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`.
124 lines
4.2 KiB
Go
124 lines
4.2 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 Queue -t gowrap_template -o queue_gen.go -l ""
|
|
|
|
import (
|
|
"context"
|
|
|
|
commonpb "go.temporal.io/api/common/v1"
|
|
_sourcePersistence "go.temporal.io/server/common/persistence"
|
|
)
|
|
|
|
type (
|
|
// faultInjectionQueue implements Queue interface with fault injection.
|
|
faultInjectionQueue struct {
|
|
_sourcePersistence.Queue
|
|
generator faultGenerator
|
|
}
|
|
)
|
|
|
|
// newFaultInjectionQueue returns faultInjectionQueue.
|
|
func newFaultInjectionQueue(
|
|
baseStore _sourcePersistence.Queue,
|
|
generator faultGenerator,
|
|
) *faultInjectionQueue {
|
|
return &faultInjectionQueue{
|
|
Queue: baseStore,
|
|
generator: generator,
|
|
}
|
|
}
|
|
|
|
// DeleteMessageFromDLQ wraps Queue.DeleteMessageFromDLQ.
|
|
func (d faultInjectionQueue) DeleteMessageFromDLQ(ctx context.Context, messageID int64) (err error) {
|
|
err = d.generator.generate("DeleteMessageFromDLQ", messageID).inject(func() error {
|
|
err = d.Queue.DeleteMessageFromDLQ(ctx, messageID)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// DeleteMessagesBefore wraps Queue.DeleteMessagesBefore.
|
|
func (d faultInjectionQueue) DeleteMessagesBefore(ctx context.Context, messageID int64) (err error) {
|
|
err = d.generator.generate("DeleteMessagesBefore", messageID).inject(func() error {
|
|
err = d.Queue.DeleteMessagesBefore(ctx, messageID)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// EnqueueMessage wraps Queue.EnqueueMessage.
|
|
func (d faultInjectionQueue) EnqueueMessage(ctx context.Context, blob *commonpb.DataBlob) (err error) {
|
|
err = d.generator.generate("EnqueueMessage", blob).inject(func() error {
|
|
err = d.Queue.EnqueueMessage(ctx, blob)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// EnqueueMessageToDLQ wraps Queue.EnqueueMessageToDLQ.
|
|
func (d faultInjectionQueue) EnqueueMessageToDLQ(ctx context.Context, blob *commonpb.DataBlob) (i1 int64, err error) {
|
|
err = d.generator.generate("EnqueueMessageToDLQ", blob).inject(func() error {
|
|
i1, err = d.Queue.EnqueueMessageToDLQ(ctx, blob)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// Init wraps Queue.Init.
|
|
func (d faultInjectionQueue) Init(ctx context.Context, blob *commonpb.DataBlob) (err error) {
|
|
err = d.generator.generate("Init", blob).inject(func() error {
|
|
err = d.Queue.Init(ctx, blob)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// RangeDeleteMessagesFromDLQ wraps Queue.RangeDeleteMessagesFromDLQ.
|
|
func (d faultInjectionQueue) RangeDeleteMessagesFromDLQ(ctx context.Context, firstMessageID int64, lastMessageID int64) (err error) {
|
|
err = d.generator.generate("RangeDeleteMessagesFromDLQ", firstMessageID).inject(func() error {
|
|
err = d.Queue.RangeDeleteMessagesFromDLQ(ctx, firstMessageID, lastMessageID)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// ReadMessages wraps Queue.ReadMessages.
|
|
func (d faultInjectionQueue) ReadMessages(ctx context.Context, lastMessageID int64, maxCount int) (qpa1 []*_sourcePersistence.QueueMessage, err error) {
|
|
err = d.generator.generate("ReadMessages", lastMessageID).inject(func() error {
|
|
qpa1, err = d.Queue.ReadMessages(ctx, lastMessageID, maxCount)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// ReadMessagesFromDLQ wraps Queue.ReadMessagesFromDLQ.
|
|
func (d faultInjectionQueue) ReadMessagesFromDLQ(ctx context.Context, firstMessageID int64, lastMessageID int64, pageSize int, pageToken []byte) (qpa1 []*_sourcePersistence.QueueMessage, ba1 []byte, err error) {
|
|
err = d.generator.generate("ReadMessagesFromDLQ", firstMessageID).inject(func() error {
|
|
qpa1, ba1, err = d.Queue.ReadMessagesFromDLQ(ctx, firstMessageID, lastMessageID, pageSize, pageToken)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// UpdateAckLevel wraps Queue.UpdateAckLevel.
|
|
func (d faultInjectionQueue) UpdateAckLevel(ctx context.Context, metadata *_sourcePersistence.InternalQueueMetadata) (err error) {
|
|
err = d.generator.generate("UpdateAckLevel", metadata).inject(func() error {
|
|
err = d.Queue.UpdateAckLevel(ctx, metadata)
|
|
return err
|
|
})
|
|
return
|
|
}
|
|
|
|
// UpdateDLQAckLevel wraps Queue.UpdateDLQAckLevel.
|
|
func (d faultInjectionQueue) UpdateDLQAckLevel(ctx context.Context, metadata *_sourcePersistence.InternalQueueMetadata) (err error) {
|
|
err = d.generator.generate("UpdateDLQAckLevel", metadata).inject(func() error {
|
|
err = d.Queue.UpdateDLQAckLevel(ctx, metadata)
|
|
return err
|
|
})
|
|
return
|
|
}
|