48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package startrpc
|
|
|
|
import (
|
|
"reflect"
|
|
|
|
conf "git.imall.cloud/openim/open-im-server-deploy/pkg/common/config"
|
|
)
|
|
|
|
func getConfig[T any](value reflect.Value) *T {
|
|
for value.Kind() == reflect.Pointer {
|
|
value = value.Elem()
|
|
}
|
|
if value.Kind() == reflect.Struct {
|
|
num := value.NumField()
|
|
for i := 0; i < num; i++ {
|
|
field := value.Field(i)
|
|
for field.Kind() == reflect.Pointer {
|
|
field = field.Elem()
|
|
}
|
|
if field.Kind() == reflect.Struct {
|
|
if elem, ok := field.Interface().(T); ok {
|
|
return &elem
|
|
}
|
|
if elem := getConfig[T](field); elem != nil {
|
|
return elem
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func getConfigRpcMaxRequestBody(value reflect.Value) *conf.MaxRequestBody {
|
|
return getConfig[conf.MaxRequestBody](value)
|
|
}
|
|
|
|
func getConfigShare(value reflect.Value) *conf.Share {
|
|
return getConfig[conf.Share](value)
|
|
}
|
|
|
|
func getConfigRateLimiter(value reflect.Value) *conf.RateLimiter {
|
|
return getConfig[conf.RateLimiter](value)
|
|
}
|
|
|
|
func getConfigCircuitBreaker(value reflect.Value) *conf.CircuitBreaker {
|
|
return getConfig[conf.CircuitBreaker](value)
|
|
}
|