mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
101cb2ff7a
GetExecution returned (nil, nil) under the in-process driver where the asynq driver returns an error, so the same contract call meant 'empty' in one deployment mode and 'failed' in the other.
83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package driver_inproc_worker
|
|
|
|
import (
|
|
"Wavelet/core/contracts"
|
|
"Wavelet/core/extpoints"
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
)
|
|
|
|
type inprocTaskService struct {
|
|
taskReg extpoints.TaskExtension
|
|
}
|
|
|
|
func newInprocTaskService(taskReg extpoints.TaskExtension) contracts.TaskService {
|
|
return &inprocTaskService{
|
|
taskReg: taskReg,
|
|
}
|
|
}
|
|
|
|
func (s *inprocTaskService) Dispatch(ctx context.Context, taskType string, payload []byte, triggeredBy string) (string, error) {
|
|
return DispatchTask(ctx, taskType, payload, triggeredBy)
|
|
}
|
|
|
|
func (s *inprocTaskService) Retry(_ context.Context, id uint64) (string, error) {
|
|
return fmt.Sprintf("inproc_retry_%d", id), nil
|
|
}
|
|
|
|
func (s *inprocTaskService) ListTasks() []contracts.TaskMetaDTO {
|
|
if s.taskReg == nil {
|
|
return nil
|
|
}
|
|
tasks := s.taskReg.Tasks()
|
|
res := make([]contracts.TaskMetaDTO, 0, len(tasks))
|
|
for _, td := range tasks {
|
|
res = append(res, contracts.TaskMetaDTO{
|
|
Name: td.Pattern,
|
|
DisplayName: td.Pattern,
|
|
MaxRetry: td.Retry,
|
|
Timeout: td.Timeout,
|
|
})
|
|
}
|
|
return res
|
|
}
|
|
|
|
func (s *inprocTaskService) GetTaskMeta(taskType string) (contracts.TaskMetaDTO, bool) {
|
|
if s.taskReg == nil {
|
|
return contracts.TaskMetaDTO{}, false
|
|
}
|
|
td, ok := s.taskReg.Get(taskType)
|
|
if !ok {
|
|
return contracts.TaskMetaDTO{}, false
|
|
}
|
|
return contracts.TaskMetaDTO{
|
|
Name: td.Pattern,
|
|
DisplayName: td.Pattern,
|
|
MaxRetry: td.Retry,
|
|
Timeout: td.Timeout,
|
|
}, true
|
|
}
|
|
|
|
func (s *inprocTaskService) ValidatePayload(_ string, payload []byte) ([]byte, error) {
|
|
return payload, nil
|
|
}
|
|
|
|
func (s *inprocTaskService) ReloadScheduler() error {
|
|
return nil
|
|
}
|
|
|
|
func (s *inprocTaskService) AppendLog(_ context.Context, _ string, _ ...any) {
|
|
}
|
|
|
|
func (s *inprocTaskService) ListExecutions(_ context.Context, _, _ string, _, _ int) ([]contracts.TaskExecutionDTO, int64, error) {
|
|
return []contracts.TaskExecutionDTO{}, 0, nil
|
|
}
|
|
|
|
func (s *inprocTaskService) GetExecution(_ context.Context, _ uint64) (*contracts.TaskExecutionDTO, error) {
|
|
return nil, errors.New("driver_inproc_worker: task executions are not tracked")
|
|
}
|