29 lines
668 B
Go
29 lines
668 B
Go
package grpc
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go-common/app/service/bbq/push/api/grpc/v1"
|
|
"go-common/app/service/bbq/push/service"
|
|
"go-common/library/net/rpc/warden"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
//New 生成rpc服务
|
|
func New(conf *warden.ServerConfig, srv *service.Service) (s *warden.Server, err error) {
|
|
s = warden.NewServer(conf)
|
|
s.Use(middleware())
|
|
v1.RegisterPushServer(s.Server(), srv)
|
|
_, err = s.Start()
|
|
return
|
|
}
|
|
|
|
func middleware() grpc.UnaryServerInterceptor {
|
|
return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
|
|
//call chain
|
|
resp, err = handler(ctx, req)
|
|
return
|
|
}
|
|
}
|