Method ApplyOrderedUpdates

common/storage/sql/postgres.go:74–127

(db *gorm.DB, tableName string, sets []OrderedUpdate, wheres []sql.NamedArg)

}

func (p *postgresHelper) ApplyOrderedUpdates(db *gorm.DB, tableName string, sets []OrderedUpdate, wheres []sql.NamedArg) (int64, error) {

	tx1 := db.Exec("CREATE EXTENSION IF NOT EXISTS pgcrypto;")
	if tx1.Error != nil {
		return 0, tx1.Error
	}

	var namedSets []string
	var namedWheres []string
	var args []interface{}

	for _, w := range wheres {
		namedWheres = append(namedWheres, "@"+w.Name)
		args = append(args, w)
	}

	var assigns []string
	assigns = append(assigns, "uuid")
	var hasName bool
	for _, u := range sets {
		if u.Key == "hash" {
			u.Value = gorm.Expr(p.Hash("uv.new_mpath1", "uv.new_mpath2", "uv.new_mpath3", "uv.new_mpath4"))
			namedSets = append(namedSets, fmt.Sprintf("%s=@%s", u.Key, u.Key))
		} else if u.Key == "hash2" {
			cName := "name"
			if hasName {
				cName = "uv.new_name"
			}
			u.Value = gorm.Expr(p.HashParent(cName, "uv.new_level", "uv.new_mpath1", "uv.new_mpath2", "uv.new_mpath3", "uv.new_mpath4"))
			namedSets = append(namedSets, fmt.Sprintf("%s=@%s", u.Key, u.Key))
		} else {
			if u.Key == "name" {
				hasName = true
			}
			switch u.Value.(type) {
			case int8, int16:
				assigns = append(assigns, fmt.Sprintf("CAST(@%s AS SMALLINT) AS new_%s", u.Key, u.Key))
			case int, int32, int64:
				assigns = append(assigns, fmt.Sprintf("CAST(@%s AS INT) AS new_%s", u.Key, u.Key))
			case bool:
				assigns = append(assigns, fmt.Sprintf("CAST(@%s AS BOOL) AS new_%s", u.Key, u.Key))
			default:
				assigns = append(assigns, fmt.Sprintf("@%s AS new_%s", u.Key, u.Key))
			}
			namedSets = append(namedSets, fmt.Sprintf("\"%s\"=uv.new_%s", u.Key, u.Key))
		}
		args = append(args, sql.Named(u.Key, u.Value))
	}

	q := fmt.Sprintf("WITH updated_values AS ( SELECT %s FROM \"%s\" WHERE %s )", strings.Join(assigns, ", "), tableName, strings.Join(namedWheres, " AND "))
	q += " " + fmt.Sprintf("UPDATE \"%s\" SET %s FROM updated_values AS uv WHERE \"%s\".uuid=uv.uuid", tableName, strings.Join(namedSets, ", "), tableName)
	tx := db.Exec(q, args...)
	return tx.RowsAffected, tx.Error
}

const (
	PostgreDriver = "postgres"