代码拉取完成,页面将自动刷新
package orm
import (
"bytes"
"errors"
"reflect"
)
func Create(db dber, v ...interface{}) error {
_, err := NewQuery(db, v...).Create()
return err
}
type insertQuery struct {
*Query
returningFields []*Field
}
var _ QueryAppender = (*insertQuery)(nil)
func (ins insertQuery) AppendQuery(b []byte, params ...interface{}) ([]byte, error) {
if ins.model == nil {
return nil, errors.New("pg: Model(nil)")
}
table := ins.model.Table()
value := ins.model.Value()
b = append(b, "INSERT INTO "...)
if len(ins.onConflict) > 0 {
b = ins.appendTableNameWithAlias(b)
} else {
b = append(b, ins.tableName...)
}
b = append(b, " ("...)
start := len(b)
for _, f := range table.Fields {
b = append(b, f.ColName...)
b = append(b, ", "...)
}
if len(b) > start {
b = b[:len(b)-2]
}
b = append(b, ") VALUES ("...)
if value.Kind() == reflect.Struct {
b = ins.appendValues(b, table.Fields, value)
} else {
for i := 0; i < value.Len(); i++ {
el := value.Index(i)
if el.Kind() == reflect.Interface {
el = el.Elem()
}
b = ins.appendValues(b, table.Fields, reflect.Indirect(el))
if i != value.Len()-1 {
b = append(b, "), ("...)
}
}
}
b = append(b, ')')
if len(ins.onConflict) > 0 {
b = append(b, " ON CONFLICT "...)
b = append(b, ins.onConflict...)
if bytes.HasSuffix(ins.onConflict, []byte("DO UPDATE")) {
var err error
b, err = ins.appendSet(b)
if err != nil {
return nil, err
}
if len(ins.where) > 0 {
b = append(b, " WHERE "...)
b = append(b, ins.where...)
}
}
}
if len(ins.returning) > 0 {
b = append(b, " RETURNING "...)
b = append(b, ins.returning...)
} else if len(ins.returningFields) > 0 {
b = ins.appendReturning(b, ins.returningFields)
}
return b, nil
}
func (ins *insertQuery) appendValues(b []byte, fields []*Field, v reflect.Value) []byte {
for i, f := range fields {
if ins.omitEmpty(f, v) {
b = append(b, "DEFAULT"...)
ins.addReturningField(f)
} else {
b = f.AppendValue(b, v, 1)
}
if i != len(fields)-1 {
b = append(b, ", "...)
}
}
return b
}
func (insertQuery) omitEmpty(f *Field, v reflect.Value) bool {
omit := f.Has(PrimaryKeyFlag)
if !omit && v.Kind() == reflect.Struct {
omit = f.OmitEmpty(v)
}
if !omit {
return false
}
return f.IsEmpty(v)
}
func (ins *insertQuery) addReturningField(field *Field) {
for _, f := range ins.returningFields {
if f == field {
return
}
}
ins.returningFields = append(ins.returningFields, field)
}
func (insertQuery) appendReturning(b []byte, fields []*Field) []byte {
b = append(b, " RETURNING "...)
for i, f := range fields {
b = append(b, f.ColName...)
if i != len(fields)-1 {
b = append(b, ", "...)
}
}
return b
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。