gocron/models/migration.go

package models

import "errors"

// 创建数据库表

type Migration struct{}

func (migration *Migration) Exec(dbName string) error {
    if !isDatabaseExist(dbName) {
        return errors.New("数据库不存在")
    }
    tables := []interface{}{
        &User{}, &Task{}, &TaskLog{}, &Host{},
    }
    for _, table := range tables {
        err := Db.Sync2(table)
        if err != nil {
            return err
        }
    }

    return nil
}

// 创建数据库
func isDatabaseExist(name string) bool {
    _, err := Db.Exec("use ?", name)

    return err != nil
}