generated from CyanAsterisk/.github
/
registry.go
44 lines (40 loc) · 1.38 KB
/
registry.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
package initialize
import (
"net"
"strconv"
"github.com/CyanAsterisk/FreeCar/server/cmd/trip/config"
"github.com/CyanAsterisk/FreeCar/server/shared/consts"
"github.com/bwmarrin/snowflake"
"github.com/cloudwego/kitex/pkg/klog"
"github.com/cloudwego/kitex/pkg/registry"
"github.com/cloudwego/kitex/pkg/utils"
"github.com/hashicorp/consul/api"
consul "github.com/kitex-contrib/registry-consul"
)
// InitRegistry to init consul
func InitRegistry(Port int) (registry.Registry, *registry.Info) {
r, err := consul.NewConsulRegister(net.JoinHostPort(
config.GlobalConsulConfig.Host,
strconv.Itoa(config.GlobalConsulConfig.Port)),
consul.WithCheck(&api.AgentServiceCheck{
Interval: consts.ConsulCheckInterval,
Timeout: consts.ConsulCheckTimeout,
DeregisterCriticalServiceAfter: consts.ConsulCheckDeregisterCriticalServiceAfter,
}))
if err != nil {
klog.Fatalf("new consul register failed: %s", err.Error())
}
// Using snowflake to generate service name.
sf, err := snowflake.NewNode(4)
if err != nil {
klog.Fatalf("generate service name failed: %s", err.Error())
}
info := ®istry.Info{
ServiceName: config.GlobalServerConfig.Name,
Addr: utils.NewNetAddr(consts.TCP, net.JoinHostPort(config.GlobalServerConfig.Host, strconv.Itoa(Port))),
Tags: map[string]string{
"ID": sf.Generate().Base36(),
},
}
return r, info
}