mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Yichao YangandGitHub
a3962bb3ae
## What changed? - Move chasm test payload store proto def to chasm package ## Why? - Follow the new convention we set up for chasm components ## How did you test it? - [x] built - [ ] run locally and tested manually - [x] covered by existing tests - [ ] added new unit test(s) - [ ] added new functional test(s)
75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
package tests
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.temporal.io/server/chasm"
|
|
"go.temporal.io/server/chasm/lib/tests/gen/testspb/v1"
|
|
)
|
|
|
|
type (
|
|
PayloadTTLPureTaskExecutor struct{}
|
|
PayloadTTLPureTaskValidator struct{}
|
|
)
|
|
|
|
func (e *PayloadTTLPureTaskExecutor) Execute(
|
|
mutableContext chasm.MutableContext,
|
|
store *PayloadStore,
|
|
_ chasm.TaskAttributes,
|
|
task *testspb.TestPayloadTTLPureTask,
|
|
) error {
|
|
_, err := store.RemovePayload(mutableContext, task.PayloadKey)
|
|
return err
|
|
}
|
|
|
|
func (v *PayloadTTLPureTaskValidator) Validate(
|
|
_ chasm.Context,
|
|
store *PayloadStore,
|
|
attributes chasm.TaskAttributes,
|
|
task *testspb.TestPayloadTTLPureTask,
|
|
) (bool, error) {
|
|
return validateTask(store, attributes, task.PayloadKey)
|
|
}
|
|
|
|
type (
|
|
PayloadTTLSideEffectTaskExecutor struct{}
|
|
PayloadTTLSideEffectTaskValidator struct{}
|
|
)
|
|
|
|
func (e *PayloadTTLSideEffectTaskExecutor) Execute(
|
|
ctx context.Context,
|
|
ref chasm.ComponentRef,
|
|
_ chasm.TaskAttributes,
|
|
task *testspb.TestPayloadTTLSideEffectTask,
|
|
) error {
|
|
_, _, err := chasm.UpdateComponent(
|
|
ctx,
|
|
ref,
|
|
(*PayloadStore).RemovePayload,
|
|
task.PayloadKey,
|
|
)
|
|
return err
|
|
}
|
|
|
|
func (v *PayloadTTLSideEffectTaskValidator) Validate(
|
|
_ chasm.Context,
|
|
store *PayloadStore,
|
|
attributes chasm.TaskAttributes,
|
|
task *testspb.TestPayloadTTLSideEffectTask,
|
|
) (bool, error) {
|
|
return validateTask(store, attributes, task.PayloadKey)
|
|
}
|
|
|
|
func validateTask(
|
|
store *PayloadStore,
|
|
attributes chasm.TaskAttributes,
|
|
payloadKey string,
|
|
) (bool, error) {
|
|
expirationTime, ok := store.State.ExpirationTimes[payloadKey]
|
|
if !ok {
|
|
return false, nil
|
|
}
|
|
|
|
return !expirationTime.AsTime().After(attributes.ScheduledTime), nil
|
|
}
|