109 lines
2.7 KiB
Go
109 lines
2.7 KiB
Go
package models
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"go-admin/common/global"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func InitDb(db *gorm.DB) (err error) {
|
|
filePath := "config/db.sql"
|
|
if global.Driver == "postgres" {
|
|
filePath := "config/db.sql"
|
|
if err = ExecSql(db, filePath); err != nil {
|
|
return err
|
|
}
|
|
filePath = "config/pg.sql"
|
|
err = ExecSql(db, filePath)
|
|
} else if global.Driver == "mysql" {
|
|
filePath = "config/db-begin-mysql.sql"
|
|
if err = ExecSql(db, filePath); err != nil {
|
|
return err
|
|
}
|
|
filePath = "config/db.sql"
|
|
if err = ExecSql(db, filePath); err != nil {
|
|
return err
|
|
}
|
|
filePath = "config/db-end-mysql.sql"
|
|
err = ExecSql(db, filePath)
|
|
} else {
|
|
err = ExecSql(db, filePath)
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return bootstrapAdmin(db, os.Getenv)
|
|
}
|
|
|
|
func bootstrapAdmin(db *gorm.DB, getenv func(string) string) error {
|
|
username := strings.TrimSpace(getenv("BELL_BOOTSTRAP_USERNAME"))
|
|
password := getenv("BELL_BOOTSTRAP_PASSWORD")
|
|
if username == "" || password == "" {
|
|
return errors.New("BELL_BOOTSTRAP_USERNAME and BELL_BOOTSTRAP_PASSWORD are required for the initial migration")
|
|
}
|
|
if len(password) < 8 {
|
|
return errors.New("BELL_BOOTSTRAP_PASSWORD must contain at least 8 characters")
|
|
}
|
|
var count int64
|
|
if err := db.Model(&SysUser{}).Where("username = ?", username).Count(&count).Error; err != nil {
|
|
return err
|
|
}
|
|
if count != 0 {
|
|
return fmt.Errorf("bootstrap user %q already exists", username)
|
|
}
|
|
return db.Create(&SysUser{
|
|
Username: username,
|
|
Password: password,
|
|
NickName: "Bell 管理员",
|
|
RoleId: 1,
|
|
DeptId: 1,
|
|
PostId: 1,
|
|
Status: "2",
|
|
ControlBy: ControlBy{
|
|
CreateBy: 1,
|
|
UpdateBy: 1,
|
|
},
|
|
}).Error
|
|
}
|
|
|
|
func ExecSql(db *gorm.DB, filePath string) error {
|
|
sql, err := Ioutil(filePath)
|
|
if err != nil {
|
|
fmt.Println("数据库基础数据初始化脚本读取失败！原因:", err.Error())
|
|
return err
|
|
}
|
|
sqlList := strings.Split(sql, ";")
|
|
for i := 0; i < len(sqlList)-1; i++ {
|
|
if strings.Contains(sqlList[i], "--") {
|
|
fmt.Println(sqlList[i])
|
|
continue
|
|
}
|
|
sql := strings.Replace(sqlList[i]+";", "\n", "", -1)
|
|
sql = strings.TrimSpace(sql)
|
|
if err = db.Exec(sql).Error; err != nil {
|
|
log.Printf("error sql: %s", sql)
|
|
if !strings.Contains(err.Error(), "Query was empty") {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func Ioutil(filePath string) (string, error) {
|
|
if contents, err := ioutil.ReadFile(filePath); err == nil {
|
|
//因为contents是[]byte类型，直接转换成string类型后会多一行空格,需要使用strings.Replace替换换行符
|
|
result := strings.Replace(string(contents), "\n", "", 1)
|
|
fmt.Println("Use ioutil.ReadFile to read a file:", result)
|
|
return result, nil
|
|
} else {
|
|
return "", err
|
|
}
|
|
}
|