1 Star 0 Fork 0

go-better/go

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
do.go 1.21 KB
一键复制 编辑 原始数据 按行查看 历史
bughou 提交于 2022-03-21 20:37 +08:00 . save
package create
import (
"errors"
"io/ioutil"
"path/filepath"
"strings"
"gitee.com/go-better/dev/config/config"
"gitee.com/go-better/dev/tool/x/release"
)
func Do(env, typ, key string, dropDB bool) error {
if dropDB {
environ := config.NewEnv(env)
switch environ.Minor() {
case `production`, `staging`, `preview`:
return errors.New("drop db is forbidden under environment: " + environ.Minor())
}
}
keys, err := getKeys(key)
if err != nil {
return err
}
for _, key := range keys {
if err := create(env, typ, key, dropDB); err != nil {
return err
}
}
return nil
}
func create(env, typ, key string, dropDB bool) error {
content, err := ioutil.ReadFile(filepath.Join(release.ProjectRoot(), `sqls`, key+`.sql`))
if err != nil {
return err
}
return Creation{env: env, typ: typ, key: key, dropDB: dropDB, sql: string(content)}.do()
}
func getKeys(key string) ([]string, error) {
var keys []string
if key != "" {
keys = []string{key}
} else {
files, err := filepath.Glob(filepath.Join(release.ProjectRoot(), `sqls`, `*.sql`))
if err != nil {
return nil, err
}
for _, file := range files {
keys = append(keys, strings.TrimSuffix(filepath.Base(file), `.sql`))
}
}
return keys, nil
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/go-better/go.git
git@gitee.com:go-better/go.git
go-better
go
go
d31700df43a9

搜索帮助