-
Notifications
You must be signed in to change notification settings - Fork 264
/
discovery.go
56 lines (50 loc) · 1.13 KB
/
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
package service
import (
"fmt"
"net"
"github.com/ixre/go2o/core/etcd"
"github.com/ixre/gof/log"
clientv3 "go.etcd.io/etcd/client/v3"
)
/**
* Copyright (C) 2007-2020 56X.NET,All rights reserved.
*
* name : discovery.go
* author : jarrysix (jarrysix#gmail.com)
* date : 2020-09-08 07:02
* description :
* history :
*/
var service = "Go2oService"
var ttl int64 = 10
// RegisterServiceDiscovery 注册服务发现
func RegisterServiceDiscovery(cfg *clientv3.Config, host string, port int) {
r, err := etcd.NewRegistry(service, ttl, *cfg)
if err != nil {
panic(err)
}
ip := host
if len(ip) == 0 {
ip = resolveIp()
}
_, err = r.Register(ip, port)
if err != nil {
panic(err)
}
log.Println(fmt.Sprintf("[ GO2O][ INFO]: service registration discovery succeeded. node: %s:%d", ip, port))
}
func resolveIp() string {
addrList, err := net.InterfaceAddrs()
if err != nil {
log.Fatalln(err.Error())
}
for _, address := range addrList {
// 检查ip地址判断是否回环地址
if i, ok := address.(*net.IPNet); ok && !i.IP.IsLoopback() {
if i.IP.To4() != nil {
return i.IP.String()
}
}
}
return "127.0.0.1"
}