bingyi/gf
1
0
Fork 0
mirror of https://gitee.com/johng/gf synced 2026-06-07 02:12:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

up

Browse Source
This commit is contained in:
John Guo
2025-12-08 14:37:35 +08:00
parent 91e3f1eab1
commit 99536c8bef
13 changed files with 137 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
contrib/drivers/pgsql/pgsql_do_insert.go
View File

@ -17,9 +17,15 @@ import (
)
// DoInsert inserts or updates data for given table.
func (d *Driver) DoInsert(ctx context.Context, link gdb.Link, table string, list gdb.List, option gdb.DoInsertOption) (result sql.Result, err error) {
// The list parameter must contain at least one record, which was previously validated.
func (d *Driver) DoInsert(
ctx context.Context,
link gdb.Link, table string, list gdb.List, option gdb.DoInsertOption,
) (result sql.Result, err error) {
switch option.InsertOption {
case gdb.InsertOptionReplace:
case
gdb.InsertOptionReplace,
gdb.InsertOptionSave:
// PostgreSQL does not support REPLACE INTO syntax, use Save (ON CONFLICT ... DO UPDATE) instead.
// Automatically detect primary keys if OnConflict is not specified.
if len(option.OnConflict) == 0 {
@ -31,10 +37,17 @@ func (d *Driver) DoInsert(ctx context.Context, link gdb.Link, table string, list
`failed to get primary keys for Replace operation`,
)
}
if len(primaryKeys) == 0 {
foundPrimaryKey := false
for _, conflictKey := range primaryKeys {
if _, ok := list[0][conflictKey]; ok {
foundPrimaryKey = true
break
}
}
if !foundPrimaryKey {
return nil, gerror.NewCode(
gcode.CodeMissingParameter,
`Replace operation requires primary key, but table has no primary key defined`,
`Please specify conflict columns or ensure the record has a primary key for Save/Replace operation`,
)
}
option.OnConflict = primaryKeys
@ -46,7 +59,7 @@ func (d *Driver) DoInsert(ctx context.Context, link gdb.Link, table string, list
tableFields, err := d.GetCore().GetDB().TableFields(ctx, table)
if err == nil {
for _, field := range tableFields {
if field.Key == "pri" {
if gstr.Equal(field.Key, "PRI") {
pkField := *field
ctx = context.WithValue(ctx, internalPrimaryKeyInCtx, pkField)
break