gochan/gochan
1
0
Fork 0
mirror of https://github.com/Eggbertx/gochan.git synced 2025-08-03 07:36:23 -07:00
Code Releases Activity

Alter table to fix MySQL collation wonkyness

Browse source
This commit is contained in:
Eggbertx 2023-03-27 16:44:59 -07:00
parent d2216d0cdf
commit 353b5d7552
1 changed files with 34 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

37
pkg/gcsql/connect.go
View file

@ -4,6 +4,8 @@ import (
"os" "os"
"regexp" "regexp"
"strings" "strings"
"github.com/gochan-org/gochan/pkg/config"
) )
var ( var (
@ -76,20 +78,49 @@ func tmpSqlAdjust() error {
} }
if numColumns == 0 { if numColumns == 0 {
query = `ALTER TABLE DBPREFIXwordfilters ADD COLUMN board_dirs varchar(255) DEFAULT '*'` query = `ALTER TABLE DBPREFIXwordfilters ADD COLUMN board_dirs varchar(255) DEFAULT '*'`
_, err = ExecSQL(query) if _, err = ExecSQL(query); err != nil {
return err
}
} }
// Yay, collation! Everybody loves MySQL's default collation!
criticalConfig := config.GetSystemCriticalConfig()
query = `ALTER DATABASE ` + criticalConfig.DBname + ` CHARACTER SET = utf8mb4 COLLATE = utf8mb4_unicode_ci`
if _, err = gcdb.db.Exec(query); err != nil {
return err
}
rows, err := QuerySQL(
`SELECT TABLE_NAME FROM information_schema.TABLES WHERE TABLE_SCHEMA = ?`,
criticalConfig.DBname)
if err != nil {
return err
}
defer rows.Close()
for rows.Next() {
var tableName string
err = rows.Scan(&tableName)
if err != nil {
return err
}
query = `ALTER TABLE ` + tableName + ` CONVERT TO CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci`
if _, err = gcdb.db.Exec(query); err != nil {
return err
}
}
err = nil
case "postgres": case "postgres":
_, err = ExecSQL(`ALTER TABLE DBPREFIXwordfilters DROP CONSTRAINT IF EXISTS board_id_fk`) _, err = ExecSQL(`ALTER TABLE DBPREFIXwordfilters DROP CONSTRAINT IF EXISTS board_id_fk`)
if err != nil { if err != nil {
return err return err
} }
_, err = ExecSQL(`ALTER TABLE DBPREFIXwordfilters ADD COLUMN board_dirs varchar(255) DEFAULT '*'`) _, err = ExecSQL(`ALTER TABLE DBPREFIXwordfilters ADD COLUMN IF NOT EXISTS board_dirs varchar(255) DEFAULT '*'`)
case "sqlite3": case "sqlite3":
_, err = ExecSQL(`PRAGMA foreign_keys = ON`) _, err = ExecSQL(`PRAGMA foreign_keys = ON`)
if err != nil { if err != nil {
return err return err
} }
_, err = ExecSQL(`ALTER TABLE DBPREFIXwordfilters ADD COLUMN board_dirs varchar(255) DEFAULT '*'`) _, err = ExecSQL(`ALTER TABLE DBPREFIXwordfilters ADD COLUMN IF NOT EXISTS board_dirs varchar(255) DEFAULT '*'`)
} }
return err return err