69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
package job
|
||
|
||
import (
|
||
"strings"
|
||
|
||
"haixun-backend/internal/library/clock"
|
||
app "haixun-backend/internal/library/errors"
|
||
"haixun-backend/internal/library/errors/code"
|
||
"haixun-backend/internal/model/ai/domain/enum"
|
||
jobentity "haixun-backend/internal/model/job/domain/entity"
|
||
jobenum "haixun-backend/internal/model/job/domain/enum"
|
||
"haixun-backend/internal/types"
|
||
)
|
||
|
||
func mapWorkerAIProvider(provider string) (enum.ProviderID, error) {
|
||
switch strings.TrimSpace(provider) {
|
||
case string(enum.ProviderOpenCode):
|
||
return enum.ProviderOpenCode, nil
|
||
case string(enum.ProviderXAI):
|
||
return enum.ProviderXAI, nil
|
||
default:
|
||
return "", app.For(code.AI).InputInvalidFormat("worker 8D 分析目前僅支援 opencode-go 與 xai，請在 AI 設定調整 research provider")
|
||
}
|
||
}
|
||
|
||
func markWorkerStep(steps []jobentity.StepProgress, stepID string, status jobenum.StepStatus, message string) []jobentity.StepProgress {
|
||
now := clock.NowUnixNano()
|
||
found := false
|
||
for i := range steps {
|
||
if steps[i].ID != stepID {
|
||
continue
|
||
}
|
||
found = true
|
||
steps[i].Status = status
|
||
steps[i].Message = message
|
||
if status == jobenum.StepStatusRunning && steps[i].StartedAt == nil {
|
||
steps[i].StartedAt = &now
|
||
}
|
||
if status == jobenum.StepStatusSucceeded || status == jobenum.StepStatusFailed {
|
||
steps[i].EndedAt = &now
|
||
}
|
||
}
|
||
if !found {
|
||
item := jobentity.StepProgress{ID: stepID, Status: status, Message: message}
|
||
if status == jobenum.StepStatusRunning {
|
||
item.StartedAt = &now
|
||
}
|
||
if status == jobenum.StepStatusSucceeded || status == jobenum.StepStatusFailed {
|
||
item.EndedAt = &now
|
||
}
|
||
steps = append(steps, item)
|
||
}
|
||
return steps
|
||
}
|
||
|
||
func toEntitySteps(steps []types.JobStepProgressData) []jobentity.StepProgress {
|
||
out := make([]jobentity.StepProgress, 0, len(steps))
|
||
for _, step := range steps {
|
||
out = append(out, jobentity.StepProgress{
|
||
ID: step.ID,
|
||
Status: jobenum.StepStatus(step.Status),
|
||
StartedAt: step.StartedAt,
|
||
EndedAt: step.EndedAt,
|
||
Message: step.Message,
|
||
})
|
||
}
|
||
return out
|
||
}
|