65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package build
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"sort"
|
|
"strings"
|
|
|
|
"git.secnex.io/secnex/pgson/schema"
|
|
"git.secnex.io/secnex/pgson/sql"
|
|
)
|
|
|
|
func Update(s *schema.Table, data map[string]any, where map[string]any) (*string, error) {
|
|
if err := sql.ValidateIdent(s.Name); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(data) == 0 {
|
|
return nil, fmt.Errorf("no columns to update")
|
|
}
|
|
|
|
if len(where) == 0 {
|
|
return nil, fmt.Errorf("WHERE clause is required for UPDATE to prevent accidental updates")
|
|
}
|
|
|
|
var setParts []string
|
|
keys := make([]string, 0, len(data))
|
|
for k := range data {
|
|
keys = append(keys, k)
|
|
}
|
|
sort.Strings(keys)
|
|
|
|
for _, k := range keys {
|
|
if err := sql.ValidateIdent(k); err != nil {
|
|
log.Printf("Invalid column identifier: %s", k)
|
|
return nil, err
|
|
}
|
|
setParts = append(setParts, fmt.Sprintf("%s = %s", sql.QuoteIdent(k), sql.QuoteValue(data[k])))
|
|
}
|
|
|
|
whereClause, err := sql.BuildWhereClause(where)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ddl := fmt.Sprintf(sql.DML_UPDATE, sql.QuoteIdent(s.Name), strings.Join(setParts, ", "), whereClause)
|
|
|
|
return &ddl, nil
|
|
}
|
|
|
|
func UpdateDeletedAt(s *schema.Table, where map[string]any) (*string, error) {
|
|
if err := sql.ValidateIdent(s.Name); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
whereClause, err := sql.BuildWhereClause(where)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ddl := fmt.Sprintf(sql.DML_UPDATE, sql.QuoteIdent(s.Name), "deleted_at = CURRENT_TIMESTAMP", whereClause)
|
|
|
|
return &ddl, nil
|
|
}
|