db

type DB interface {
	// EnqueueJob enqueues a query to be run in a specific repository at a
	// specific point in the commit history.
	//
	// Input QueryJob must have the Repository, CommitHash, Query fields
	// populated.
	//
	// On exit, QueryJob has the ID and QueueTime fields populated.
	//
	// If there is an existing non-failed job with the same Repository, CommitHash, and
	// Query, enqueue requests should deduplicate to the same request ID; failed
	// jobs are ignored for the purposes of this deduplication.
	EnqueueJob(context.Context, *QueryJob) error

	DequeueJob(ctx context.Context, workerName string) (*QueryJob, error)

	GetJob(ctx context.Context, id string) (*QueryJob, error)

	FinishJob(ctx context.Context, id string, status string, result string) error

	io.Closer
}

type Fake struct {
	Queue []FakeQueueEntry

	EnqueueJobErr error
	GetJobErr     error
	FinishJobErr  error
}

type FakeQueueEntry struct {
	Job *QueryJob
	Err error
}

type QueryJob struct {
	Repository  string     `datastore:"repository"`
	CommitHash  string     `datastore:"commit_hash"`
	Query       string     `datastore:"query_string"`
	ID          string     `datastore:"id"`
	Status      string     `datastore:"status"`
	Worker      *string    `datastore:"worker"`
	QueueTime   time.Time  `datastore:"queue_time"`
	StartTime   *time.Time `datastore:"start_time"`
	FinishTime  *time.Time `datastore:"finish_time"`
	ResultURL   *string    `datastore:"result_url"`
	ResultError *string    `datastore:"result_error"`
}