jy_docs/rpc/userlib/userlib.go

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()

	var c config.Config
	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)
		srv := server.NewUserLibServer(ctx)
		s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
			userlib.RegisterUserLibServer(grpcServer, srv)
		})
		defer s.Stop()
		fmt.Printf("Starting rpc server at %s...\n", c.ListenOn)
		s.Start()
	} else {
		log.Fatal("【jy_docs】 read config error!")
	}
}