feat: auto start workers

backend/modules/middlewares/m_check_ua.go

@@ -0,0 +1,17 @@
+package middlewares
+
+import (
+	"strings"
+
+	"github.com/gofiber/fiber/v3"
+)
+
+func (m *Middlewares) CheckUA(ctx fiber.Ctx) error {
+	keyword := strings.ToLower("MicroMessenger")
+	userAgent := ctx.GetReqHeaders()["User-Agent"][0]
+
+	if strings.Contains(userAgent, keyword) {
+		return ctx.SendString("")
+	}
+	return ctx.Next()
+}

backend/modules/middlewares/mid_debug.go

@@ -2,7 +2,6 @@ package middlewares
 
 import (
 	"github.com/gofiber/fiber/v3"
-	log "github.com/sirupsen/logrus"
 )
 
 func (f *Middlewares) DebugMode(c fiber.Ctx) error {
@@ -10,6 +9,5 @@ func (f *Middlewares) DebugMode(c fiber.Ctx) error {
 	// host := c.BaseURL()
 	// fmt.Println(strings.Split(c.Path(), "/"))
 	// return c.SendString("ABC" + c.Params("+"))
-	log.SetLevel(log.DebugLevel)
 	return c.Next()
 }

backend/modules/workers/provider.gen.go

@@ -0,0 +1,31 @@
+package workers
+
+import (
+	"backend/modules/commands/store"
+	"backend/providers/storage"
+
+	"git.ipao.vip/rogeecn/atom"
+	"git.ipao.vip/rogeecn/atom/container"
+	"git.ipao.vip/rogeecn/atom/contracts"
+	"git.ipao.vip/rogeecn/atom/utils/opt"
+)
+
+func Provide(opts ...opt.Option) error {
+	if err := container.Container.Provide(func(
+		cmdStore *store.StoreMedias,
+		storage *storage.Config,
+	) (contracts.Initial, error) {
+		obj := &Controller{
+			cmdStore: cmdStore,
+			storage:  storage,
+		}
+		if err := obj.Prepare(); err != nil {
+			return nil, err
+		}
+		return obj, nil
+	}, atom.GroupInitial); err != nil {
+		return err
+	}
+
+	return nil
+}

backend/modules/workers/workers.go

@@ -0,0 +1,39 @@
+package workers
+
+import (
+	"time"
+
+	"backend/modules/commands/store"
+	"backend/providers/storage"
+
+	_ "git.ipao.vip/rogeecn/atom"
+	_ "git.ipao.vip/rogeecn/atom/contracts"
+	"github.com/sirupsen/logrus"
+)
+
+// @provider:except	contracts.Initial atom.GroupInitial
+type Controller struct {
+	log      *logrus.Entry `inject:"false"`
+	cmdStore *store.StoreMedias
+	storage  *storage.Config
+}
+
+func (c *Controller) Prepare() error {
+	c.log = logrus.WithField("module", "workers.worker")
+	c.log.Info("start workers")
+
+	go c.store()
+
+	return nil
+}
+
+func (c *Controller) store() {
+	ticker := time.NewTicker(time.Minute * 5)
+
+	for range ticker.C {
+		c.log.WithField("action", "store").Info("start to run store")
+		if err := c.cmdStore.RunE(c.storage.Path); err != nil {
+			c.log.WithField("action", "store").WithError(err).Error("run store cmd failed")
+		}
+	}
+}

backend/pkg/service/http/http.go

@@ -3,10 +3,12 @@ package http
 import (
 	"path/filepath"
 
+	"backend/modules/commands/store"
 	"backend/modules/medias"
 	"backend/modules/middlewares"
 	"backend/modules/users"
 	wechatModule "backend/modules/wechat"
+	"backend/modules/workers"
 	"backend/providers/app"
 	"backend/providers/hashids"
 	"backend/providers/http"
@@ -49,6 +51,8 @@ func Command() atom.Option {
 			users.Provide,
 			medias.Provide,
 			wechatModule.Provide,
+			store.Provide,
+			workers.Provide,
 		)),
 	)
 }
@@ -70,10 +74,12 @@ func Serve(cmd *cobra.Command, args []string) error {
 		if http.App.Mode == app.AppModeDevelopment {
 			log.SetLevel(log.DebugLevel)
 		}
+
 		engine := http.Service.Engine
 		mid := http.Middlewares
 
 		engine.Use(mid.DebugMode)
+		engine.Use(mid.CheckUA)
 		engine.Use(mid.ProcessResponse)
 		engine.Use(mid.WeChatVerify)