package main import ( "app.yhyue.com/moapp/jy_docs/rpc/stdlib/type/stdlib" "flag" "log" "app.yhyue.com/moapp/jy_docs/rpc/stdlib/internal/config" "app.yhyue.com/moapp/jy_docs/rpc/stdlib/internal/server" "app.yhyue.com/moapp/jy_docs/rpc/stdlib/internal/svc" jyDocsRpcpartner "app.yhyue.com/moapp/jy_docs/services/partner" jyDocsRpcUtil "app.yhyue.com/moapp/jy_docs/services/util" "github.com/zeromicro/go-zero/core/conf" "github.com/zeromicro/go-zero/zrpc" "google.golang.org/grpc" ) var configFile = flag.String("f", "etc/stdlib.yaml", "the config file") func main() { flag.Parse() var c config.Config conf.MustLoad(*configFile, &c) if c.JyDocsMysqlDB.DataSourceName != "" && c.EsConfig.Addr != "" { jyDocsRpcUtil.InitDB(c.JyDocsMysqlDB.DataSourceName, c.JyDocsMysqlDB.DriverName, c.JyDocsMysqlDB.MaxOpenConn, c.JyDocsMysqlDB.MaxIdleConn) jyDocsRpcUtil.InitEs(c.EsConfig.Addr, c.EsConfig.Pool, c.EsConfig.UserName, c.EsConfig.Password) jyDocsRpcpartner.InitDocClass() // jyDocsRpcUtil.InitOss(c.FileSystemConf) ctx := svc.NewServiceContext(c) srv := server.NewStdlibServer(ctx) s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) { stdlib.RegisterStdlibServer(grpcServer, srv) }) defer s.Stop() log.Printf("Starting rpc server at %s...\n", c.ListenOn) s.Start() } else { log.Fatal("【jy_docs】 read config error!") } }