|
@@ -17,21 +17,20 @@ import (
|
|
|
)
|
|
|
|
|
|
var configFile = flag.String("f", "etc/userlib.yaml", "the config file")
|
|
|
-var c config.Config
|
|
|
func main() {
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
|
- conf.MustLoad(*configFile, &c)
|
|
|
- if c.JyDocsMysqlDB.DataSourceName != "" {
|
|
|
- jyDocsRpcUtil.InitDB(c.JyDocsMysqlDB.DataSourceName, c.JyDocsMysqlDB.DriverName, c.JyDocsMysqlDB.MaxOpenConn, c.JyDocsMysqlDB.MaxIdleConn)
|
|
|
- ctx := svc.NewServiceContext(c)
|
|
|
+ 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(c.RpcServerConf, func(grpcServer *grpc.Server) {
|
|
|
+ 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", c.ListenOn)
|
|
|
+ fmt.Printf("Starting rpc server at %s...\n", config.Configs.ListenOn)
|
|
|
s.Start()
|
|
|
} else {
|
|
|
log.Fatal("【jy_docs】 read config error!")
|