fix(db): update Returning method to accept multiple columns

This commit is contained in:
Hein
2026-05-27 14:11:20 +02:00
parent 4018af0636
commit 69cc3e2839
2 changed files with 4 additions and 4 deletions
+2 -2
View File
@@ -1489,7 +1489,7 @@ func (b *BunInsertQuery) OnConflict(action string) common.InsertQuery {
func (b *BunInsertQuery) Returning(columns ...string) common.InsertQuery { func (b *BunInsertQuery) Returning(columns ...string) common.InsertQuery {
if len(columns) > 0 { if len(columns) > 0 {
b.query = b.query.Returning(columns[0]) b.query = b.query.Returning(strings.Join(columns, ", "))
} }
return b return b
} }
@@ -1606,7 +1606,7 @@ func (b *BunUpdateQuery) Where(query string, args ...interface{}) common.UpdateQ
func (b *BunUpdateQuery) Returning(columns ...string) common.UpdateQuery { func (b *BunUpdateQuery) Returning(columns ...string) common.UpdateQuery {
if len(columns) > 0 { if len(columns) > 0 {
b.query = b.query.Returning(columns[0]) b.query = b.query.Returning(strings.Join(columns, ", "))
} }
return b return b
} }
+2 -2
View File
@@ -1218,8 +1218,8 @@ func (h *Handler) handleCreate(ctx context.Context, w common.ResponseWriter, dat
if provider, ok := modelValue.(common.TableNameProvider); !ok || provider.TableName() == "" { if provider, ok := modelValue.(common.TableNameProvider); !ok || provider.TableName() == "" {
query = query.Table(tableName) query = query.Table(tableName)
} }
fields := reflection.GetSQLModelColumns(model)
query = query.Returning("*") query = query.Returning(fields...)
// Execute BeforeScan hooks - pass query chain so hooks can modify it // Execute BeforeScan hooks - pass query chain so hooks can modify it
itemHookCtx := &HookContext{ itemHookCtx := &HookContext{