99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package service
|
|
|
|
import (
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"git.apinb.com/bsm-sdk/core/conf"
|
|
"git.apinb.com/bsm-sdk/core/env"
|
|
"git.apinb.com/bsm-sdk/core/print"
|
|
"git.apinb.com/bsm-sdk/core/vars"
|
|
clientv3 "go.etcd.io/etcd/client/v3"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type (
|
|
// RegisterFn defines the method to register a server.
|
|
RegisterFn func(*grpc.Server)
|
|
|
|
Service struct {
|
|
GrpcSrv *grpc.Server
|
|
Opts *Options
|
|
}
|
|
|
|
Options struct {
|
|
Addr string
|
|
Conf *conf.MicroServiceConf
|
|
EtcdClient *clientv3.Client
|
|
}
|
|
)
|
|
|
|
func New(srv *grpc.Server, opts *Options) *Service {
|
|
return &Service{GrpcSrv: srv, Opts: opts}
|
|
}
|
|
|
|
func Addr(ip string, port int) string {
|
|
return net.JoinHostPort(ip, strconv.Itoa(port))
|
|
}
|
|
|
|
func (s *Service) Start() {
|
|
print.Info("[BSM - %s] Service Starting ...", vars.ServiceKey)
|
|
|
|
// register to etcd.
|
|
if s.Opts.Conf != nil && s.Opts.Conf.Enable {
|
|
if s.Opts.EtcdClient == nil {
|
|
print.Error("[BSM Register] Etcd Client is nil.")
|
|
os.Exit(1)
|
|
}
|
|
// get methods
|
|
methods := FoundGrpcMethods(s.GrpcSrv)
|
|
|
|
// set router key
|
|
routerKey := vars.ServiceRootPrefix + "Router/" + env.Runtime.Workspace + "/" + strings.ToLower(vars.ServiceKey) + "/" + s.Opts.Addr
|
|
|
|
// register to etcd
|
|
register, err := RegisterService(s.Opts.EtcdClient, routerKey, methods, vars.ServiceLease)
|
|
if err != nil {
|
|
log.Panicf("[ERROR] %s Service Register:%s \n", vars.ServiceKey, err.Error())
|
|
}
|
|
|
|
anonKey := vars.ServiceRootPrefix + "Router/" + env.Runtime.Workspace + "/"
|
|
|
|
register.SetAnonymous(anonKey, s.Opts.Conf.Anonymous)
|
|
|
|
// service register lease
|
|
go register.ListenLeaseRespChan()
|
|
}
|
|
|
|
// run grpc srv.
|
|
tcpListen, err := net.Listen("tcp", s.Opts.Addr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
print.Success("[BSM - %s] Service Grpc Register & Runing Success!", vars.ServiceKey)
|
|
if err := s.GrpcSrv.Serve(tcpListen); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
}
|
|
|
|
func (s *Service) Stop() {
|
|
s.GrpcSrv.GracefulStop()
|
|
}
|
|
|
|
// found grpc methods.
|
|
func FoundGrpcMethods(s *grpc.Server) []string {
|
|
var mothods []string
|
|
for key, srv := range s.GetServiceInfo() {
|
|
srvName := strings.Split(key, ".")[1]
|
|
for _, mn := range srv.Methods {
|
|
mothods = append(mothods, srvName+"."+mn.Name)
|
|
}
|
|
}
|
|
return mothods
|
|
}
|