mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
feat(drivers): provide TaskService implementation in inproc worker
This commit is contained in:
@@ -10,6 +10,7 @@ import (
|
||||
"time"
|
||||
|
||||
"Wavelet/core"
|
||||
"Wavelet/core/contracts"
|
||||
)
|
||||
|
||||
const (
|
||||
@@ -98,10 +99,13 @@ func (p *Plugin) Manifest() core.Manifest {
|
||||
}
|
||||
}
|
||||
|
||||
// Apply registers the worker driver into the Context.
|
||||
// Apply registers the worker driver and provides contracts.TaskService.
|
||||
func (p *Plugin) Apply(ctx *core.Context) error {
|
||||
p.coreCtx = ctx
|
||||
|
||||
taskSvc := newInprocTaskService(ctx.Tasks())
|
||||
core.Provide[contracts.TaskService](ctx, taskSvc)
|
||||
|
||||
ctx.OnDispose(func() error {
|
||||
shutdownCtx, cancel := context.WithTimeout(context.Background(), p.shutdownTimeout)
|
||||
defer cancel()
|
||||
|
||||
@@ -0,0 +1,82 @@
|
||||
// Copyright 2026 Arctel.net
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
package driver_inproc_worker
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"Wavelet/core/contracts"
|
||||
"Wavelet/core/extpoints"
|
||||
)
|
||||
|
||||
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(ctx 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, _ string, _, _ int) ([]contracts.TaskExecutionDTO, int64, error) {
|
||||
return []contracts.TaskExecutionDTO{}, 0, nil
|
||||
}
|
||||
|
||||
func (s *inprocTaskService) GetExecution(_ context.Context, _ uint64) (*contracts.TaskExecutionDTO, error) {
|
||||
return nil, nil
|
||||
}
|
||||
Reference in New Issue
Block a user