refactor: 代码改进

yangyi · yangyi · commit 17e30e3d8cb5 · 2025-08-03T14:00:00.000+08:00
diff --git a/db.go b/db.go
@@ -186,7 +186,7 @@ func doUpsert(ctx context.Context, ent Entity, db DB) error {
 		return fmt.Errorf("get metadata, %w", err)
 	}
 
-	for _, col := range md.Columns {
+	for _, col := range md.PrimaryKeys {
 		if col.AutoIncrement {
 			return fmt.Errorf("upsert not support auto increment primary key %q", col.DBField)
 		}

Original file line number	Diff line number	Diff line change
`@@ -186,7 +186,7 @@ func doUpsert(ctx context.Context, ent Entity, db DB) error {`
`186`	`186`	`return fmt.Errorf("get metadata, %w", err)`
`187`	`187`	`}`
`188`	`188`
`189`		`- for _, col := range md.Columns {`
	`189`	`+ for _, col := range md.PrimaryKeys {`
`190`	`190`	`if col.AutoIncrement {`
`191`	`191`	`return fmt.Errorf("upsert not support auto increment primary key %q", col.DBField)`
`192`	`192`	`}`