exec.go
2.7 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package starrocks
import (
"bytes"
"fmt"
"github.com/google/uuid"
"gitlab.fjmaimaimai.com/allied-creation/character-library-metadata-bastion/pkg/domain"
"gorm.io/gorm"
"html"
"html/template"
)
func Insert(db *gorm.DB, tableName string, fields []*domain.FieldValue) error {
value := map[string]interface{}{}
for _, f := range fields {
if f.Field.Flag == domain.PKField && f.Value == "" {
//continue
id, _ := uuid.NewUUID()
f.Value = id.String()
}
value[f.Field.SQLName] = f.TypeValue()
}
//tx := db.Table(tableName).Create(value)
//return tx.Error
sql := db.ToSQL(func(tx *gorm.DB) *gorm.DB {
return tx.Table(tableName).Create(value)
})
tx := db.Exec(sql)
return tx.Error
}
func Update(db *gorm.DB, tableName string, fields []*domain.FieldValue) error {
value := map[string]interface{}{}
var pk *domain.FieldValue
for _, f := range fields {
if f.Field.Flag == domain.PKField && f.Value != "" {
pk = f
}
value[f.Field.SQLName] = f.TypeValue()
}
if pk == nil && pk.Value != "" && pk.TypeValue() != nil {
return fmt.Errorf("主键不能为空")
}
//tx := db.Table(tableName).Where("? = ?", pk.Field.SQLName, pk.Value).Updates(value)
//return tx.Error
sql := db.ToSQL(func(tx *gorm.DB) *gorm.DB {
return tx.Table(tableName).Create(value)
})
tx := db.Exec(sql)
return tx.Error
}
func Delete(db *gorm.DB, tableName string, fields []*domain.FieldValue) error {
var pk *domain.FieldValue
for _, f := range fields {
if f.Field.Flag == domain.PKField && f.Value != "" {
pk = f
}
}
if pk == nil {
return fmt.Errorf("主键不能为空")
}
//tx := db.Table(tableName).Delete("? = ?", pk.Field.SQLName, pk.Value)
//return tx.Error
sql := db.ToSQL(func(tx *gorm.DB) *gorm.DB {
return tx.Table(tableName).Delete("? = ?", pk.Field.SQLName, pk.Value)
})
tx := db.Exec(sql)
return tx.Error
}
//DROP VIEW IF EXISTS
func DropView(db *gorm.DB, tableName string) error {
tx := db.Exec("DROP VIEW IF EXISTS " + tableName)
return tx.Error
}
func Exec(db *gorm.DB, sql string) error {
tx := db.Exec(sql)
return tx.Error
}
func CreateViewSql(viewName string, fields []*domain.Field, selectSql string) string {
sql := `
DROP VIEW IF EXISTS {{.ViewName}};
create view {{.ViewName}}(
{{.Field}}
) as(
{{.Sql}}
)
`
tmp := template.New("ViewCreator")
tmp.Parse(sql)
buf := bytes.NewBuffer(nil)
bufField := bytes.NewBuffer(nil)
for i, f := range fields {
bufField.WriteString(fmt.Sprintf(`%s COMMENT "%v"`, f.SQLName, f.Name))
if i != len(fields)-1 {
bufField.WriteString(",\n")
}
}
tmp.Execute(buf, map[string]interface{}{
"ViewName": viewName,
"Field": bufField.String(),
"Sql": selectSql,
})
return html.UnescapeString(buf.String())
}