chore: Enhancing migration speed

main.go

@@ -2,9 +2,11 @@ package main
 
 import (
 	"context"
+	"fmt"
 	"log"
 	"os"
 	"os/signal"
+	"strings"
 	"syscall"
 	"time"
 
@@ -13,12 +15,26 @@ import (
 	"git.wh64.net/muffin/goMuffin/configs"
 	"git.wh64.net/muffin/goMuffin/databases"
 	"git.wh64.net/muffin/goMuffin/handler"
+	dbmigrate "git.wh64.net/muffin/goMuffin/scripts/dbMigrate"
+	deleteallcommands "git.wh64.net/muffin/goMuffin/scripts/deleteAllCommands"
 	"github.com/bwmarrin/discordgo"
 )
 
 func main() {
 	config := configs.Config
 
+	if len(os.Args) > 1 {
+		switch strings.ToLower(os.Args[1]) {
+		case "dbmigrate":
+			dbmigrate.DBMigrate()
+		case "deleteallcommands":
+			deleteallcommands.DeleteAllCommands()
+		default:
+			panic(fmt.Errorf("[goMuffin] 명령어 인자에는 dbmigrate나 deleteallcommands만 올 수 있어요."))
+		}
+		return
+	}
+
 	dg, err := discordgo.New("Bot " + config.Bot.Token)
 	if err != nil {
 		log.Println("[goMuffin] 봇의 세션을 만들수가 없어요.")

scripts/dbMigrate/main.go

@@ -1,4 +1,4 @@
-package main
+package dbmigrate
 
 import (
 	"context"
@@ -9,8 +9,9 @@ import (
 	"time"
 
 	"git.wh64.net/muffin/goMuffin/configs"
-	"git.wh64.net/muffin/goMuffin/databases"
+
 	_ "github.com/go-sql-driver/mysql"
+	"go.mongodb.org/mongo-driver/v2/bson"
 	"go.mongodb.org/mongo-driver/v2/mongo"
 	"go.mongodb.org/mongo-driver/v2/mongo/options"
 )
@@ -18,9 +19,10 @@ import (
 var wg sync.WaitGroup
 
 // 이 스크립트는 MariaDB -> MongoDB로의 전환을 위해 만들었음.
-func main() {
+func DBMigrate() {
 	mariaURL := os.Getenv("PREVIOUS_DATABASE_URL")
 	mongoURL := configs.Config.DatabaseURL
+	dbName := configs.Config.DBName
 
 	dbConnectionQuery := "?parseTime=true"
 
@@ -30,8 +32,7 @@ func main() {
 	go func() {
 		defer wg.Done()
 
-		var text, persona string
-		var createdAt time.Time
+		newDataList := []any{}
 		mariaDB, err := sql.Open("mysql", mariaURL+dbConnectionQuery)
 		if err != nil {
 			panic(err)
@@ -54,6 +55,9 @@ func main() {
 		i := 1
 
 		for rows.Next() {
+			var text, persona string
+			var createdAt time.Time
+
 			fmt.Printf("statement %d\n", i)
 			err = rows.Scan(&text, &persona, &createdAt)
 			if err != nil {
@@ -63,21 +67,25 @@ func main() {
 			if text == "" {
 				text = "살ㄹ려주세요"
 			}
-			databases.Texts.InsertOne(context.TODO(), databases.InsertText{
-				Text:      text,
-				Persona:   persona,
-				CreatedAt: createdAt,
+
+			newDataList = append(newDataList, bson.M{
+				"text":       text,
+				"persona":    persona,
+				"created_at": createdAt,
 			})
 			i++
 		}
+		_, err = mongoDB.Database(dbName).Collection("text").InsertMany(context.TODO(), newDataList)
+		if err != nil {
+			panic(err)
+		}
 	}()
 
 	// nsfw_content -> text
 	go func() {
 		defer wg.Done()
 
-		var text, persona string
-		var createdAt time.Time
+		newDataList := []any{}
 		mariaDB, err := sql.Open("mysql", mariaURL+dbConnectionQuery)
 		if err != nil {
 			panic(err)
@@ -100,6 +108,9 @@ func main() {
 		i := 1
 
 		for rows.Next() {
+			var text, persona string
+			var createdAt time.Time
+
 			fmt.Printf("nsfw_content %d\n", i)
 			err = rows.Scan(&text, &persona, &createdAt)
 			if err != nil {
@@ -109,21 +120,26 @@ func main() {
 			if text == "" {
 				text = "살ㄹ려주세요"
 			}
-			databases.Texts.InsertOne(context.TODO(), databases.InsertText{
-				Text:      text,
-				Persona:   persona,
-				CreatedAt: createdAt,
+
+			newDataList = append(newDataList, bson.M{
+				"text":       text,
+				"persona":    persona,
+				"created_at": createdAt,
 			})
 			i++
 		}
+
+		_, err = mongoDB.Database(dbName).Collection("text").InsertMany(context.TODO(), newDataList)
+		if err != nil {
+			panic(err)
+		}
 	}()
 
 	// learn -> learn
 	go func() {
 		defer wg.Done()
 
-		var command, result, userId string
-		var createdAt time.Time
+		newDataList := []any{}
 		mariaDB, err := sql.Open("mysql", mariaURL+dbConnectionQuery)
 		if err != nil {
 			panic(err)
@@ -146,22 +162,31 @@ func main() {
 		i := 1
 
 		for rows.Next() {
+			var command, result, userId string
+			var createdAt time.Time
+
 			fmt.Printf("learn %d\n", i)
 			err = rows.Scan(&command, &result, &userId, &createdAt)
 			if err != nil {
 				panic(err)
 			}
 
-			databases.Learns.InsertOne(context.TODO(), databases.InsertLearn{
-				Command:   command,
-				Result:    result,
-				UserId:    userId,
-				CreatedAt: createdAt,
+			newDataList = append(newDataList, bson.M{
+				"command":    command,
+				"result":     result,
+				"user_id":    userId,
+				"created_at": createdAt,
 			})
 			i++
 		}
+
+		_, err = mongoDB.Database(dbName).Collection("learn").InsertMany(context.TODO(), newDataList)
+		if err != nil {
+			panic(err)
+		}
 	}()
 
 	// 모든 고루틴이 끝날 떄 까지 대기
 	wg.Wait()
+	fmt.Println("데이터 마이그레이션이 끝났어요.")
 }

scripts/deleteAllCommands/main.go

@@ -1,4 +1,4 @@
-package main
+package deleteallcommands
 
 import (
 	"flag"
@@ -8,29 +8,24 @@ import (
 	"os"
 	"strings"
 
+	"git.wh64.net/muffin/goMuffin/configs"
 	"github.com/bwmarrin/discordgo"
-	"github.com/joho/godotenv"
 )
 
-func main() {
-	godotenv.Load()
+func DeleteAllCommands() {
 	var answer string
-	id := flag.String("id", "", "discordBot's token")
+	id := flag.String("id", "", "디스코드 봇의 토큰")
 
 	flag.Parse()
 
-	fmt.Printf("Do you want to delete all commands? [y/N]: ")
+	fmt.Printf("정말로 모든 명령어를 삭제하시겠어요? [y/N]: ")
 	fmt.Scanf("%s", &answer)
-	if strings.ToLower(answer) != "y" {
+	if strings.ToLower(answer) != "y" && strings.ToLower(answer) != "yes" {
 		os.Exit(1)
 	}
 
-	if os.Getenv("BOT_TOKEN") == "" {
-		panic(fmt.Errorf("You need a BOT_TOKEN environment."))
-	}
-
 	if *id == "" {
-		panic(fmt.Errorf("You need a --id flag value."))
+		panic(fmt.Errorf("--id 플래그의 값이 필요해요."))
 	}
 
 	c := http.Client{}
@@ -39,7 +34,7 @@ func main() {
 		panic(err)
 	}
 
-	req.Header.Add("Authorization", "Bot "+os.Getenv("BOT_TOKEN"))
+	req.Header.Add("Authorization", "Bot "+configs.Config.Bot.Token)
 
 	resp, err := c.Do(req)
 	if err != nil {