fix: scripts folder
This commit is contained in:
parent
55d09e27c1
commit
b391bde9b6
3 changed files with 1 additions and 242 deletions
|
@ -1,192 +0,0 @@
|
||||||
package dbmigrate
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"database/sql"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"git.wh64.net/muffin/goMuffin/configs"
|
|
||||||
|
|
||||||
_ "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"
|
|
||||||
)
|
|
||||||
|
|
||||||
var wg sync.WaitGroup
|
|
||||||
|
|
||||||
// 이 스크립트는 MariaDB -> MongoDB로의 전환을 위해 만들었음.
|
|
||||||
func DBMigrate() {
|
|
||||||
mariaURL := os.Getenv("PREVIOUS_DATABASE_URL")
|
|
||||||
mongoURL := configs.Config.DatabaseURL
|
|
||||||
dbName := configs.Config.DBName
|
|
||||||
|
|
||||||
dbConnectionQuery := "?parseTime=true"
|
|
||||||
|
|
||||||
wg.Add(3)
|
|
||||||
|
|
||||||
// statement -> text
|
|
||||||
go func() {
|
|
||||||
defer wg.Done()
|
|
||||||
|
|
||||||
newDataList := []any{}
|
|
||||||
mariaDB, err := sql.Open("mysql", mariaURL+dbConnectionQuery)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
mongoDB, err := mongo.Connect(options.Client().ApplyURI(mongoURL))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer mongoDB.Disconnect(context.TODO())
|
|
||||||
defer mariaDB.Close()
|
|
||||||
rows, err := mariaDB.Query("select text, persona, created_at from statement;")
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
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 {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if text == "" {
|
|
||||||
text = "살ㄹ려주세요"
|
|
||||||
}
|
|
||||||
|
|
||||||
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()
|
|
||||||
|
|
||||||
newDataList := []any{}
|
|
||||||
mariaDB, err := sql.Open("mysql", mariaURL+dbConnectionQuery)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
mongoDB, err := mongo.Connect(options.Client().ApplyURI(mongoURL))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer mongoDB.Disconnect(context.TODO())
|
|
||||||
defer mariaDB.Close()
|
|
||||||
rows, err := mariaDB.Query("select text, persona, created_at from nsfw_content;")
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
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 {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if text == "" {
|
|
||||||
text = "살ㄹ려주세요"
|
|
||||||
}
|
|
||||||
|
|
||||||
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()
|
|
||||||
|
|
||||||
newDataList := []any{}
|
|
||||||
mariaDB, err := sql.Open("mysql", mariaURL+dbConnectionQuery)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
mongoDB, err := mongo.Connect(options.Client().ApplyURI(mongoURL))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer mongoDB.Disconnect(context.TODO())
|
|
||||||
defer mariaDB.Close()
|
|
||||||
rows, err := mariaDB.Query("select command, result, user_id, created_at from learn;")
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
|
|
||||||
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("데이터 마이그레이션이 끝났어요.")
|
|
||||||
}
|
|
|
@ -25,7 +25,7 @@ func DeleteAllCommands() {
|
||||||
}
|
}
|
||||||
|
|
||||||
if *id == "" {
|
if *id == "" {
|
||||||
panic(fmt.Errorf("--id 플래그의 값이 필요해요."))
|
panic(fmt.Errorf("--id 플래그의 값이 필요해요"))
|
||||||
}
|
}
|
||||||
|
|
||||||
c := http.Client{}
|
c := http.Client{}
|
||||||
|
|
|
@ -1,49 +0,0 @@
|
||||||
package deleteallcommands
|
|
||||||
|
|
||||||
import (
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"net/http"
|
|
||||||
"os"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"git.wh64.net/muffin/goMuffin/configs"
|
|
||||||
"github.com/bwmarrin/discordgo"
|
|
||||||
)
|
|
||||||
|
|
||||||
func DeleteAllCommands() {
|
|
||||||
var answer string
|
|
||||||
id := flag.String("id", "", "디스코드 봇의 토큰")
|
|
||||||
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
fmt.Printf("정말로 모든 명령어를 삭제하시겠어요? [y/N]: ")
|
|
||||||
fmt.Scanf("%s", &answer)
|
|
||||||
if strings.ToLower(answer) != "y" && strings.ToLower(answer) != "yes" {
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
if *id == "" {
|
|
||||||
panic(fmt.Errorf("--id 플래그의 값이 필요해요."))
|
|
||||||
}
|
|
||||||
|
|
||||||
c := http.Client{}
|
|
||||||
req, err := http.NewRequest("PUT", discordgo.EndpointApplicationGlobalCommands(*id), nil)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
req.Header.Add("Authorization", "Bot "+configs.Config.Bot.Token)
|
|
||||||
|
|
||||||
resp, err := c.Do(req)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
bytes, err := io.ReadAll(resp.Body)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
fmt.Println(string(bytes))
|
|
||||||
}
|
|
Loading…
Reference in a new issue