实现持久化 assignment handoff

internal/taskworker/worker.go

@@ -6,8 +6,8 @@ import (
 	"errors"
 	"strconv"
 
+	"git.ddupan.top/panxiao81/gitea-dynamic-runner/internal/taskassignment"
 	"git.ddupan.top/panxiao81/gitea-dynamic-runner/internal/taskidentity"
-	"git.ddupan.top/panxiao81/gitea-dynamic-runner/internal/taskscheduler"
 )
 
 // Phase is observed from Kubernetes or OpenSandbox, never stored by the worker.
@@ -38,7 +38,7 @@ type Metadata struct {
 // Every method must be idempotent.
 type Backend interface {
 	Find(context.Context, string) (*Executor, error)
-	Create(context.Context, taskscheduler.Assignment, Metadata) (*Executor, error)
+	Create(context.Context, taskassignment.Assignment, Metadata) (*Executor, error)
 	BindIdentity(context.Context, *Executor, taskidentity.Identity) error
 	Delete(context.Context, *Executor) error
 }
@@ -59,7 +59,7 @@ type Worker struct {
 
 // Handle performs one reconciliation. Done means the queue message may be
 // acknowledged. A false result should remain pending and be reconciled again.
-func (w Worker) Handle(ctx context.Context, assignment taskscheduler.Assignment) (done bool, err error) {
+func (w Worker) Handle(ctx context.Context, assignment taskassignment.Assignment) (done bool, err error) {
 	if w.Backend == nil || w.Tasks == nil {
 		return false, errors.New("backend and Gitea task state are required")
 	}
@@ -114,11 +114,12 @@ func (w Worker) Handle(ctx context.Context, assignment taskscheduler.Assignment)
 }
 
 // BackendMetadata is the shared metadata contract for Pods and OpenSandbox.
-func BackendMetadata(assignment taskscheduler.Assignment) Metadata {
+func BackendMetadata(assignment taskassignment.Assignment) Metadata {
 	return Metadata{
 		Labels: map[string]string{
 			"ci.ddupan.top/assignment-id": assignment.ID,
 			"ci.ddupan.top/task-id":       strconv.FormatInt(assignment.Task.GetId(), 10),
+			"ci.ddupan.top/backend":       string(assignment.Backend),
 		},
 		Annotations: map[string]string{
 			"ci.ddupan.top/repository": assignment.Identity.Repository,

internal/taskworker/worker_test.go

@@ -6,8 +6,8 @@ import (
 
 	runnerv1 "gitea.dev/actionslib/runner/v1"
 
+	"git.ddupan.top/panxiao81/gitea-dynamic-runner/internal/taskassignment"
 	"git.ddupan.top/panxiao81/gitea-dynamic-runner/internal/taskidentity"
-	"git.ddupan.top/panxiao81/gitea-dynamic-runner/internal/taskscheduler"
 )
 
 type fakeBackend struct {
@@ -18,7 +18,7 @@ type fakeBackend struct {
 }
 
 func (b *fakeBackend) Find(context.Context, string) (*Executor, error) { return b.executor, nil }
-func (b *fakeBackend) Create(_ context.Context, _ taskscheduler.Assignment, _ Metadata) (*Executor, error) {
+func (b *fakeBackend) Create(_ context.Context, _ taskassignment.Assignment, _ Metadata) (*Executor, error) {
 	b.created++
 	b.executor = &Executor{Name: "executor", IdentityTarget: "pod-uid", Phase: PhaseRunning}
 	return b.executor, nil
@@ -45,10 +45,11 @@ func (t *fakeTasks) Report(_ context.Context, _ int64, phase Phase) error {
 	return nil
 }
 
-func assignment() taskscheduler.Assignment {
-	return taskscheduler.Assignment{
-		ID:   "gitea-task-42",
-		Task: &runnerv1.Task{Id: 42},
+func assignment() taskassignment.Assignment {
+	return taskassignment.Assignment{
+		ID:      "gitea-task-42",
+		Backend: taskassignment.BackendPod,
+		Task:    &runnerv1.Task{Id: 42},
 		Identity: taskidentity.Identity{
 			Repository: "owner/repo",
 			Task:       "publish",