feat: 重建 Bell GoAdmin 产品骨架 (#62)

2026-08-27 18:20:40 +08:00
parent b37c350096
commit f5a3dd796b
715 changed files with 75278 additions and 0 deletions
@@ -0,0 +1,108 @@
+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
+	}
+}