WIP: convert GORM to sqlc, for jobs/tasks #104304
@ -572,11 +572,18 @@ func (db *DB) SaveTask(ctx context.Context, t *Task) error {
|
||||
}
|
||||
|
||||
func (db *DB) SaveTaskStatus(ctx context.Context, t *Task) error {
|
||||
tx := db.gormDB.WithContext(ctx).
|
||||
Select("Status").
|
||||
Save(t)
|
||||
if tx.Error != nil {
|
||||
return taskError(tx.Error, "saving task")
|
||||
queries, err := db.queries()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = queries.UpdateTaskStatus(ctx, sqlc.UpdateTaskStatusParams{
|
||||
UpdatedAt: db.now(),
|
||||
Status: string(t.Status),
|
||||
ID: int64(t.ID),
|
||||
})
|
||||
if err != nil {
|
||||
return taskError(err, "saving task status")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
@ -88,3 +88,9 @@ UPDATE tasks SET
|
||||
commands = @commands,
|
||||
activity = @activity
|
||||
WHERE id=@id;
|
||||
|
||||
-- name: UpdateTaskStatus :exec
|
||||
UPDATE tasks SET
|
||||
updated_at = @updated_at,
|
||||
status = @status
|
||||
WHERE id=@id;
|
||||
|
@ -423,3 +423,21 @@ func (q *Queries) UpdateTask(ctx context.Context, arg UpdateTaskParams) error {
|
||||
)
|
||||
return err
|
||||
}
|
||||
|
||||
const updateTaskStatus = `-- name: UpdateTaskStatus :exec
|
||||
UPDATE tasks SET
|
||||
updated_at = ?1,
|
||||
status = ?2
|
||||
WHERE id=?3
|
||||
`
|
||||
|
||||
type UpdateTaskStatusParams struct {
|
||||
UpdatedAt sql.NullTime
|
||||
Status string
|
||||
ID int64
|
||||
}
|
||||
|
||||
func (q *Queries) UpdateTaskStatus(ctx context.Context, arg UpdateTaskStatusParams) error {
|
||||
_, err := q.db.ExecContext(ctx, updateTaskStatus, arg.UpdatedAt, arg.Status, arg.ID)
|
||||
return err
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user