htwkmooc/poseidon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Recover Runner Allocations on startup.

Browse Source
This commit is contained in:
Maximilian Paß
2023-03-31 17:26:44 +01:00
committed by Sebastian Serth
parent 038d71ff51
commit 8950ce29d8
5 changed files with 89 additions and 909 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
internal/nomad/api_querier.go
View File

@@ -39,11 +39,14 @@ type apiQuerier interface {
stdin io.Reader, stdout, stderr io.Writer) (int, error)
// listJobs loads all jobs with the specified prefix.
listJobs(prefix string) (allocationListStub []*nomadApi.JobListStub, err error)
listJobs(prefix string) (jobListStub []*nomadApi.JobListStub, err error)
// job returns the job of the given jobID.
job(jobID string) (job *nomadApi.Job, err error)
// listAllocations loads all allocations.
listAllocations() (allocationListStub []*nomadApi.AllocationListStub, err error)
// allocation returns the first allocation of the given job.
allocation(jobID string) (*nomadApi.Allocation, error)
@@ -228,6 +231,14 @@ func (nc *nomadAPIClient) job(jobID string) (job *nomadApi.Job, err error) {
return
}
func (nc *nomadAPIClient) listAllocations() ([]*nomadApi.AllocationListStub, error) {
allocationListStubs, _, err := nc.client.Allocations().List(nc.queryOptions())
if err != nil {
return nil, fmt.Errorf("error listing Nomad allocations: %w", err)
}
return allocationListStubs, nil
}
func (nc *nomadAPIClient) allocation(jobID string) (alloc *nomadApi.Allocation, err error) {
allocs, _, err := nc.client.Jobs().Allocations(jobID, false, nil)
if err != nil {