jyMicroservices/jyBXSubscribe/rpc/bxsubscribe.go

package main

import (
	MC "app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/endless"

	logrusx "app.yhyue.com/moapp/jylogx/logx"
	"flag"
	"fmt"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gcfg"
	"github.com/zeromicro/go-zero/core/conf"
	IC "jyBXSubscribe/rpc/init"
	"jyBXSubscribe/rpc/internal/server"
	"jyBXSubscribe/rpc/internal/svc"
	"jyBXSubscribe/rpc/type/bxsubscribe"
	"log"

	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/service"
	"github.com/zeromicro/go-zero/zrpc"
	"google.golang.org/grpc"
	"google.golang.org/grpc/reflection"
)

var configF = flag.String("cf", "etc/bxsubscribe.yaml", "the config file")
var dbFile = flag.String("df", "etc/db.yaml", "the db file")
var logFile = flag.String("lf", "etc/logs.yaml", "the log file")

func main() {
	//初始化基本配置
	g.Cfg().GetAdapter().(*gcfg.AdapterFile).SetFileName("etc/config.yaml")
	//logx.SetWriter(logx.NewWriter(jylog.GetWriter()))
	conf.MustLoad(*configF, &IC.C)
	//初始化数据库配置
	conf.MustLoad(*dbFile, &IC.DB)
	//初始化日志信息
	conf.MustLoad(*logFile, &IC.Logc)
	IC.InitC()

	go func() {
		err := endless.ListenAndServe(":"+MC.InterfaceToStr(IC.C.Webrpcport), nil, func() {})
		if err != nil {
			log.Println("ListenAndServe: ", err)
		}
	}()

	ctx := svc.NewServiceContext(IC.C)
	svr := server.NewBxsubscribeServer(ctx)

	s := zrpc.MustNewServer(IC.C.RpcServerConf, func(grpcServer *grpc.Server) {
		bxsubscribe.RegisterBxsubscribeServer(grpcServer, svr)

		if IC.C.Mode == service.DevMode || IC.C.Mode == service.TestMode {
			reflection.Register(grpcServer)
		}
	})

	//日志记录
	logx.SetWriter(logrusx.NewLogrusWriter())
	//s.AddUnaryInterceptors(util.CustomInterceptors)
	defer s.Stop()
	fmt.Printf("Starting rpc server at %s...\n", IC.C.ListenOn)
	s.Start()
}