66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"framework_v2/internal/providers"
|
|
"framework_v2/internal/providers/interceptor"
|
|
"github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/auth"
|
|
"github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/logging"
|
|
"github.com/spf13/cobra"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/reflection"
|
|
|
|
"net"
|
|
)
|
|
|
|
var rpcCommand = &cobra.Command{
|
|
Use: "serve",
|
|
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
providers.InitConfig()
|
|
providers.InitLogger()
|
|
providers.InitEnt()
|
|
providers.InitRedis()
|
|
providers.InitJwksRefresh()
|
|
providers.InitAsynQClient()
|
|
|
|
StartSpiderService()
|
|
},
|
|
}
|
|
|
|
func StartSpiderService() {
|
|
if providers.Config.GRPC.GrpcListenAddr == "" {
|
|
providers.Config.GRPC.GrpcListenAddr = "0.0.0.0:8081"
|
|
}
|
|
|
|
lis, err := net.Listen("tcp", providers.Config.GRPC.GrpcListenAddr)
|
|
if err != nil {
|
|
panic("failed to listen: " + err.Error())
|
|
}
|
|
providers.Logger.Info("Server listening at " + providers.Config.GRPC.GrpcListenAddr)
|
|
|
|
var opts = []grpc.ServerOption{
|
|
grpc.ChainUnaryInterceptor(
|
|
logging.UnaryServerInterceptor(interceptor.ZapLogInterceptor()),
|
|
auth.UnaryServerInterceptor(interceptor.JwtAuth),
|
|
),
|
|
grpc.ChainStreamInterceptor(
|
|
logging.StreamServerInterceptor(interceptor.ZapLogInterceptor()),
|
|
auth.StreamServerInterceptor(interceptor.JwtAuth),
|
|
),
|
|
}
|
|
grpcServer := grpc.NewServer(opts...)
|
|
|
|
reflection.Register(grpcServer)
|
|
|
|
// 如需实现，需要手动实现这里
|
|
//library.RegisterLibraryServiceServer(grpcServer, &services.LibraryService{})
|
|
//document.RegisterDocumentServiceServer(grpcServer, &services.DocumentService{})
|
|
|
|
err = grpcServer.Serve(lis)
|
|
if err != nil {
|
|
panic(err)
|
|
return
|
|
}
|
|
|
|
}
|