You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.3 KiB
Go
91 lines
2.3 KiB
Go
package internal
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/tal-tech/go-zero/core/proc"
|
|
"github.com/tal-tech/go-zero/core/stat"
|
|
"github.com/tal-tech/go-zero/zrpc/internal/serverinterceptors"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type (
|
|
// ServerOption defines the method to customize a rpcServerOptions.
|
|
ServerOption func(options *rpcServerOptions)
|
|
|
|
rpcServerOptions struct {
|
|
metrics *stat.Metrics
|
|
}
|
|
|
|
rpcServer struct {
|
|
name string
|
|
*baseRpcServer
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
InitLogger()
|
|
}
|
|
|
|
// NewRpcServer returns a Server.
|
|
func NewRpcServer(address string, opts ...ServerOption) Server {
|
|
var options rpcServerOptions
|
|
for _, opt := range opts {
|
|
opt(&options)
|
|
}
|
|
if options.metrics == nil {
|
|
options.metrics = stat.NewMetrics(address)
|
|
}
|
|
|
|
return &rpcServer{
|
|
baseRpcServer: newBaseRpcServer(address, options.metrics),
|
|
}
|
|
}
|
|
|
|
func (s *rpcServer) SetName(name string) {
|
|
s.name = name
|
|
s.baseRpcServer.SetName(name)
|
|
}
|
|
|
|
func (s *rpcServer) Start(register RegisterFn) error {
|
|
lis, err := net.Listen("tcp", s.address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
unaryInterceptors := []grpc.UnaryServerInterceptor{
|
|
serverinterceptors.UnaryTracingInterceptor(s.name),
|
|
serverinterceptors.UnaryCrashInterceptor(),
|
|
serverinterceptors.UnaryStatInterceptor(s.metrics),
|
|
serverinterceptors.UnaryPrometheusInterceptor(),
|
|
serverinterceptors.UnaryBreakerInterceptor(),
|
|
serverinterceptors.UnaryOpenTracingInterceptor(),
|
|
}
|
|
unaryInterceptors = append(unaryInterceptors, s.unaryInterceptors...)
|
|
streamInterceptors := []grpc.StreamServerInterceptor{
|
|
serverinterceptors.StreamCrashInterceptor,
|
|
serverinterceptors.StreamBreakerInterceptor,
|
|
serverinterceptors.StreamOpenTracingInterceptor(),
|
|
}
|
|
streamInterceptors = append(streamInterceptors, s.streamInterceptors...)
|
|
options := append(s.options, WithUnaryServerInterceptors(unaryInterceptors...),
|
|
WithStreamServerInterceptors(streamInterceptors...))
|
|
server := grpc.NewServer(options...)
|
|
register(server)
|
|
// we need to make sure all others are wrapped up
|
|
// so we do graceful stop at shutdown phase instead of wrap up phase
|
|
waitForCalled := proc.AddWrapUpListener(func() {
|
|
server.GracefulStop()
|
|
})
|
|
defer waitForCalled()
|
|
|
|
return server.Serve(lis)
|
|
}
|
|
|
|
// WithMetrics returns a func that sets metrics to a Server.
|
|
func WithMetrics(metrics *stat.Metrics) ServerOption {
|
|
return func(options *rpcServerOptions) {
|
|
options.metrics = metrics
|
|
}
|
|
}
|