micro/micro.go at v4 · unistack-org/micro · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package micro

import (
	"reflect"

	"go.unistack.org/micro/v4/broker"
	"go.unistack.org/micro/v4/client"
	"go.unistack.org/micro/v4/codec"
	"go.unistack.org/micro/v4/flow"
	"go.unistack.org/micro/v4/fsm"
	"go.unistack.org/micro/v4/logger"
	"go.unistack.org/micro/v4/meter"
	"go.unistack.org/micro/v4/register"
	"go.unistack.org/micro/v4/resolver"
	"go.unistack.org/micro/v4/router"
	"go.unistack.org/micro/v4/selector"
	"go.unistack.org/micro/v4/server"
	"go.unistack.org/micro/v4/store"
	"go.unistack.org/micro/v4/sync"
	"go.unistack.org/micro/v4/tracer"
)

func As(b any, target any) bool {
	if b == nil {
		return false
	}
	if target == nil {
		return false
	}
	val := reflect.ValueOf(target)
	typ := val.Type()
	if typ.Kind() != reflect.Pointer || val.IsNil() {
		return false
	}
	targetType := typ.Elem()
	if targetType.Kind() != reflect.Interface {
		switch {
		case targetType.Implements(brokerType):
			break
		case targetType.Implements(loggerType):
			break
		case targetType.Implements(clientType):
			break
		case targetType.Implements(serverType):
			break
		case targetType.Implements(codecType):
			break
		case targetType.Implements(flowType):
			break
		case targetType.Implements(fsmType):
			break
		case targetType.Implements(meterType):
			break
		case targetType.Implements(registerType):
			break
		case targetType.Implements(resolverType):
			break
		case targetType.Implements(selectorType):
			break
		case targetType.Implements(storeType):
			break
		case targetType.Implements(syncType):
			break
		case targetType.Implements(serviceType):
			break
		case targetType.Implements(routerType):
			break
		case targetType.Implements(tracerType):
			break
		default:
			return false
		}
	}
	if reflect.TypeOf(b).AssignableTo(targetType) {
		val.Elem().Set(reflect.ValueOf(b))
		return true
	}
	return false
}

var (
	brokerType   = reflect.TypeFor[broker.Broker]()
	loggerType   = reflect.TypeFor[logger.Logger]()
	clientType   = reflect.TypeFor[client.Client]()
	serverType   = reflect.TypeFor[server.Server]()
	codecType    = reflect.TypeFor[codec.Codec]()
	flowType     = reflect.TypeFor[flow.Flow]()
	fsmType      = reflect.TypeFor[fsm.FSM]()
	meterType    = reflect.TypeFor[meter.Meter]()
	registerType = reflect.TypeFor[register.Register]()
	resolverType = reflect.TypeFor[resolver.Resolver]()
	routerType   = reflect.TypeFor[router.Router]()
	selectorType = reflect.TypeFor[selector.Selector]()
	storeType    = reflect.TypeFor[store.Store]()
	syncType     = reflect.TypeFor[sync.Sync]()
	tracerType   = reflect.TypeFor[tracer.Tracer]()
	serviceType  = reflect.TypeFor[Service]()
)