bxsubscribe.go 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. package main
  2. import (
  3. logrusx "app.yhyue.com/moapp/jylogx/logx"
  4. IC "bp.jydev.jianyu360.cn/BaseService/jyMicroservices/jyBXSubscribe/rpc/init"
  5. "bp.jydev.jianyu360.cn/BaseService/jyMicroservices/jyBXSubscribe/rpc/internal/server"
  6. "bp.jydev.jianyu360.cn/BaseService/jyMicroservices/jyBXSubscribe/rpc/internal/svc"
  7. "bp.jydev.jianyu360.cn/BaseService/jyMicroservices/jyBXSubscribe/rpc/type/bxsubscribe"
  8. "flag"
  9. "fmt"
  10. _ "github.com/ClickHouse/clickhouse-go/v2"
  11. "github.com/gogf/gf/v2/frame/g"
  12. "github.com/gogf/gf/v2/os/gcfg"
  13. "github.com/zeromicro/go-zero/core/conf"
  14. "github.com/zeromicro/go-zero/core/logx"
  15. "github.com/zeromicro/go-zero/core/service"
  16. "github.com/zeromicro/go-zero/zrpc"
  17. "google.golang.org/grpc"
  18. "google.golang.org/grpc/reflection"
  19. )
  20. var configF = flag.String("cf", "etc/bxsubscribe.yaml", "the config file")
  21. var dbFile = flag.String("df", "etc/db.yaml", "the db file")
  22. var logFile = flag.String("lf", "etc/logs.yaml", "the log file")
  23. func main() {
  24. //初始化基本配置
  25. g.Cfg().GetAdapter().(*gcfg.AdapterFile).SetFileName("etc/config.yaml")
  26. //logx.SetWriter(logx.NewWriter(jylog.GetWriter()))
  27. conf.MustLoad(*configF, &IC.C)
  28. //初始化数据库配置
  29. conf.MustLoad(*dbFile, &IC.DB)
  30. //初始化日志信息
  31. conf.MustLoad(*logFile, &IC.Logc)
  32. IC.InitC()
  33. //go func() {
  34. // err := endless.ListenAndServe(":"+MC.InterfaceToStr(IC.C.Webrpcport), nil, func() {})
  35. // if err != nil {
  36. // log.Println("ListenAndServe: ", err)
  37. // }
  38. //}()
  39. ctx := svc.NewServiceContext(IC.C)
  40. svr := server.NewBxsubscribeServer(ctx)
  41. s := zrpc.MustNewServer(IC.C.RpcServerConf, func(grpcServer *grpc.Server) {
  42. bxsubscribe.RegisterBxsubscribeServer(grpcServer, svr)
  43. if IC.C.Mode == service.DevMode || IC.C.Mode == service.TestMode {
  44. reflection.Register(grpcServer)
  45. }
  46. })
  47. //日志记录
  48. logx.SetWriter(logrusx.NewLogrusWriter())
  49. //s.AddUnaryInterceptors(util.CustomInterceptors)
  50. defer s.Stop()
  51. fmt.Printf("Starting rpc server at %s...\n", IC.C.ListenOn)
  52. s.Start()
  53. }