Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-15 13:57:10 +08:00
commit 8679200f41
1897 changed files with 257900 additions and 0 deletions
@@ -0,0 +1,77 @@
+package main
+
+import (
+	"sync"
+
+	"91porn-server/common/conf"
+	"91porn-server/common/db"
+	"91porn-server/common/log"
+	"91porn-server/models"
+	"91porn-server/models/Init"
+	"91porn-server/skd/skdg"
+)
+
+/*
+  本函数会将mongo库内的video_info数据, audiobook数据, fiction数据写入es
+  一般在项目部署, 数据导入完成以后运行一次便可
+*/
+
+func init() {
+	cfg := skdg.GlobalConfig{}
+	log.Info("load config start...")
+	if err := conf.LoadJSON("config/skd.json", &cfg); err != nil {
+		log.Error("startUp get config error: %v", log.E(err))
+		panic(err)
+	}
+	skdg.Conf = &cfg
+	log.Info("load config end..")
+	log.Init(log.Options{Level: skdg.Conf.Log.Level})
+	log.Info("daichong conf Domain", log.Any("donmain", skdg.Conf.DaiChong.Domain))
+}
+
+func main() {
+	ds := db.InitDS(initMongodbParam(models.RegisterPool))
+	if len(ds) > 0 {
+		for k, v := range ds {
+			switch k {
+			case models.VideoDb:
+				skdg.VideoDB = v
+			case models.LogDb:
+				skdg.LogDB = v
+			case models.StatDb:
+				skdg.StatDB = v
+			}
+		}
+	}
+	skdg.InitElastic()
+	Init.InitMongo()
+	log.Info("所有同步开始")
+	wg := new(sync.WaitGroup)
+	wg.Add(3)
+	go syncAudioBook(wg)
+	go syncVideoInfo(wg)
+	go syncFiction(wg)
+	wg.Wait()
+	log.Info("所有同步结束")
+}
+
+func initMongodbParam(registerPool []models.Register) ([]db.DBmap, []db.Register) {
+	pri := db.DBmap{
+		Key: models.VideoDb,
+		URL: skdg.Conf.Mongo.VideoDbUrl,
+	}
+	sta := db.DBmap{
+		Key: models.StatDb,
+		URL: skdg.Conf.Mongo.StatDbUrl,
+	}
+	log := db.DBmap{
+		Key: models.LogDb,
+		URL: skdg.Conf.Mongo.LogDbUrl,
+	}
+	dbgister := make([]db.Register, len(registerPool))
+	for i, v := range registerPool {
+		dbgister[i] = db.Register{Key: v.Key, Table: v.Table}
+	}
+	dbmaps := []db.DBmap{pri, sta, log}
+	return dbmaps, dbgister
+}