ai update.
This commit is contained in:
@@ -1,3 +1,5 @@
|
||||
// Package service 提供微服务管理功能
|
||||
// 包括服务启动、注册、网关代理等核心功能
|
||||
package service
|
||||
|
||||
import (
|
||||
@@ -21,50 +23,59 @@ import (
|
||||
)
|
||||
|
||||
type (
|
||||
// RegisterFn defines the method to register a server.
|
||||
// RegisterFn 定义服务注册函数类型
|
||||
RegisterFn func(*grpc.Server)
|
||||
|
||||
// Service 微服务实例
|
||||
Service struct {
|
||||
GrpcSrv *grpc.Server
|
||||
Opts *Options
|
||||
GrpcSrv *grpc.Server // gRPC服务器实例
|
||||
Opts *Options // 服务配置选项
|
||||
}
|
||||
|
||||
// Options 服务配置选项
|
||||
Options struct {
|
||||
Addr string
|
||||
EtcdClient *clientv3.Client
|
||||
MsConf *conf.MicroServiceConf
|
||||
GatewayConf *conf.GatewayConf
|
||||
GatewayCtx context.Context
|
||||
GatewayMux *gwRuntime.ServeMux
|
||||
Addr string // 服务监听地址
|
||||
EtcdClient *clientv3.Client // Etcd客户端
|
||||
MsConf *conf.MicroServiceConf // 微服务配置
|
||||
GatewayConf *conf.GatewayConf // 网关配置
|
||||
GatewayCtx context.Context // 网关上下文
|
||||
GatewayMux *gwRuntime.ServeMux // 网关多路复用器
|
||||
}
|
||||
)
|
||||
|
||||
// New 创建新的服务实例
|
||||
// srv: gRPC服务器实例
|
||||
// opts: 服务配置选项
|
||||
func New(srv *grpc.Server, opts *Options) *Service {
|
||||
return &Service{GrpcSrv: srv, Opts: opts}
|
||||
}
|
||||
|
||||
// Addr formats ip and port into host:port.
|
||||
// Addr 将IP和端口格式化为host:port格式
|
||||
// ip: IP地址
|
||||
// port: 端口号
|
||||
func Addr(ip string, port int) string {
|
||||
return net.JoinHostPort(ip, strconv.Itoa(port))
|
||||
}
|
||||
|
||||
// Start 启动服务
|
||||
// 包括etcd注册、gRPC服务启动、HTTP网关启动
|
||||
func (s *Service) Start() {
|
||||
printer.Info("[BSM - %s] Service Starting ...", vars.ServiceKey)
|
||||
|
||||
// register to etcd.
|
||||
// 注册到etcd
|
||||
if s.Opts.MsConf != nil && s.Opts.MsConf.Enable {
|
||||
if s.Opts.EtcdClient == nil {
|
||||
printer.Error("[BSM Register] Etcd Client is nil.")
|
||||
os.Exit(1)
|
||||
}
|
||||
printer.Info("[BSM - %s] Registering Service to Etcd ...", vars.ServiceKey)
|
||||
// get methods for gateway/router discovery
|
||||
// 获取gRPC方法用于网关/路由发现
|
||||
methods := FoundGrpcMethods(s.GrpcSrv)
|
||||
|
||||
// set router key
|
||||
// 设置路由键
|
||||
routerKey := vars.ServiceRootPrefix + "Router/" + env.Runtime.Workspace + "/" + strings.ToLower(vars.ServiceKey) + "/" + s.Opts.Addr
|
||||
|
||||
// register to etcd with lease
|
||||
// 使用租约注册到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())
|
||||
@@ -74,11 +85,11 @@ func (s *Service) Start() {
|
||||
|
||||
register.SetAnonymous(anonKey, s.Opts.MsConf.Anonymous)
|
||||
|
||||
// service register lease
|
||||
// 服务注册租约监听
|
||||
go register.ListenLeaseRespChan()
|
||||
}
|
||||
|
||||
// run grpc srv.
|
||||
// 启动gRPC服务
|
||||
tcpListen, err := net.Listen("tcp", s.Opts.Addr)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
@@ -91,6 +102,7 @@ func (s *Service) Start() {
|
||||
}()
|
||||
printer.Success("[BSM - %s] Grpc %s Runing Success !", vars.ServiceKey, s.Opts.Addr)
|
||||
|
||||
// 启动HTTP网关
|
||||
if s.Opts.GatewayConf != nil && s.Opts.GatewayConf.Enable {
|
||||
addr := Addr("0.0.0.0", s.Opts.GatewayConf.Port)
|
||||
go s.Gateway(s.Opts.Addr, addr)
|
||||
@@ -98,20 +110,26 @@ func (s *Service) Start() {
|
||||
printer.Success("[BSM - %s] Http %s Runing Success!", vars.ServiceKey, addr)
|
||||
}
|
||||
|
||||
// 阻塞主线程
|
||||
select {}
|
||||
}
|
||||
|
||||
// Gateway 启动HTTP网关服务
|
||||
// grpcAddr: gRPC服务地址
|
||||
// httpAddr: HTTP服务地址
|
||||
func (s *Service) Gateway(grpcAddr string, httpAddr string) {
|
||||
// 1. 定义一个context
|
||||
// 定义上下文
|
||||
_, cancel := context.WithCancel(s.Opts.GatewayCtx)
|
||||
defer cancel()
|
||||
|
||||
// 不因 HTTP 启动失败而导致 panic
|
||||
// 启动HTTP服务,不因HTTP启动失败而导致panic
|
||||
if err := http.ListenAndServe(httpAddr, s.Opts.GatewayMux); err != nil {
|
||||
printer.Error("[BSM - %s] Http Serve Error: %v", vars.ServiceKey, err)
|
||||
}
|
||||
}
|
||||
|
||||
// Use 执行初始化函数
|
||||
// initFunc: 初始化函数
|
||||
func (s *Service) Use(initFunc func() error) {
|
||||
err := (initFunc)()
|
||||
if err != nil {
|
||||
@@ -120,11 +138,14 @@ func (s *Service) Use(initFunc func() error) {
|
||||
}
|
||||
}
|
||||
|
||||
// Stop 优雅停止服务
|
||||
func (s *Service) Stop() {
|
||||
s.GrpcSrv.GracefulStop()
|
||||
}
|
||||
|
||||
// found grpc methods.
|
||||
// FoundGrpcMethods 发现gRPC服务中的所有方法
|
||||
// s: gRPC服务器实例
|
||||
// 返回: 方法名列表
|
||||
func FoundGrpcMethods(s *grpc.Server) []string {
|
||||
var mothods []string
|
||||
for key, srv := range s.GetServiceInfo() {
|
||||
|
||||
Reference in New Issue
Block a user