-
Notifications
You must be signed in to change notification settings - Fork 31
/
broker.go
32 lines (27 loc) · 1.02 KB
/
broker.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package mq
import (
"context"
"github.com/runabol/tork"
)
type Provider func() (Broker, error)
const (
BROKER_INMEMORY = "inmemory"
BROKER_RABBITMQ = "rabbitmq"
TOPIC_JOB = "job.*"
TOPIC_JOB_COMPLETED = "job.completed"
TOPIC_JOB_FAILED = "job.failed"
)
// Broker is the message-queue, pub/sub mechanism used for delivering tasks.
type Broker interface {
Queues(ctx context.Context) ([]QueueInfo, error)
PublishTask(ctx context.Context, qname string, t *tork.Task) error
SubscribeForTasks(qname string, handler func(t *tork.Task) error) error
PublishHeartbeat(ctx context.Context, n *tork.Node) error
SubscribeForHeartbeats(handler func(n *tork.Node) error) error
PublishJob(ctx context.Context, j *tork.Job) error
SubscribeForJobs(handler func(j *tork.Job) error) error
Shutdown(ctx context.Context) error
PublishEvent(ctx context.Context, topic string, event any) error
SubscribeForEvents(ctx context.Context, pattern string, handler func(event any)) error
HealthCheck(ctx context.Context) error
}