Skip to content

Commit 207bd69

Browse files
committed
Add missing :exec and :one to delete and update queries
1 parent 8c5de14 commit 207bd69

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

cmd/sqlcup/main.go

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -252,14 +252,20 @@ func writeCreateQuery(w io.Writer, args *scaffoldCommandArgs) {
252252

253253
//goland:noinspection GoUnhandledErrorResult
254254
func writeDeleteQuery(w io.Writer, args *scaffoldCommandArgs) {
255-
fmt.Fprintf(w, "-- name: Delete%s\n", args.SingularEntity)
255+
fmt.Fprintf(w, "-- name: Delete%s :exec\n", args.SingularEntity)
256256
fmt.Fprintf(w, "DELETE FROM %s\n", args.Table)
257257
fmt.Fprintf(w, "WHERE %s = ?;", args.IDColumn.Name)
258258
}
259259

260260
//goland:noinspection GoUnhandledErrorResult
261261
func writeUpdateQuery(w io.Writer, args *scaffoldCommandArgs) {
262-
fmt.Fprintf(w, "-- name: Update%s\n", args.SingularEntity)
262+
var mode string
263+
if args.NoReturningClause {
264+
mode = ":exec"
265+
} else {
266+
mode = ":one"
267+
}
268+
fmt.Fprintf(w, "-- name: Update%s %s\n", args.SingularEntity, mode)
263269
fmt.Fprintf(w, "UPDATE %s\n", args.Table)
264270
fmt.Fprintf(w, "SET\n")
265271
for i, col := range args.NonIDColumns {

0 commit comments

Comments
 (0)