mirror of
https://gitee.com/johng/gf.git
synced 2024-11-29 18:57:44 +08:00
47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package pgsql
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/gogf/gf/v2/database/gdb"
|
|
"github.com/gogf/gf/v2/errors/gcode"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
)
|
|
|
|
// 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) {
|
|
switch option.InsertOption {
|
|
case gdb.InsertOptionReplace:
|
|
return nil, gerror.NewCode(
|
|
gcode.CodeNotSupported,
|
|
`Replace operation is not supported by pgsql driver`,
|
|
)
|
|
|
|
case gdb.InsertOptionIgnore:
|
|
return nil, gerror.NewCode(
|
|
gcode.CodeNotSupported,
|
|
`Insert ignore operation is not supported by pgsql driver`,
|
|
)
|
|
|
|
case gdb.InsertOptionDefault:
|
|
tableFields, err := d.GetCore().GetDB().TableFields(ctx, table)
|
|
if err == nil {
|
|
for _, field := range tableFields {
|
|
if field.Key == "pri" {
|
|
pkField := *field
|
|
ctx = context.WithValue(ctx, internalPrimaryKeyInCtx, pkField)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return d.Core.DoInsert(ctx, link, table, list, option)
|
|
}
|