wechat.go 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. // wechat
  2. //
  3. // Description: wechat service
  4. //
  5. // Schemes: http, https
  6. // Host: localhost:0
  7. // BasePath: /
  8. // Version: 0.0.1
  9. // SecurityDefinitions:
  10. // Token:
  11. // type: apiKey
  12. // name: Authorization
  13. // in: header
  14. // Security:
  15. // - Token: []
  16. // Consumes:
  17. // - application/json
  18. //
  19. // Produces:
  20. // - application/json
  21. //
  22. // swagger:meta
  23. package main
  24. import (
  25. "flag"
  26. "fmt"
  27. "github.com/zeromicro/go-zero/core/logx"
  28. "wechat-api/crontask"
  29. "wechat-api/internal/config"
  30. "wechat-api/internal/handler"
  31. "wechat-api/internal/service/MessageHandlers"
  32. "wechat-api/internal/svc"
  33. "github.com/robfig/cron/v3"
  34. "github.com/zeromicro/go-zero/core/conf"
  35. "github.com/zeromicro/go-zero/rest"
  36. )
  37. var configFile = flag.String("f", "etc/wechat.yaml", "the config file")
  38. func main() {
  39. flag.Parse()
  40. var c config.Config
  41. conf.MustLoad(*configFile, &c, conf.UseEnv())
  42. server := rest.MustNewServer(c.RestConf, rest.WithCors(c.CROSConf.Address))
  43. defer server.Stop()
  44. ctx := svc.NewServiceContext(c)
  45. handler.RegisterHandlers(server, ctx)
  46. //个微处理程序 没有彻底完成之前不能开,和cow重复了
  47. if len(ctx.WechatWs) == 0 {
  48. fmt.Println("wechat ws is nil")
  49. } else if ctx.WechatWs["default"] != nil {
  50. //var ic channel.IChannel
  51. for _, ws := range ctx.WechatWs {
  52. //switch ws.CTypes {
  53. //case "wechat":
  54. // ic = channel.NewWechatChannel(ws, ctx)
  55. //}
  56. //ws.RegisterMessageHandler(ic.OnMessage)
  57. ws.RegisterMessageHandler(MessageHandlers.NewFriendPushNoticeHandler(ctx).Handler)
  58. }
  59. logx.Info("注册个微处理通道~")
  60. }
  61. //for {
  62. // time.Sleep(time.Second * 10)
  63. //}
  64. cronCtx := cron.New(cron.WithChain(
  65. cron.SkipIfStillRunning(cron.DefaultLogger),
  66. cron.Recover(cron.DefaultLogger),
  67. ))
  68. crontask.ScheduleRun(cronCtx, ctx)
  69. go cronCtx.Start()
  70. defer cronCtx.Stop()
  71. fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port)
  72. server.Start()
  73. }