mirror of
https://github.com/gogf/gf.git
synced 2025-04-05 03:05:05 +08:00
91 lines
2.3 KiB
Go
91 lines
2.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 (
|
|
"fmt"
|
|
|
|
"github.com/gogf/gf/v2/database/gdb"
|
|
"github.com/gogf/gf/v2/errors/gcode"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/text/gstr"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
)
|
|
|
|
// FormatUpsert returns SQL clause of type upsert for PgSQL.
|
|
// For example: ON CONFLICT (id) DO UPDATE SET ...
|
|
func (d *Driver) FormatUpsert(columns []string, list gdb.List, option gdb.DoInsertOption) (string, error) {
|
|
if len(option.OnConflict) == 0 {
|
|
return "", gerror.NewCode(
|
|
gcode.CodeMissingParameter, `Please specify conflict columns`,
|
|
)
|
|
}
|
|
|
|
var onDuplicateStr string
|
|
if option.OnDuplicateStr != "" {
|
|
onDuplicateStr = option.OnDuplicateStr
|
|
} else if len(option.OnDuplicateMap) > 0 {
|
|
for k, v := range option.OnDuplicateMap {
|
|
if len(onDuplicateStr) > 0 {
|
|
onDuplicateStr += ","
|
|
}
|
|
switch v.(type) {
|
|
case gdb.Raw, *gdb.Raw:
|
|
onDuplicateStr += fmt.Sprintf(
|
|
"%s=%s",
|
|
d.Core.QuoteWord(k),
|
|
v,
|
|
)
|
|
case gdb.Counter, *gdb.Counter:
|
|
var counter gdb.Counter
|
|
switch value := v.(type) {
|
|
case gdb.Counter:
|
|
counter = value
|
|
case *gdb.Counter:
|
|
counter = *value
|
|
}
|
|
operator, columnVal := "+", counter.Value
|
|
if columnVal < 0 {
|
|
operator, columnVal = "-", -columnVal
|
|
}
|
|
onDuplicateStr += fmt.Sprintf(
|
|
"%s=EXCLUDED.%s%s%s",
|
|
d.QuoteWord(k),
|
|
d.QuoteWord(counter.Field),
|
|
operator,
|
|
gconv.String(columnVal),
|
|
)
|
|
default:
|
|
onDuplicateStr += fmt.Sprintf(
|
|
"%s=EXCLUDED.%s",
|
|
d.Core.QuoteWord(k),
|
|
d.Core.QuoteWord(gconv.String(v)),
|
|
)
|
|
}
|
|
}
|
|
} else {
|
|
for _, column := range columns {
|
|
// If it's SAVE operation, do not automatically update the creating time.
|
|
if d.Core.IsSoftCreatedFieldName(column) {
|
|
continue
|
|
}
|
|
if len(onDuplicateStr) > 0 {
|
|
onDuplicateStr += ","
|
|
}
|
|
onDuplicateStr += fmt.Sprintf(
|
|
"%s=EXCLUDED.%s",
|
|
d.Core.QuoteWord(column),
|
|
d.Core.QuoteWord(column),
|
|
)
|
|
}
|
|
}
|
|
|
|
conflictKeys := gstr.Join(option.OnConflict, ",")
|
|
|
|
return fmt.Sprintf("ON CONFLICT (%s) DO UPDATE SET ", conflictKeys) + onDuplicateStr, nil
|
|
}
|