feat: add worker distributor and model registry

2025-01-18 01:46:37 +01:00
parent 54aa310583
commit 2cdab4a1ab
28 changed files with 1169 additions and 29 deletions
--- a/internal/worker/repositories/querier.go
+++ b/internal/worker/repositories/querier.go
@@ -11,8 +11,9 @@ import (
 )

 type Querier interface {
+	GetWorkers(ctx context.Context) ([]*GetWorkersRow, error)
 	Ping(ctx context.Context) (int32, error)
-	RegisterWorker(ctx context.Context, workerID uuid.UUID) error
+	RegisterWorker(ctx context.Context, arg *RegisterWorkerParams) error
 	UpdateWorkerHeartbeat(ctx context.Context, workerID uuid.UUID) error
 }