threshold/internal/scheduler/job/query.go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: BUSL-1.1

package job

const runJobsQuery = `
	insert into job_run (
	  job_plugin_id, job_name, controller_id
	)
	select
	  j.plugin_id, j."name", ?
	from job j
	where next_scheduled_run <= current_timestamp
	order by next_scheduled_run asc
	on conflict
	  (job_plugin_id, job_name)
	    where status = 'running'
	do nothing
	returning *;
`

const upsertJobQuery = `
	insert into job (
	  plugin_id, 
	  name, 
	  description, 
	  next_scheduled_run 
	) values (
	  @plugin_id, -- plugin_id
	  @name, -- name
	  @description, -- description
	  wt_add_seconds_to_now(@next_scheduled_run) -- next_scheduled_run
	)
	on conflict on constraint  
	  job_pkey
	do update set 
	  description = @description
	returning *;
`

const setNextScheduledRunIfSoonerQuery = `
	update
	  job
	set
	  next_scheduled_run = least(wt_add_seconds_to_now(?), next_scheduled_run)
	where
	  plugin_id = ?
	  and name = ?
	returning *;
`

const setNextScheduledRunQuery = `
	update
	  job
	set
	  next_scheduled_run = wt_add_seconds_to_now(?)
	where
	  plugin_id = ?
	  and name = ?
	returning *;
`

const updateProgressQuery = `
	update
	  job_run
	set
	  completed_count = ?,
	  total_count = ?,
	  retries_count = ?
	where
	  private_id = ?
	  and status = 'running'
	returning *;
`

const completeRunQuery = `
	delete from job_run
	where
	  private_id = ?
	  and status = 'running'
	returning *;
`

const failRunQuery = `
	update
	  job_run
	set
	  completed_count = ?,
	  total_count     = ?,
	  retries_count   = ?,
	  status          = 'failed',
	  end_time        = current_timestamp
	where
	  private_id = ?
	  and status = 'running'
	returning *;
`

const interruptRunsQuery = `
	update
	  job_run
	set 
	  status = 'interrupted',
	  end_time = current_timestamp
	where
	  update_time <= wt_add_seconds_to_now(?)
	  and status = 'running'
      %s
	returning *;
`

const deleteJobByName = `
	delete 
	from job 
	where 
	  name = ?
`