-
-
Notifications
You must be signed in to change notification settings - Fork 1.5k
/
gsvc_discovery.go
109 lines (97 loc) · 2.96 KB
/
gsvc_discovery.go
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
99
100
101
102
103
104
105
106
107
108
109
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://github.com/gogf/gf.
package gsvc
import (
"context"
"time"
"github.com/gogf/gf/v2/container/gmap"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/internal/intlog"
"github.com/gogf/gf/v2/util/gutil"
)
// watchedServiceMap stores used service
var watchedServiceMap = gmap.New(true)
// ServiceWatch is used to watch the service status.
type ServiceWatch func(service Service)
// Get retrieves and returns the service by service name.
func Get(ctx context.Context, name string) (service Service, err error) {
return GetAndWatch(ctx, name, nil)
}
// GetAndWatch is used to getting the service with custom watch callback function.
func GetAndWatch(ctx context.Context, name string, watch ServiceWatch) (service Service, err error) {
v := watchedServiceMap.GetOrSetFuncLock(name, func() interface{} {
var (
services []Service
watcher Watcher
)
services, err = Search(ctx, SearchInput{
Name: name,
})
if err != nil {
return nil
}
if len(services) == 0 {
err = gerror.NewCodef(gcode.CodeNotFound, `service not found with name "%s"`, name)
return nil
}
// Just pick one if multiple.
service = services[0]
// Watch the service changes in goroutine.
if watch != nil {
if watcher, err = Watch(ctx, service.GetPrefix()); err != nil {
return nil
}
go watchAndUpdateService(watcher, service, watch)
}
return service
})
if v != nil {
service = v.(Service)
}
return
}
// watchAndUpdateService watches and updates the service in memory if it is changed.
func watchAndUpdateService(watcher Watcher, service Service, watchFunc ServiceWatch) {
var (
ctx = context.Background()
err error
services []Service
)
for {
time.Sleep(time.Second)
services, err = watcher.Proceed()
if err != nil {
intlog.Errorf(ctx, `%+v`, err)
continue
}
if len(services) > 0 {
watchedServiceMap.Set(service.GetName(), services[0])
if watchFunc != nil {
gutil.TryCatch(ctx, func(ctx context.Context) {
watchFunc(services[0])
}, func(ctx context.Context, exception error) {
intlog.Errorf(ctx, `%+v`, exception)
})
}
}
}
}
// Search searches and returns services with specified condition.
func Search(ctx context.Context, in SearchInput) ([]Service, error) {
if defaultRegistry == nil {
return nil, gerror.NewCodef(gcode.CodeNotImplemented, `no Registry is registered`)
}
ctx, _ = context.WithTimeout(ctx, defaultTimeout)
return defaultRegistry.Search(ctx, in)
}
// Watch watches specified condition changes.
func Watch(ctx context.Context, key string) (Watcher, error) {
if defaultRegistry == nil {
return nil, gerror.NewCodef(gcode.CodeNotImplemented, `no Registry is registered`)
}
return defaultRegistry.Watch(ctx, key)
}