migrate from github

2026-07-15 22:42:31 +08:00
commit 3ad559f05e
262 changed files with 21637 additions and 0 deletions
@@ -0,0 +1,46 @@
+package main
+
+import (
+	"net/http"
+	"sync"
+	"time"
+
+	"github.com/XingfenD/yoresee_doc/internal/bootstrap"
+	"github.com/XingfenD/yoresee_doc/internal/service/mq_service"
+	"github.com/XingfenD/yoresee_doc/internal/utils"
+	"github.com/XingfenD/yoresee_doc/pkg/constant"
+	"github.com/XingfenD/yoresee_doc/pkg/key"
+	"github.com/XingfenD/yoresee_doc/pkg/mq"
+	"github.com/sirupsen/logrus"
+)
+
+func main() {
+	initializer := bootstrap.NewInitializer().
+		InitConfig().
+		InitPostgres().
+		InitRedis().
+		InitElasticsearchAllowFail().
+		InitMQ().
+		InitRepository().
+		InitService()
+	if err := initializer.Err(); err != nil {
+		logrus.Fatalf("Init snapshot-worker failed: %v", err)
+	}
+
+	w := &Worker{
+		collabCoreHTTP: utils.GetEnvVar("COLLAB_CORE_HTTP", "http://collab-core:1234"),
+		mqGroup:        utils.GetEnvVar("SNAPSHOT_MQ_GROUP", "snapshot-worker"),
+		topic:          constant.DirtyDocTopicDefault,
+		mqBackend:      mq.BackendRabbitMQ,
+		client:         &http.Client{Timeout: 10 * time.Second},
+		dirtySetKey:    key.KeyCollabDirtyDocSet(),
+		inFlight:       &sync.Map{},
+	}
+
+	logrus.Infof("Snapshot worker started: topic=%s group=%s collabCore=%s", w.topic, w.mqGroup, w.collabCoreHTTP)
+
+	go w.runMQConsumer(mq_service.MQSvc)
+	go w.runScanLoop()
+
+	initializer.ShutdownOnSignal(0)
+}