jy_docs/rpc/partnerlib/partnerlib.go

package main

import (
	IC "app.yhyue.com/moapp/jy_docs/rpc/partnerlib/init"
	"app.yhyue.com/moapp/jy_docs/rpc/partnerlib/internal/server"
	"app.yhyue.com/moapp/jy_docs/rpc/partnerlib/internal/svc"
	"app.yhyue.com/moapp/jy_docs/rpc/partnerlib/type/partnerlib"
	MC "app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/endless"
	logrusx "app.yhyue.com/moapp/jylogx/logx"
	"fmt"
	"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"
	"log"
)

func main() {
	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.NewPartnerServer(ctx)

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

		if IC.C.Mode == service.DevMode || IC.C.Mode == service.TestMode {
			reflection.Register(grpcServer)
		}
	})
	defer s.Stop()
	//日志记录
	logx.SetWriter(logrusx.NewLogrusWriter())
	//crontab.StartTask()
	fmt.Printf("Starting rpc server at %s...\n", IC.C.ListenOn)
	s.Start()
}