suguo
/
snippet
Template
1
0
Fork 0
Code Issues Pull Requests Releases 7 Activity

grpc更新

This commit is contained in:
suguo.yao 2021-09-16 10:38:37 +08:00
parent b048a2e1f2
commit 666348870e
1 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
grpc/grpc.go
View File

@ -25,9 +25,9 @@ var conf *Config
func Init(config *Config) *grpc.Server { func Init(config *Config) *grpc.Server {
if config == nil { if config == nil {
config = &Config{ config = &Config{
Address: "", Address: "0.0.0.0",
Port: 0, Port: 0,
AppName: "dev", AppName: "unknown",
} }
} }
conf = config conf = config
@ -50,24 +50,19 @@ func Start() {
//注册反射 用于grpcurl调试 //注册反射 用于grpcurl调试
reflection.Register(rpc) reflection.Register(rpc)
// grpc服务启动 // grpc服务启动，本局IP支持
go func() { go func() {
addr, err := net.ResolveTCPAddr("tcp", fmt.Sprintf("%s:%d", conf.Address, conf.Port)) addr, err := net.ResolveTCPAddr("tcp", fmt.Sprintf("%s:%d", conf.Address, conf.Port))
if err != nil { if err != nil {
log.Fatal("net.ResolveTCPAddr", err) log.Fatal("net.ResolveTCPAddr", err)
} }
if addr.IP.String() != conf.Address {
conf.Address = addr.IP.String()
}
if addr.Port != conf.Port {
conf.Port = addr.Port
}
log.Printf("starting grpc service on %s:%d", addr.IP, addr.Port)
lis, err := net.ListenTCP("tcp", addr) lis, err := net.ListenTCP("tcp", addr)
if err != nil { if err != nil {
log.Fatal("fail to open port: ", err) log.Fatal("fail to open port: ", err)
} }
defer lis.Close() defer lis.Close()
conf.Port = lis.Addr().(*net.TCPAddr).Port
log.Printf("starting grpc service on %s:%d", addr.IP, conf.Port)
err = rpc.Serve(lis) err = rpc.Serve(lis)
if err != nil { if err != nil {
log.Fatal("fail to open microservice: ", err) log.Fatal("fail to open microservice: ", err)