123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384 |
- package main
- import (
- "flag"
- "fmt"
- "github.com/zeromicro/go-zero/core/logx"
- "wechat-api/crontask"
- "wechat-api/internal/config"
- "wechat-api/internal/handler"
- "wechat-api/internal/service/MessageHandlers"
- "wechat-api/internal/svc"
- "github.com/robfig/cron/v3"
- "github.com/zeromicro/go-zero/core/conf"
- "github.com/zeromicro/go-zero/rest"
- )
- var configFile = flag.String("f", "etc/wechat.yaml", "the config file")
- func main() {
- flag.Parse()
- var c config.Config
- conf.MustLoad(*configFile, &c, conf.UseEnv())
- server := rest.MustNewServer(c.RestConf, rest.WithCors(c.CROSConf.Address))
- defer server.Stop()
- ctx := svc.NewServiceContext(c)
- handler.RegisterHandlers(server, ctx)
-
- if len(ctx.WechatWs) == 0 {
- fmt.Println("wechat ws is nil")
- } else if ctx.WechatWs["default"] != nil {
-
- for _, ws := range ctx.WechatWs {
-
-
-
-
-
- ws.RegisterMessageHandler(MessageHandlers.NewFriendPushNoticeHandler(ctx).Handler)
- }
- logx.Info("注册个微处理通道~")
- }
-
-
-
- cronCtx := cron.New(cron.WithChain(
- cron.SkipIfStillRunning(cron.DefaultLogger),
- cron.Recover(cron.DefaultLogger),
- ))
- crontask.ScheduleRun(cronCtx, ctx)
- go cronCtx.Start()
- defer cronCtx.Stop()
- fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port)
- server.Start()
- }
|