OPC/chorus
2
0
Fork 0
Code Issues 11 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 实现安全多 Provider worker (#22)

Browse Source
This commit is contained in:
ila
2026-08-21 21:00:25 +08:00
parent 9b2827a727
commit 35fd0342f0
22 changed files with 1498 additions and 375 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/core/queue/controller.go
+8
View File
@@ -40,6 +40,14 @@ func (c *Controller) AssignProvider(ctx context.Context, generationID uint64, le
return c.repository.AssignProvider(ctx, generationID, leaseToken, providerModelID)
}
func (c *Controller) BeginProviderAttempt(ctx context.Context, generationID uint64, leaseToken string, routeMemberID, providerModelID uint64) (model.Attempt, bool, error) {
return c.repository.BeginProviderAttempt(ctx, generationID, leaseToken, routeMemberID, providerModelID)
}
func (c *Controller) FinishProviderAttempt(ctx context.Context, generationID uint64, leaseToken string, attempt model.Attempt) (bool, error) {
return c.repository.FinishProviderAttempt(ctx, generationID, leaseToken, attempt)
}
func (c *Controller) Succeed(ctx context.Context, generationID uint64, leaseToken string, outputs []model.GenerationOutput, attempt model.Attempt) (bool, error) {
return c.repository.Succeed(ctx, generationID, leaseToken, outputs, attempt)
}