45 lines
1015 B
Go
45 lines
1015 B
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
service "user-srv/Service"
|
||
|
|
"user-srv/config"
|
||
|
|
|
||
|
|
pb "user-srv/pb/api"
|
||
|
|
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
"google.golang.org/grpc/health"
|
||
|
|
"google.golang.org/grpc/health/grpc_health_v1"
|
||
|
|
)
|
||
|
|
|
||
|
|
// var userService = service.UserService{}
|
||
|
|
|
||
|
|
func StartGrpcServer() error {
|
||
|
|
listener, err := net.Listen("tcp", ":"+fmt.Sprintf("%d", config.GrpcPort))
|
||
|
|
if err != nil {
|
||
|
|
logrus.WithFields(logrus.Fields{
|
||
|
|
"func": "StartGrpcServer",
|
||
|
|
}).Warnf("net.Listen: %s", err.Error())
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
// // 注册到consul
|
||
|
|
// if err := ConsulGrpc(); err != nil {
|
||
|
|
// log.Fatalln("consulGrpc failed: ", err)
|
||
|
|
// }
|
||
|
|
server := grpc.NewServer()
|
||
|
|
pb.RegisterUserServiceServer(server, &service.UserService{})
|
||
|
|
// 健康检查
|
||
|
|
grpc_health_v1.RegisterHealthServer(server, health.NewServer())
|
||
|
|
err = server.Serve(listener)
|
||
|
|
if err != nil {
|
||
|
|
logrus.WithFields(logrus.Fields{
|
||
|
|
"func": "StartGrpcServer",
|
||
|
|
}).Warnf("server.Serve: %s", err.Error())
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
|
||
|
|
}
|