migrate from github

cmd/db_init/other_user.go

@@ -0,0 +1,57 @@
+package main
+
+import (
+	"strconv"
+
+	"github.com/XingfenD/yoresee_doc/internal/model"
+	"github.com/XingfenD/yoresee_doc/internal/utils"
+	"github.com/sirupsen/logrus"
+	"gorm.io/gorm"
+)
+
+func createTestUserIntx(tx *gorm.DB) error {
+	logrus.Println("Creating test user in transaction...")
+	return createUserWithUsername(tx, "test")
+}
+
+func createNormalUserInTx(tx *gorm.DB) error {
+	logrus.Println("Creating 100 users in transaction...")
+
+	for i := 1; i <= 20; i++ {
+		username := "user" + strconv.Itoa(i)
+		if err := createUserWithUsername(tx, username); err != nil {
+			return err
+		}
+
+		var user model.User
+		if err := tx.Where("username = ?", username).First(&user).Error; err != nil {
+			return err
+		}
+		userNum := strconv.Itoa(i)
+		logrus.Printf("User%s created successfully with ID: %s in transaction.\n", userNum, user.ExternalID)
+
+		logrus.Printf("Creating knowledge base for user%s...\n", userNum)
+
+		kbExternalID := utils.GenerateExternalID(utils.ExternalIDKnowledgeBase)
+		knowledgeBase := model.KnowledgeBase{
+			ExternalID:    kbExternalID,
+			Name:          "User" + userNum + "'s Knowledge Base",
+			Description:   "Knowledge base created by user" + userNum,
+			CreatorUserID: user.ID,
+			IsPublic:      false,
+		}
+
+		var kbCount int64
+		tx.Model(&model.KnowledgeBase{}).Where("name = ?", knowledgeBase.Name).Count(&kbCount)
+		if kbCount == 0 {
+			if err := tx.Create(&knowledgeBase).Error; err != nil {
+				return err
+			}
+			logrus.Printf("Knowledge base created successfully for user%s: %s\n", userNum, knowledgeBase.Name)
+		} else {
+			logrus.Printf("Knowledge base for user%s already exists.\n", userNum)
+		}
+	}
+
+	return nil
+}