mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-01-24 11:36:27 +08:00
97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
// Package hggen
|
|
// @Link https://github.com/bufanyun/hotgo
|
|
// @Copyright Copyright (c) 2022 HotGo CLI
|
|
// @Author Ms <133814250@qq.com>
|
|
// @License https://github.com/bufanyun/hotgo/blob/master/LICENSE
|
|
//
|
|
package views
|
|
|
|
import (
|
|
"context"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/text/gstr"
|
|
"hotgo/internal/consts"
|
|
"hotgo/internal/model/input/sysin"
|
|
"io/ioutil"
|
|
"strings"
|
|
)
|
|
|
|
// parseServFunName 解析业务服务名称
|
|
func (l *gCurd) parseServFunName(templateGroup, varName string) string {
|
|
templateGroup = gstr.UcFirst(templateGroup)
|
|
if gstr.HasPrefix(varName, templateGroup) && varName != templateGroup {
|
|
return varName
|
|
}
|
|
|
|
return templateGroup + varName
|
|
}
|
|
|
|
// getPkField 获取主键
|
|
func (l *gCurd) getPkField(in *CurdPreviewInput) *sysin.GenCodesColumnListModel {
|
|
if len(in.masterFields) == 0 {
|
|
panic("getPkField masterFields uninitialized.")
|
|
}
|
|
for _, field := range in.masterFields {
|
|
if field.Index == consts.GenCodesIndexPK {
|
|
return field
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// hasEffectiveJoin 存在有效的关联表
|
|
func hasEffectiveJoins(joins []*CurdOptionsJoin) bool {
|
|
for _, join := range joins {
|
|
if isEffectiveJoin(join) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func isEffectiveJoin(join *CurdOptionsJoin) bool {
|
|
return join.Alias != "" && join.Field != "" && join.LinkTable != "" && join.MasterField != "" && join.DaoName != "" && join.LinkMode > 0
|
|
}
|
|
|
|
// formatComment formats the comment string to fit the golang code without any lines.
|
|
func formatComment(comment string) string {
|
|
comment = gstr.ReplaceByArray(comment, g.SliceStr{
|
|
"\n", " ",
|
|
"\r", " ",
|
|
})
|
|
comment = gstr.Replace(comment, `\n`, " ")
|
|
comment = gstr.Trim(comment)
|
|
return comment
|
|
}
|
|
|
|
// 移除末尾的换行符
|
|
func removeEndWrap(comment string) string {
|
|
if len(comment) > 2 && comment[len(comment)-2:] == " \n" {
|
|
comment = comment[:len(comment)-2]
|
|
}
|
|
return comment
|
|
}
|
|
|
|
// ImportSql 导出sql文件
|
|
func ImportSql(ctx context.Context, path string) error {
|
|
rows, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
sqlArr := strings.Split(string(rows), "\n")
|
|
for _, sql := range sqlArr {
|
|
sql = strings.TrimSpace(sql)
|
|
if sql == "" || strings.HasPrefix(sql, "--") {
|
|
continue
|
|
}
|
|
exec, err := g.DB().Exec(ctx, sql)
|
|
g.Log().Infof(ctx, "views.ImportSql sql:%v, exec:%+v, err:%+v", sql, exec, err)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|