110 lines
4.4 KiB
Go
110 lines
4.4 KiB
Go
//go:build ignore
|
|
|
|
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
)
|
|
|
|
func main() {
|
|
dsn := strings.TrimPrefix(os.Getenv("IM_MYSQL_DSN"), "mysql://")
|
|
if dsn == "" {
|
|
fatal("IM_MYSQL_DSN is required")
|
|
}
|
|
db, err := sql.Open("mysql", dsn)
|
|
if err != nil {
|
|
fatal(err)
|
|
}
|
|
defer db.Close()
|
|
if err = db.Ping(); err != nil {
|
|
fatal(err)
|
|
}
|
|
|
|
statements := []string{
|
|
"CREATE DATABASE IF NOT EXISTS hotgo CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci",
|
|
`CREATE TABLE IF NOT EXISTS hotgo.im_tenant(
|
|
id BIGINT NOT NULL AUTO_INCREMENT,
|
|
code VARCHAR(64) NOT NULL,
|
|
name VARCHAR(128) NOT NULL,
|
|
tenant_desc VARCHAR(256),
|
|
db_name VARCHAR(128) NOT NULL DEFAULT '',
|
|
db_dsn VARCHAR(1024) NOT NULL DEFAULT '',
|
|
state SMALLINT NOT NULL,
|
|
created_at DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
|
|
created_by BIGINT NOT NULL,
|
|
updated_at DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) ON UPDATE CURRENT_TIMESTAMP(3),
|
|
updated_by BIGINT NOT NULL,
|
|
PRIMARY KEY(id),
|
|
UNIQUE INDEX uk_im_tenant_code(code)
|
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci`,
|
|
`CREATE TABLE IF NOT EXISTS hotgo.im_tenant_config(
|
|
id BIGINT NOT NULL AUTO_INCREMENT,
|
|
tenant_id BIGINT NOT NULL,
|
|
max_users BIGINT UNSIGNED NOT NULL DEFAULT 0 COMMENT '0 means unlimited',
|
|
max_groups BIGINT UNSIGNED NOT NULL DEFAULT 0 COMMENT '0 means unlimited',
|
|
max_group_members INT UNSIGNED NOT NULL DEFAULT 0 COMMENT '0 means unlimited',
|
|
created_at DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
|
|
created_by BIGINT NOT NULL,
|
|
updated_at DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3) ON UPDATE CURRENT_TIMESTAMP(3),
|
|
updated_by BIGINT NOT NULL,
|
|
PRIMARY KEY(id),
|
|
UNIQUE INDEX uk_im_tenant_config_tenant_id(tenant_id),
|
|
CONSTRAINT fk_im_tenant_config_tenant FOREIGN KEY(tenant_id) REFERENCES hotgo.im_tenant(id) ON DELETE CASCADE
|
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci`,
|
|
"CREATE DATABASE IF NOT EXISTS tinode CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci",
|
|
}
|
|
for _, stmt := range statements {
|
|
if _, err = db.Exec(stmt); err != nil {
|
|
fatal(fmt.Errorf("exec failed: %s: %w", stmt, err))
|
|
}
|
|
}
|
|
for _, stmt := range []string{
|
|
"ALTER TABLE hotgo.im_tenant ADD COLUMN tenant_desc VARCHAR(256) AFTER name",
|
|
"ALTER TABLE hotgo.im_tenant ADD COLUMN db_name VARCHAR(128) NOT NULL DEFAULT '' AFTER tenant_desc",
|
|
"ALTER TABLE hotgo.im_tenant ADD COLUMN db_dsn VARCHAR(1024) NOT NULL DEFAULT '' AFTER db_name",
|
|
"ALTER TABLE tinode.im_tenant ADD COLUMN tenant_desc VARCHAR(256) AFTER name",
|
|
"ALTER TABLE tinode.im_tenant ADD COLUMN db_name VARCHAR(128) NOT NULL DEFAULT '' AFTER tenant_desc",
|
|
"ALTER TABLE tinode.im_tenant ADD COLUMN db_dsn VARCHAR(1024) NOT NULL DEFAULT '' AFTER db_name",
|
|
} {
|
|
if _, err = db.Exec(stmt); err != nil && !strings.Contains(err.Error(), "Duplicate column name") {
|
|
fatal(fmt.Errorf("exec failed: %s: %w", stmt, err))
|
|
}
|
|
}
|
|
if _, err = db.Exec(`INSERT INTO hotgo.im_tenant(code,name,tenant_desc,db_name,db_dsn,state,created_by,updated_by)
|
|
VALUES('default','Default Tenant',NULL,'tinode','',1,0,0)
|
|
ON DUPLICATE KEY UPDATE name=VALUES(name), db_name=VALUES(db_name), updated_at=CURRENT_TIMESTAMP(3)`); err != nil {
|
|
fatal(err)
|
|
}
|
|
if _, err = db.Exec(`INSERT INTO hotgo.im_tenant_config(tenant_id,max_users,max_groups,max_group_members,created_by,updated_by)
|
|
SELECT id,0,0,0,0,0 FROM hotgo.im_tenant WHERE code='default'
|
|
ON DUPLICATE KEY UPDATE updated_at=CURRENT_TIMESTAMP(3)`); err != nil {
|
|
fatal(err)
|
|
}
|
|
if _, err = db.Exec("UPDATE tinode.im_tenant SET db_name='tinode' WHERE code='default' AND (db_name='' OR db_name IS NULL)"); err != nil {
|
|
fatal(err)
|
|
}
|
|
|
|
var code, dbName string
|
|
var state int
|
|
if err = db.QueryRow("SELECT code, db_name, state FROM hotgo.im_tenant WHERE code='default'").Scan(&code, &dbName, &state); err != nil {
|
|
fatal(err)
|
|
}
|
|
fmt.Printf("hotgo.im_tenant default: code=%s db_name=%s state=%d\n", code, dbName, state)
|
|
var columnCount int
|
|
if err = db.QueryRow(`SELECT COUNT(*) FROM information_schema.COLUMNS
|
|
WHERE TABLE_SCHEMA='tinode' AND TABLE_NAME='im_tenant' AND COLUMN_NAME IN ('tenant_desc','db_name','db_dsn')`).Scan(&columnCount); err != nil {
|
|
fatal(err)
|
|
}
|
|
fmt.Printf("tinode.im_tenant routing columns: %d/3\n", columnCount)
|
|
}
|
|
|
|
func fatal(v any) {
|
|
fmt.Fprintln(os.Stderr, v)
|
|
os.Exit(1)
|
|
}
|