Method JobCancel

riverdriver/riversqlite/river_sqlite_driver.go:251–287

(ctx context.Context, params *riverdriver.JobCancelParams)

}

func (e *Executor) JobCancel(ctx context.Context, params *riverdriver.JobCancelParams) (*rivertype.JobRow, error) {
	// Unlike Postgres, this must be carried out in two operations because
	// SQLite doesn't support CTEs containing `UPDATE`. As long as the job
	// exists and is not running, only one database operation is needed, but if
	// the initial update comes back empty, it does one more fetch to return the
	// most appropriate error.
	return dbutil.WithTxV(ctx, e, func(ctx context.Context, execTx riverdriver.ExecutorTx) (*rivertype.JobRow, error) { // TODO
		dbtx := templateReplaceWrapper{dbtx: e.driver.UnwrapTx(execTx), replacer: &e.driver.replacer}

		cancelledAt, err := params.CancelAttemptedAt.UTC().MarshalJSON()
		if err != nil {
			return nil, err
		}
		cancelledAt = cancelledAt[1 : len(cancelledAt)-1] // remove quotes around the time so we don't end up with doubled up quotes

		job, err := dbsqlc.New().JobCancel(schemaTemplateParam(ctx, params.Schema), dbtx, &dbsqlc.JobCancelParams{
			ID:                params.ID,
			CancelAttemptedAt: string(cancelledAt),
			Now:               timeStringNullable(params.Now),
		})
		if err != nil {
			if errors.Is(err, sql.ErrNoRows) {
				job, err := execTx.JobGetByID(ctx, &riverdriver.JobGetByIDParams{
					ID:     params.ID,
					Schema: params.Schema,
				})
				if err != nil {
					return nil, interpretError(err)
				}
				return job, nil
			}

			return nil, interpretError(err)
		}
		return jobRowFromInternal(job)
	})
}

func (e *Executor) JobCountByAllStates(ctx context.Context, params *riverdriver.JobCountByAllStatesParams) (map[rivertype.JobState]int, error) {
	counts, err := dbsqlc.New().JobCountByAllStates(schemaTemplateParam(ctx, params.Schema), e.dbtx)