openbilibili/app/service/bbq/push/server/grpc/server.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
	}
}