feat: 自动清理终态 Pod 执行器
This commit is contained in:
@@ -57,7 +57,7 @@ type Facade struct {
|
||||
Registry *Registry
|
||||
Capabilities Capabilities
|
||||
Upstream Upstream
|
||||
OnTerminal func(taskassignment.Assignment)
|
||||
OnTerminal func(context.Context, taskassignment.Assignment) error
|
||||
}
|
||||
|
||||
func (f *Facade) Handler() (string, http.Handler) {
|
||||
@@ -101,7 +101,9 @@ func (f *Facade) UpdateTask(ctx context.Context, request *connect.Request[runner
|
||||
}
|
||||
response, err := f.Upstream.UpdateTask(ctx, connect.NewRequest(request.Msg))
|
||||
if err == nil && request.Msg.GetState().GetResult() != runnerv1.Result_RESULT_UNSPECIFIED && f.OnTerminal != nil {
|
||||
f.OnTerminal(assignment)
|
||||
if terminalErr := f.OnTerminal(ctx, assignment); terminalErr != nil {
|
||||
return nil, connect.NewError(connect.CodeUnavailable, terminalErr)
|
||||
}
|
||||
}
|
||||
return response, err
|
||||
}
|
||||
|
||||
@@ -170,11 +170,12 @@ func TestFacadeSignalsTerminalTaskAfterUpstreamAcceptsIt(t *testing.T) {
|
||||
t.Fatal(err)
|
||||
}
|
||||
completed := 0
|
||||
facade.OnTerminal = func(got taskassignment.Assignment) {
|
||||
facade.OnTerminal = func(_ context.Context, got taskassignment.Assignment) error {
|
||||
if got.ID != assignment.ID {
|
||||
t.Fatalf("terminal assignment = %s", got.ID)
|
||||
}
|
||||
completed++
|
||||
return nil
|
||||
}
|
||||
if _, err := facade.UpdateTask(ctx, authenticatedRequest(&runnerv1.UpdateTaskRequest{
|
||||
State: &runnerv1.TaskState{Id: 42, Result: runnerv1.Result_RESULT_SUCCESS},
|
||||
|
||||
Reference in New Issue
Block a user