package main import ( jyDocsRpcUtil "app.yhyue.com/moapp/jy_docs/services/util" "flag" "fmt" "log" "app.yhyue.com/moapp/jy_docs/rpc/userlib/internal/config" "app.yhyue.com/moapp/jy_docs/rpc/userlib/internal/server" "app.yhyue.com/moapp/jy_docs/rpc/userlib/internal/svc" "app.yhyue.com/moapp/jy_docs/rpc/userlib/userlib" "github.com/tal-tech/go-zero/core/conf" "github.com/tal-tech/go-zero/zrpc" "google.golang.org/grpc" ) var configFile = flag.String("f", "etc/userlib.yaml", "the config file") func main() { flag.Parse() conf.MustLoad(*configFile, &config.Configs) if config.Configs.JyDocsMysqlDB.DataSourceName != "" { jyDocsRpcUtil.InitDB(config.Configs.JyDocsMysqlDB.DataSourceName, config.Configs.JyDocsMysqlDB.DriverName, config.Configs.JyDocsMysqlDB.MaxOpenConn, config.Configs.JyDocsMysqlDB.MaxIdleConn) ctx := svc.NewServiceContext(config.Configs) srv := server.NewUserLibServer(ctx) s := zrpc.MustNewServer(config.Configs.RpcServerConf, func(grpcServer *grpc.Server) { userlib.RegisterUserLibServer(grpcServer, srv) }) defer s.Stop() fmt.Printf("Starting rpc server at %s...\n", config.Configs.ListenOn) s.Start() } else { log.Fatal("【jy_docs】 read config error!") } }