Method DoInsert

contrib/drivers/pgsql/pgsql_do_insert.go:21–84

DoInsert inserts or updates data for given table. The list parameter must contain at least one record, which was previously validated.

(
	ctx context.Context,
	link gdb.Link, table string, list gdb.List, option gdb.DoInsertOption,
)

// DoInsert inserts or updates data for given table.
// 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.InsertOptionSave,
		gdb.InsertOptionReplace:
		// 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 {
			primaryKeys, err := d.Core.GetPrimaryKeys(ctx, table)
			if err != nil {
				return nil, gerror.WrapCode(
					gcode.CodeInternalError,
					err,
					`failed to get primary keys for Save/Replace operation`,
				)
			}
			foundPrimaryKey := false
			for _, primaryKey := range primaryKeys {
				for dataKey := range list[0] {
					if strings.EqualFold(dataKey, primaryKey) {
						foundPrimaryKey = true
						break
					}
				}
				if foundPrimaryKey {
					break
				}
			}
			if !foundPrimaryKey {
				return nil, gerror.NewCodef(
					gcode.CodeMissingParameter,
					`Replace/Save operation requires conflict detection: `+
						`either specify OnConflict() columns or ensure table '%s' has a primary key in the data`,
					table,
				)
			}
			// TODO consider composite primary keys.
			option.OnConflict = primaryKeys
		}
		// Treat Replace as Save operation
		option.InsertOption = gdb.InsertOptionSave

	// pgsql support InsertIgnore natively, so no need to set primary key in context.
	case gdb.InsertOptionIgnore, gdb.InsertOptionDefault:
		// Get table fields to retrieve the primary key TableField object (not just the name)
		// because DoExec needs the `TableField.Type` to determine if LastInsertId is supported.
		tableFields, err := d.GetCore().GetDB().TableFields(ctx, table)
		if err == nil {
			for _, field := range tableFields {
				if strings.EqualFold(field.Key, "pri") {
					pkField := *field
					ctx = context.WithValue(ctx, internalPrimaryKeyInCtx, pkField)
					break
				}
			}