psrpc/pkg/server/registration.go

// Copyright 2023 LiveKit, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package server

import "reflect"

type Registerer struct {
	register   any
	deregister any
}

func NewRegisterer(register, deregister any) Registerer {
	return Registerer{register, deregister}
}

func anySliceReflectValues(anys []any) []reflect.Value {
	vals := make([]reflect.Value, len(anys))
	for i, a := range anys {
		vals[i] = reflect.ValueOf(a)
	}
	return vals
}

type RegistererSlice []Registerer

func (rs RegistererSlice) Register(params ...any) error {
	paramVals := anySliceReflectValues(params)
	for i, r := range rs {
		ret := reflect.ValueOf(r.register).Call(paramVals)
		if !ret[0].IsNil() {
			rs[:i].Deregister(params...)
			return ret[0].Interface().(error)
		}
	}
	return nil
}

func (rs RegistererSlice) Deregister(params ...any) {
	paramVals := anySliceReflectValues(params)
	for _, r := range rs {
		reflect.ValueOf(r.deregister).Call(paramVals)
	}
}