|
| 1 | +package services |
| 2 | + |
| 3 | +import ( |
| 4 | + "context" |
| 5 | + "encoding/json" |
| 6 | + "fmt" |
| 7 | + "time" |
| 8 | + |
| 9 | + "github.com/NdoleStudio/httpsms/pkg/telemetry" |
| 10 | + cloudevents "github.com/cloudevents/sdk-go/v2" |
| 11 | + "github.com/google/uuid" |
| 12 | +) |
| 13 | + |
| 14 | +type inMemoryPushQueue struct { |
| 15 | + getEventDispatcher func() *EventDispatcher |
| 16 | + queueConfig PushQueueConfig |
| 17 | + logger telemetry.Logger |
| 18 | + tracer telemetry.Tracer |
| 19 | +} |
| 20 | + |
| 21 | +// NewGooglePushQueue creates a new googlePushQueue |
| 22 | +func NewInMemoryPushQueue( |
| 23 | + logger telemetry.Logger, |
| 24 | + tracer telemetry.Tracer, |
| 25 | + queueConfig PushQueueConfig, |
| 26 | + getEventDispatcher func() *EventDispatcher, |
| 27 | +) PushQueue { |
| 28 | + return &inMemoryPushQueue{ |
| 29 | + tracer: tracer, |
| 30 | + logger: logger, |
| 31 | + queueConfig: queueConfig, |
| 32 | + getEventDispatcher: getEventDispatcher, |
| 33 | + } |
| 34 | +} |
| 35 | + |
| 36 | +// Enqueue a task to the queue |
| 37 | +func (queue *inMemoryPushQueue) Enqueue(ctx context.Context, task *PushQueueTask, timeout time.Duration) (queueID string, err error) { |
| 38 | + ctx, span := queue.tracer.Start(ctx) |
| 39 | + defer span.End() |
| 40 | + |
| 41 | + ctxLogger := queue.tracer.CtxLogger(queue.logger, span) |
| 42 | + queueID = uuid.New().String() |
| 43 | + |
| 44 | + go func() { |
| 45 | + time.Sleep(timeout) |
| 46 | + var event cloudevents.Event |
| 47 | + json.Unmarshal(task.Body, &event) |
| 48 | + queue.getEventDispatcher().DispatchSync(ctx, event) |
| 49 | + }() |
| 50 | + |
| 51 | + ctxLogger.Info(fmt.Sprintf( |
| 52 | + "item added to [%s] queue with id [%s] and schedule [%s]", |
| 53 | + queue.queueConfig.Name, |
| 54 | + queueID, |
| 55 | + time.Now().UTC().Add(timeout), |
| 56 | + )) |
| 57 | + |
| 58 | + return queueID, nil |
| 59 | +} |
0 commit comments