复制项目

pkg/common/startrpc/tools.go

@@ -0,0 +1,47 @@
+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)
+}