mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Stephan BehnkeandGitHub
6dc37be79e
## What changed Replaced one `WithFxOptionsForService` with fault injection. ## Why We want to eliminate `WithFxOptionsForService` as it is blocking us from migrating away from the `onebox.go` approach (which duplicates the fx setup) since we don't want to expose an equivalent method in `temporal/fx.go`.
78 lines
2.7 KiB
Go
78 lines
2.7 KiB
Go
// Code generated by gowrap. DO NOT EDIT.
|
|
// template: gowrap_template
|
|
// gowrap: http://github.com/hexdigest/gowrap
|
|
|
|
package faultinjection
|
|
|
|
//go:generate gowrap gen -p go.temporal.io/server/common/persistence -i QueueV2 -t gowrap_template -o queue_v2_gen.go -l ""
|
|
|
|
import (
|
|
"context"
|
|
|
|
_sourcePersistence "go.temporal.io/server/common/persistence"
|
|
)
|
|
|
|
type (
|
|
// faultInjectionQueueV2 implements QueueV2 interface with fault injection.
|
|
faultInjectionQueueV2 struct {
|
|
_sourcePersistence.QueueV2
|
|
generator faultGenerator
|
|
}
|
|
)
|
|
|
|
// newFaultInjectionQueueV2 returns faultInjectionQueueV2.
|
|
func newFaultInjectionQueueV2(
|
|
baseStore _sourcePersistence.QueueV2,
|
|
generator faultGenerator,
|
|
) *faultInjectionQueueV2 {
|
|
return &faultInjectionQueueV2{
|
|
QueueV2: baseStore,
|
|
generator: generator,
|
|
}
|
|
}
|
|
|
|
// CreateQueue wraps QueueV2.CreateQueue.
|
|
func (d faultInjectionQueueV2) CreateQueue(ctx context.Context, request *_sourcePersistence.InternalCreateQueueRequest) (ip1 *_sourcePersistence.InternalCreateQueueResponse, err error) {
|
|
err = d.generator.generate("CreateQueue", request).inject(func() error {
|
|
ip1, err = d.QueueV2.CreateQueue(ctx, request)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// EnqueueMessage wraps QueueV2.EnqueueMessage.
|
|
func (d faultInjectionQueueV2) EnqueueMessage(ctx context.Context, request *_sourcePersistence.InternalEnqueueMessageRequest) (ip1 *_sourcePersistence.InternalEnqueueMessageResponse, err error) {
|
|
err = d.generator.generate("EnqueueMessage", request).inject(func() error {
|
|
ip1, err = d.QueueV2.EnqueueMessage(ctx, request)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// ListQueues wraps QueueV2.ListQueues.
|
|
func (d faultInjectionQueueV2) ListQueues(ctx context.Context, request *_sourcePersistence.InternalListQueuesRequest) (ip1 *_sourcePersistence.InternalListQueuesResponse, err error) {
|
|
err = d.generator.generate("ListQueues", request).inject(func() error {
|
|
ip1, err = d.QueueV2.ListQueues(ctx, request)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// RangeDeleteMessages wraps QueueV2.RangeDeleteMessages.
|
|
func (d faultInjectionQueueV2) RangeDeleteMessages(ctx context.Context, request *_sourcePersistence.InternalRangeDeleteMessagesRequest) (ip1 *_sourcePersistence.InternalRangeDeleteMessagesResponse, err error) {
|
|
err = d.generator.generate("RangeDeleteMessages", request).inject(func() error {
|
|
ip1, err = d.QueueV2.RangeDeleteMessages(ctx, request)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// ReadMessages wraps QueueV2.ReadMessages.
|
|
func (d faultInjectionQueueV2) ReadMessages(ctx context.Context, request *_sourcePersistence.InternalReadMessagesRequest) (ip1 *_sourcePersistence.InternalReadMessagesResponse, err error) {
|
|
err = d.generator.generate("ReadMessages", request).inject(func() error {
|
|
ip1, err = d.QueueV2.ReadMessages(ctx, request)
|
|
return err
|
|
})
|
|
return
|
|
}
|