particle-os/debian-forge-composer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

pkg/jobqueue: add arch to worker

Browse source
This commit is contained in:
Sanne Raymaekers 2023-12-08 12:20:36 +01:00
parent 8a8b1406fb
commit ac854b7cc8
4 changed files with 28 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

16
pkg/jobqueue/dbjobqueue/dbjobqueue.go
View file

@ -124,7 +124,7 @@ const (
SET heartbeat = now()
WHERE worker_id = $1`
sqlQueryWorkers = `
SELECT worker_id
SELECT worker_id, arch
FROM workers
WHERE age(now(), heartbeat) > $1`
sqlDeleteWorker = `
@ -746,7 +746,7 @@ func (q *DBJobQueue) UpdateWorkerStatus(workerID uuid.UUID) error {
}
return nil
}
func (q *DBJobQueue) Workers(olderThan time.Duration) ([]uuid.UUID, error) {
func (q *DBJobQueue) Workers(olderThan time.Duration) ([]jobqueue.Worker, error) {
conn, err := q.pool.Acquire(context.Background())
if err != nil {
return nil, err
@ -759,23 +759,27 @@ func (q *DBJobQueue) Workers(olderThan time.Duration) ([]uuid.UUID, error) {
}
defer rows.Close()
workerIDs := make([]uuid.UUID, 0)
workers := make([]jobqueue.Worker, 0)
for rows.Next() {
var w uuid.UUID
err = rows.Scan(&w)
var a string
err = rows.Scan(&w, &a)
if err != nil {
// Log the error and try to continue with the next row
q.logger.Error(err, "Unable to read token from heartbeats")
continue
}
workerIDs = append(workerIDs, w)
workers = append(workers, jobqueue.Worker{
ID: w,
Arch: a,
})
}
if rows.Err() != nil {
q.logger.Error(rows.Err(), "Error reading tokens from heartbeats")
return nil, rows.Err()
}
return workerIDs, nil
return workers, nil
}
func (q *DBJobQueue) DeleteWorker(workerID uuid.UUID) error {

7
pkg/jobqueue/jobqueue.go
View file

@ -88,7 +88,7 @@ type JobQueue interface {
UpdateWorkerStatus(workerID uuid.UUID) error
// Get a list of workers which haven't been updated in the specified time frame
Workers(olderThan time.Duration) ([]uuid.UUID, error)
Workers(olderThan time.Duration) ([]Worker, error)
// Deletes the worker
DeleteWorker(workerID uuid.UUID) error
@ -115,3 +115,8 @@ var (
ErrActiveJobs = errors.New("worker has active jobs associated with it")
ErrWorkerNotExist = errors.New("worker does not exist")
)
type Worker struct {
ID uuid.UUID
Arch string
}