jy_docs/rpc/stdlib/stdlib.go

package main

import (
	"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"
	"app.yhyue.com/moapp/jy_docs/rpc/stdlib/stdlib"
	jyDocsRpcUtil "app.yhyue.com/moapp/jy_docs/services/util"
	"flag"
	"github.com/tal-tech/go-zero/core/conf"
	"github.com/tal-tech/go-zero/zrpc"
	"google.golang.org/grpc"
	"log"
)

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)
		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!")
	}

}