gocron/routers/install/install.go

package install

import (
    "github.com/ouqiang/gocron/models"
    "github.com/ouqiang/gocron/modules/app"
    "github.com/ouqiang/gocron/modules/setting"
    "github.com/ouqiang/gocron/modules/utils"
    "gopkg.in/macaron.v1"
    "strconv"
    "github.com/ouqiang/gocron/modules/logger"
    "github.com/go-macaron/binding"
)

// 系统安装

type InstallForm struct {
    DbType        string `binding:"IN(mysql)"`
    DbHost        string `binding:"Required"`
    DbPort        int    `binding:"Required;Range(1,65535)"`
    DbUsername    string `binding:"Required"`
    DbPassword    string `binding:"Required"`
    DbName        string `binding:"Required"`
    DbTablePrefix string
    AdminUsername string `binding:"Required;MinSize(3)"`
    AdminPassword string `binding:"Required;MinSize(6)"`
    AdminEmail    string `binding:"Email"`
}

func(f InstallForm) Error(ctx *macaron.Context, errs binding.Errors)  {
    logger.Error(errs)
}

func Create(ctx *macaron.Context) {
    if app.Installed {
        ctx.Redirect("/")
    }
    ctx.Data["Title"] = "安装"
    ctx.Data["DisableNav"] = true
    ctx.HTML(200, "install/create")
}

// 安装
func Store(ctx *macaron.Context, form InstallForm) string {
    json := utils.Json{}
    if app.Installed {
        logger.Warn("系统重复安装")
        return json.Failure(utils.ResponseFailure, "系统已安装!")
    }
    err := testDbConnection(form)
    if err != nil {
        logger.Error(err)
        return json.Failure(utils.ResponseFailure, "数据库连接失败")
    }
    // 写入数据库配置
    err = writeConfig(form)
    if err != nil {
        logger.Error(err)
        return json.Failure(utils.ResponseFailure, "数据库配置写入文件失败")
    }

    app.InitDb()
    // 创建数据库表
    migration := new(models.Migration)
    err = migration.Exec(form.DbName)
    if err != nil {
        logger.Error(err)
        return json.Failure(utils.ResponseFailure, "创建数据库表失败")
    }

    // 创建管理员账号
    err = createAdminUser(form)
    if err != nil {
        logger.Error(err)
        return json.Failure(utils.ResponseFailure, "创建管理员账号失败")
    }

    // 创建安装锁
    err = app.CreateInstallLock()
    if err != nil {
        logger.Error(err)
        return json.Failure(utils.ResponseFailure, "创建文件安装锁失败")
    }

    app.Installed = true
    // 初始化定时任务等
    app.InitResource()

    return json.Success("安装成功", nil)
}

// 数据库配置写入文件
func writeConfig(form InstallForm) error {
    dbConfig := map[string]map[string]string{
        "db": map[string]string{
            "engine":   form.DbType,
            "host":     form.DbHost,
            "port":     strconv.Itoa(form.DbPort),
            "user":     form.DbUsername,
            "password": form.DbPassword,
            "database": form.DbName,
            "prefix":   form.DbTablePrefix,
            "charset":  "utf8",
        },
    }

    return setting.Write(dbConfig, app.AppConfig)
}

// 创建管理员账号
func createAdminUser(form InstallForm) error {
    user := new(models.User)
    user.Name = form.AdminUsername
    user.Password = form.AdminPassword
    user.Email = form.AdminEmail
    user.IsAdmin = 1
    _, err := user.Create()

    return err
}

// 测试数据库连接
func testDbConnection(form InstallForm) error {
    var dbConfig map[string]string = make(map[string]string)
    dbConfig["engine"] = form.DbType
    dbConfig["host"] = form.DbHost
    dbConfig["port"] = strconv.Itoa(form.DbPort)
    dbConfig["user"] = form.DbUsername
    dbConfig["password"] = form.DbPassword
    dbConfig["charset"] = "utf8"
    db, err := models.CreateTmpDb(dbConfig)
    if err != nil {
        return err
    }

    defer  db.Close()
    err = db.Ping()

    return err

}