Skip to content

Commit 97a13f2

Browse files
authored
Merge pull request #313 from cschleiden/start-cache-eviction
Start cache eviction
2 parents 2e4b887 + 5b0ce1e commit 97a13f2

File tree

3 files changed

+17
-0
lines changed

3 files changed

+17
-0
lines changed

internal/worker/activity.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,10 @@ func (atw *ActivityTaskWorker) Complete(ctx context.Context, event *history.Even
5050
return nil
5151
}
5252

53+
func (atw *ActivityTaskWorker) Start(ctx context.Context) error {
54+
return nil
55+
}
56+
5357
func (atw *ActivityTaskWorker) Execute(ctx context.Context, task *backend.ActivityTask) (*history.Event, error) {
5458
a := task.Event.Attributes.(*history.ActivityScheduledAttributes)
5559
ametrics := atw.backend.Metrics().WithTags(metrics.Tags{metrickeys.ActivityName: a.Name})

internal/worker/worker.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ import (
1212
)
1313

1414
type TaskWorker[Task, Result any] interface {
15+
Start(context.Context) error
1516
Get(context.Context) (*Task, error)
1617
Extend(context.Context, *Task) error
1718
Execute(context.Context, *Task) (*Result, error)
@@ -55,6 +56,10 @@ func NewWorker[Task, TaskResult any](
5556
}
5657

5758
func (w *Worker[Task, TaskResult]) Start(ctx context.Context) error {
59+
if err := w.tw.Start(ctx); err != nil {
60+
return fmt.Errorf("starting task worker: %w", err)
61+
}
62+
5863
w.pollersWg.Add(w.options.Pollers)
5964

6065
for i := 0; i < w.options.Pollers; i++ {

internal/worker/workflow.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,14 @@ type WorkflowTaskWorker struct {
5353
logger *slog.Logger
5454
}
5555

56+
func (wtw *WorkflowTaskWorker) Start(ctx context.Context) error {
57+
if wtw.cache != nil {
58+
go wtw.cache.StartEviction(ctx)
59+
}
60+
61+
return nil
62+
}
63+
5664
// Complete implements TaskWorker.
5765
func (wtw *WorkflowTaskWorker) Complete(ctx context.Context, result *workflow.ExecutionResult, t *backend.WorkflowTask) error {
5866
state := result.State

0 commit comments

Comments
 (0)