task.go 813 B

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. package main
  2. import (
  3. "flag"
  4. "fmt"
  5. "github.com/zeromicro/go-zero/core/conf"
  6. "github.com/zeromicro/go-zero/core/logx"
  7. "github.com/zeromicro/go-zero/core/service"
  8. "paopaoimtask/internal/config"
  9. "paopaoimtask/internal/handler"
  10. "paopaoimtask/internal/svc"
  11. "paopaoimtask/pkg/log"
  12. )
  13. var configFile = flag.String("f", "etc/task-mq.yaml", "the config file")
  14. func main() {
  15. flag.Parse()
  16. var c config.Config
  17. conf.MustLoad(*configFile, &c)
  18. if err := c.SetUp(); err != nil {
  19. panic(err)
  20. }
  21. ctx := svc.NewServiceContext(c)
  22. writer := log.NewZapWriter()
  23. logx.SetWriter(writer)
  24. listen := handler.NewListen(ctx)
  25. serviceGroup := service.NewServiceGroup()
  26. for _, s := range listen.Services() {
  27. serviceGroup.Add(s)
  28. }
  29. fmt.Printf("starting task server at %v....\n", c.ListenOn)
  30. serviceGroup.Start()
  31. }