biservice.go 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. // Code generated by goctl. DO NOT EDIT!
  2. // Source: biService.proto
  3. package main
  4. import (
  5. "bp.jydev.jianyu360.cn/BaseService/biService/entity"
  6. "bp.jydev.jianyu360.cn/BaseService/biService/rpc/internal/config"
  7. "bp.jydev.jianyu360.cn/BaseService/biService/rpc/internal/server"
  8. "bp.jydev.jianyu360.cn/BaseService/biService/rpc/internal/svc"
  9. "bp.jydev.jianyu360.cn/BaseService/biService/rpc/pb"
  10. "bp.jydev.jianyu360.cn/BaseService/fileCenter/rpc/filecenter"
  11. "flag"
  12. "fmt"
  13. "github.com/nsqio/go-nsq"
  14. "github.com/zeromicro/go-zero/core/conf"
  15. "github.com/zeromicro/go-zero/core/logx"
  16. "github.com/zeromicro/go-zero/zrpc"
  17. "google.golang.org/grpc"
  18. )
  19. var configFile = flag.String("f", "etc/biservice.yaml", "the config file")
  20. func main() {
  21. flag.Parse()
  22. var c config.Config
  23. conf.MustLoad(*configFile, &c)
  24. ctx := svc.NewServiceContext(c)
  25. srv := server.NewBiServiceServer(ctx)
  26. logx.MustSetup(c.Logx)
  27. entity.PublicKey = c.PublicKey
  28. entity.InitMysql(c.Mysql.JianYu, c.Mysql.JyDoc, c.Mysql.Bi, c.Mysql.Tidb, c.Mysql.BiTidb, c.Mysql.CallTidb, c.Mysql.BiService)
  29. entity.InitMail(c.Mail)
  30. entity.InitMongo(c.Mongo.Qfw.MongodbAddr, c.Mongo.Qfw.DbName, c.Mongo.Qfw.Size)
  31. entity.InitBiddingMgo(c.Mongo.Bidding.MongodbAddr, c.Mongo.Bidding.DbName, c.Mongo.Bidding.UserName, c.Mongo.Bidding.Password, c.Mongo.Bidding.Size)
  32. entity.InitQyxyMgo(c.Mongo.Qyxy.MongodbAddr, c.Mongo.Qyxy.DbName, c.Mongo.Qyxy.UserName, c.Mongo.Qyxy.Password, c.Mongo.Qyxy.Size)
  33. entity.InitEs(c.Es.Version, c.Es.Address, c.Es.UserName, c.Es.Password, c.Es.DbSize)
  34. entity.InitEntEs(c.EntEs.Version, c.EntEs.Address, c.EntEs.UserName, c.EntEs.Password, c.EntEs.DbSize)
  35. entity.InitArea()
  36. entity.InitRedis(c.RedisAddress)
  37. entity.InitConfig(c.ExportDirectory, c.ComFileDir, c.UpdateProjectUrl, c.ExportUrl, c.ComFileUrl, c.ExportCount)
  38. //合力亿捷
  39. entity.GetHlyj(c.Hlyj.Appid, c.Hlyj.Account, c.Hlyj.Secret, c.Hlyj.TokenUrl, c.Hlyj.CallUrl, c.Hlyj.Integratedid, c.Hlyj.CallFlag)
  40. entity.InitMiddleground(c.RpcServerConf.Etcd.Hosts, c.ResourceCenterKey)
  41. entity.FileCenterRpc = filecenter.NewFileCenter(zrpc.MustNewClient(c.FileCenterRpc))
  42. entity.OssBucketName = c.OssBucketName
  43. entity.OssUrl = c.OssUrl
  44. //nsq
  45. config := nsq.NewConfig()
  46. consumer, err := nsq.NewConsumer(c.TopicName, "jy_position_sync", config)
  47. if err != nil {
  48. fmt.Println(err)
  49. }
  50. consumer.AddHandler(&entity.Handler{})
  51. err = consumer.ConnectToNSQLookupd(c.NsqUrl)
  52. if err != nil {
  53. fmt.Println(err)
  54. }
  55. s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
  56. pb.RegisterBiServiceServer(grpcServer, srv)
  57. })
  58. defer s.Stop()
  59. fmt.Printf("Starting rpc server at %s...\n", c.ListenOn)
  60. s.Start()
  61. }