-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
64 lines (55 loc) · 1.49 KB
/
main.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
package main
import (
"flag"
"fmt"
"net"
"os"
"os/signal"
"syscall"
"github.com/palp1tate/brevinect/service/meeting/global"
"github.com/palp1tate/brevinect/service/meeting/initialize"
"github.com/palp1tate/brevinect/util"
"go.uber.org/zap"
)
func main() {
initialize.InitConfig()
initialize.InitLogger()
initialize.InitMySQL()
host := global.ServerConfig.Service.Host
port := flag.Int("p", 0, "端口号")
flag.Parse()
if *port == 0 {
*port, _ = util.GetFreePort()
}
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", *port))
if err != nil {
zap.S().Panic("failed to listen:" + err.Error())
}
server, closer := initialize.InitGRPC()
zap.S().Info("host: ", host)
zap.S().Info("port: ", *port)
go func() {
err = server.Serve(lis)
if err != nil {
zap.S().Panic("failed to serve:" + err.Error())
}
}()
client := initialize.NewRegistryClient(global.ServerConfig.Consul.Host, global.ServerConfig.Consul.Port)
serviceName := global.ServerConfig.Service.Name
serviceTags := global.ServerConfig.Service.Tags
serviceId := util.GenerateUUID()
err = client.Register(host, *port, serviceName, serviceTags, serviceId)
if err != nil {
zap.S().Panic("服务注册失败:", err.Error())
}
quit := make(chan os.Signal)
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
<-quit
defer closer.Close()
server.GracefulStop()
if err = client.DeRegister(serviceId); err != nil {
zap.S().Warnf("%s注销失败", serviceName)
} else {
zap.S().Infof("%s注销成功", serviceName)
}
}