176 lines
3.2 KiB
Go
176 lines
3.2 KiB
Go
package db
|
|
|
|
import (
|
|
"allapp-go/internal/middleware"
|
|
"context"
|
|
"fmt"
|
|
"sort"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func (c *Client) Update(
|
|
ctx context.Context,
|
|
table string,
|
|
keyField string,
|
|
data map[string]any,
|
|
) error {
|
|
keyVal, err := validateData(table, keyField, data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !auditExcludeTables[table] {
|
|
now := time.Now()
|
|
|
|
delete(data, "update_time")
|
|
delete(data, "update_by")
|
|
|
|
data["update_time"] = now
|
|
|
|
userID, flag := middleware.GetUserID(ctx)
|
|
if flag {
|
|
data["update_by"] = userID
|
|
}
|
|
}
|
|
|
|
// 👉 表名加引号（支持 schema）
|
|
tableParts := strings.Split(table, ".")
|
|
for i, p := range tableParts {
|
|
tableParts[i] = `"` + p + `"`
|
|
}
|
|
table = strings.Join(tableParts, ".")
|
|
|
|
// 👉 keyField 加引号
|
|
keyField = `"` + keyField + `"`
|
|
|
|
var (
|
|
setClauses []string
|
|
args []any
|
|
i = 1
|
|
)
|
|
|
|
for col, val := range data {
|
|
if col == keyField {
|
|
continue
|
|
}
|
|
|
|
// 👉 列名加引号
|
|
colQuoted := `"` + col + `"`
|
|
|
|
setClauses = append(setClauses, fmt.Sprintf("%s=$%d", colQuoted, i))
|
|
args = append(args, val)
|
|
i++
|
|
}
|
|
|
|
where := fmt.Sprintf("%s=$%d", keyField, i)
|
|
args = append(args, keyVal)
|
|
|
|
sql := fmt.Sprintf(
|
|
"UPDATE %s SET %s WHERE %s",
|
|
table,
|
|
strings.Join(setClauses, ", "),
|
|
where,
|
|
)
|
|
|
|
_, err = c.exec().Exec(ctx, sql, args...)
|
|
return err
|
|
}
|
|
|
|
func (c *Client) BatchUpdate(
|
|
ctx context.Context,
|
|
table string,
|
|
keyField string,
|
|
dataList []map[string]any,
|
|
) error {
|
|
|
|
if len(dataList) == 0 {
|
|
return fmt.Errorf("数据不能为空")
|
|
}
|
|
|
|
first := dataList[0]
|
|
if _, err := validateData(table, keyField, first); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 👉 表名加引号（支持 schema）
|
|
tableParts := strings.Split(table, ".")
|
|
for i, p := range tableParts {
|
|
tableParts[i] = `"` + p + `"`
|
|
}
|
|
table = strings.Join(tableParts, ".")
|
|
|
|
// 👉 keyField 加引号
|
|
keyFieldQuoted := `"` + keyField + `"`
|
|
|
|
var columns []string
|
|
for col := range first {
|
|
if col != keyField {
|
|
columns = append(columns, col)
|
|
}
|
|
}
|
|
sort.Strings(columns)
|
|
|
|
var (
|
|
args []any
|
|
argIndex = 1
|
|
)
|
|
|
|
var setClauses []string
|
|
|
|
for _, col := range columns {
|
|
colQuoted := `"` + col + `"`
|
|
|
|
var caseBuilder strings.Builder
|
|
caseBuilder.WriteString(fmt.Sprintf("%s = CASE %s ", colQuoted, keyFieldQuoted))
|
|
|
|
for _, data := range dataList {
|
|
keyVal, err := validateData(table, keyField, data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
val, ok := data[col]
|
|
if !ok {
|
|
return fmt.Errorf("缺少字段: %s", col)
|
|
}
|
|
|
|
caseBuilder.WriteString(fmt.Sprintf(
|
|
"WHEN $%d THEN $%d ",
|
|
argIndex,
|
|
argIndex+1,
|
|
))
|
|
|
|
args = append(args, keyVal, val)
|
|
argIndex += 2
|
|
}
|
|
|
|
caseBuilder.WriteString("END")
|
|
setClauses = append(setClauses, caseBuilder.String())
|
|
}
|
|
|
|
var wherePlaceholders []string
|
|
|
|
for _, data := range dataList {
|
|
keyVal, ok := data[keyField]
|
|
if !ok || keyVal == nil {
|
|
return fmt.Errorf("缺少主键字段值: %s", keyField)
|
|
}
|
|
|
|
wherePlaceholders = append(wherePlaceholders, fmt.Sprintf("$%d", argIndex))
|
|
args = append(args, keyVal)
|
|
argIndex++
|
|
}
|
|
|
|
sql := fmt.Sprintf(
|
|
"UPDATE %s SET %s WHERE %s IN (%s)",
|
|
table,
|
|
strings.Join(setClauses, ", "),
|
|
keyFieldQuoted,
|
|
strings.Join(wherePlaceholders, ", "),
|
|
)
|
|
|
|
_, err := c.exec().Exec(ctx, sql, args...)
|
|
return err
|
|
}
|