fix(go.sum): update ResolveSpec dependency to v1.0.87

2026-06-23 13:17:16 +02:00
parent 0227912325
commit 1adf50e3db
2436 changed files with 1078758 additions and 114 deletions
@@ -0,0 +1,234 @@
+package sqlserver
+
+import (
+	"reflect"
+
+	"gorm.io/gorm"
+	"gorm.io/gorm/callbacks"
+	"gorm.io/gorm/clause"
+)
+
+func Create(db *gorm.DB) {
+	if db.Error != nil {
+		return
+	}
+
+	if db.Statement.Schema != nil && !db.Statement.Unscoped {
+		for _, c := range db.Statement.Schema.CreateClauses {
+			db.Statement.AddClause(c)
+		}
+	}
+
+	hasOutput := false
+	if db.Statement.SQL.String() == "" {
+		var (
+			values                  = callbacks.ConvertToCreateValues(db.Statement)
+			c                       = db.Statement.Clauses["ON CONFLICT"]
+			onConflict, hasConflict = c.Expression.(clause.OnConflict)
+		)
+
+		if hasConflict {
+			if len(db.Statement.Schema.PrimaryFields) > 0 {
+				columnsMap := map[string]bool{}
+				for _, column := range values.Columns {
+					columnsMap[column.Name] = true
+				}
+
+				for _, field := range db.Statement.Schema.PrimaryFields {
+					if _, ok := columnsMap[field.DBName]; !ok {
+						hasConflict = false
+					}
+				}
+			} else {
+				hasConflict = false
+			}
+		}
+
+		if hasConflict {
+			hasOutput = MergeCreate(db, onConflict, values)
+		} else {
+			setIdentityInsert := false
+
+			if db.Statement.Schema != nil {
+				if field := db.Statement.Schema.PrioritizedPrimaryField; field != nil && field.AutoIncrement {
+					switch db.Statement.ReflectValue.Kind() {
+					case reflect.Struct:
+						_, isZero := field.ValueOf(db.Statement.Context, db.Statement.ReflectValue)
+						setIdentityInsert = !isZero
+					case reflect.Slice, reflect.Array:
+						for i := 0; i < db.Statement.ReflectValue.Len(); i++ {
+							obj := db.Statement.ReflectValue.Index(i)
+							if reflect.Indirect(obj).Kind() == reflect.Struct {
+								_, isZero := field.ValueOf(db.Statement.Context, db.Statement.ReflectValue.Index(i))
+								setIdentityInsert = !isZero
+							}
+							break
+						}
+					}
+
+					if setIdentityInsert {
+						db.Statement.WriteString("SET IDENTITY_INSERT ")
+						db.Statement.WriteQuoted(db.Statement.Table)
+						db.Statement.WriteString(" ON;")
+					}
+				}
+			}
+
+			db.Statement.AddClauseIfNotExists(clause.Insert{})
+			db.Statement.Build("INSERT")
+			db.Statement.WriteByte(' ')
+
+			db.Statement.AddClause(values)
+			if values, ok := db.Statement.Clauses["VALUES"].Expression.(clause.Values); ok {
+				if len(values.Columns) > 0 {
+					db.Statement.WriteByte('(')
+					for idx, column := range values.Columns {
+						if idx > 0 {
+							db.Statement.WriteByte(',')
+						}
+						db.Statement.WriteQuoted(column)
+					}
+					db.Statement.WriteByte(')')
+
+					hasOutput = outputInserted(db)
+
+					db.Statement.WriteString(" VALUES ")
+
+					for idx, value := range values.Values {
+						if idx > 0 {
+							db.Statement.WriteByte(',')
+						}
+
+						db.Statement.WriteByte('(')
+						db.Statement.AddVar(db.Statement, value...)
+						db.Statement.WriteByte(')')
+					}
+
+					db.Statement.WriteString(";")
+				} else {
+					db.Statement.WriteString("DEFAULT VALUES;")
+				}
+			}
+
+			if setIdentityInsert {
+				db.Statement.WriteString("SET IDENTITY_INSERT ")
+				db.Statement.WriteQuoted(db.Statement.Table)
+				db.Statement.WriteString(" OFF;")
+			}
+		}
+	}
+
+	if !db.DryRun && db.Error == nil {
+		if db.Statement.Schema != nil && hasOutput {
+			rows, err := db.Statement.ConnPool.QueryContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)
+			if db.AddError(err) == nil {
+				defer rows.Close()
+				gorm.Scan(rows, db, gorm.ScanUpdate|gorm.ScanOnConflictDoNothing)
+				if db.Statement.Result != nil {
+					db.Statement.Result.RowsAffected = db.RowsAffected
+				}
+			}
+		} else {
+			result, err := db.Statement.ConnPool.ExecContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)
+			if db.AddError(err) == nil {
+				db.RowsAffected, _ = result.RowsAffected()
+				if db.Statement.Result != nil {
+					db.Statement.Result.Result = result
+					db.Statement.Result.RowsAffected = db.RowsAffected
+				}
+			}
+		}
+	}
+}
+
+func MergeCreate(db *gorm.DB, onConflict clause.OnConflict, values clause.Values) bool {
+	db.Statement.WriteString("MERGE INTO ")
+	db.Statement.WriteQuoted(db.Statement.Table)
+	db.Statement.WriteString(" USING (VALUES")
+	for idx, value := range values.Values {
+		if idx > 0 {
+			db.Statement.WriteByte(',')
+		}
+
+		db.Statement.WriteByte('(')
+		db.Statement.AddVar(db.Statement, value...)
+		db.Statement.WriteByte(')')
+	}
+
+	db.Statement.WriteString(") AS excluded (")
+	for idx, column := range values.Columns {
+		if idx > 0 {
+			db.Statement.WriteByte(',')
+		}
+		db.Statement.WriteQuoted(column.Name)
+	}
+	db.Statement.WriteString(") ON ")
+
+	var where clause.Where
+	for _, field := range db.Statement.Schema.PrimaryFields {
+		where.Exprs = append(where.Exprs, clause.Eq{
+			Column: clause.Column{Table: db.Statement.Table, Name: field.DBName},
+			Value:  clause.Column{Table: "excluded", Name: field.DBName},
+		})
+	}
+	where.Build(db.Statement)
+
+	if len(onConflict.DoUpdates) > 0 {
+		db.Statement.WriteString(" WHEN MATCHED THEN UPDATE SET ")
+		onConflict.DoUpdates.Build(db.Statement)
+	}
+
+	db.Statement.WriteString(" WHEN NOT MATCHED THEN INSERT (")
+
+	written := false
+	for _, column := range values.Columns {
+		if db.Statement.Schema.PrioritizedPrimaryField == nil || !db.Statement.Schema.PrioritizedPrimaryField.AutoIncrement || db.Statement.Schema.PrioritizedPrimaryField.DBName != column.Name {
+			if written {
+				db.Statement.WriteByte(',')
+			}
+			written = true
+			db.Statement.WriteQuoted(column.Name)
+		}
+	}
+
+	db.Statement.WriteString(") VALUES (")
+
+	written = false
+	for _, column := range values.Columns {
+		if db.Statement.Schema.PrioritizedPrimaryField == nil || !db.Statement.Schema.PrioritizedPrimaryField.AutoIncrement || db.Statement.Schema.PrioritizedPrimaryField.DBName != column.Name {
+			if written {
+				db.Statement.WriteByte(',')
+			}
+			written = true
+			db.Statement.WriteQuoted(clause.Column{
+				Table: "excluded",
+				Name:  column.Name,
+			})
+		}
+	}
+
+	db.Statement.WriteString(")")
+	hasOutput := outputInserted(db)
+	db.Statement.WriteString(";")
+	return hasOutput
+}
+
+func outputInserted(db *gorm.DB) (hasOutput bool) {
+	if db.Statement.Schema != nil && len(db.Statement.Schema.FieldsWithDefaultDBValue) > 0 {
+		for _, field := range db.Statement.Schema.FieldsWithDefaultDBValue {
+			if hasOutput {
+				db.Statement.WriteString(",")
+			}
+			if field.Readable {
+				if !hasOutput {
+					db.Statement.WriteString(" OUTPUT INSERTED.")
+					hasOutput = true
+				} else {
+					db.Statement.WriteString(" INSERTED.")
+				}
+				db.Statement.AddVar(db.Statement, clause.Column{Name: field.DBName})
+			}
+		}
+	}
+	return hasOutput
+}