package main import ( "flag" "fmt" "bp.jydev.jianyu360.cn/BaseService/fileCenter/rpc/internal/config" "bp.jydev.jianyu360.cn/BaseService/fileCenter/rpc/internal/server" "bp.jydev.jianyu360.cn/BaseService/fileCenter/rpc/internal/svc" "bp.jydev.jianyu360.cn/BaseService/fileCenter/rpc/pb" "bp.jydev.jianyu360.cn/BaseService/fileCenter/entity" "github.com/zeromicro/go-zero/core/conf" "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 configFile = flag.String("f", "etc/filecenter.yaml", "the config file") // var logFile = flag.String("lf", "etc/log.yaml", "the config file") func main() { flag.Parse() var c config.Config conf.MustLoad(*configFile, &config.C) c = config.C //初始化日志信息 conf.MustLoad(*logFile, &config.LogC) if len(config.LogC.Level) > 0 { for _, v := range config.LogC.Level { logx.MustSetup(logx.LogConf{ Mode: config.LogC.Mode, Path: config.LogC.Path, Level: v, KeepDays: config.LogC.KeepDays, }) logx.Info(v, "--日志记录") } } //oss初始化 entity.OssInit(c.Oss.OssEndpoint, c.Oss.OssAccessKeyID, c.Oss.OssAccessKeySecret) ctx := svc.NewServiceContext(c) svr := server.NewFileCenterServer(ctx) s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) { pb.RegisterFileCenterServer(grpcServer, svr) if c.Mode == service.DevMode || c.Mode == service.TestMode { reflection.Register(grpcServer) } }) defer s.Stop() fmt.Printf("Starting rpc server at %s...\n", c.ListenOn) s.Start() }