worker

Documentation

Index

• func EmitHeartbeat(nc *nats.Conn, payload []byte) error
• func HeartbeatLoop(ctx context.Context, nc *nats.Conn, payloadFn func() ([]byte, error))
• func HeartbeatPayload(workerID, pool string, activeJobs, maxParallel int, cpuLoad float32) ([]byte, error)
• type Handler
• type NATSConn
• type Worker
  • func (w *Worker) Start() error

Functions

func EmitHeartbeat

func EmitHeartbeat(nc *nats.Conn, payload []byte) error

EmitHeartbeat publishes a heartbeat once. Call repeatedly on a ticker.

func HeartbeatLoop

func HeartbeatLoop(ctx context.Context, nc *nats.Conn, payloadFn func() ([]byte, error))

HeartbeatLoop emits heartbeats until ctx is done.

func HeartbeatPayload

func HeartbeatPayload(workerID, pool string, activeJobs, maxParallel int, cpuLoad float32) ([]byte, error)

HeartbeatPayload returns a protobuf-encoded heartbeat envelope.

Types

type Handler

type Handler func(context.Context, *agentv1.JobRequest) (*agentv1.JobResult, error)

Handler processes a JobRequest and returns a JobResult.

type NATSConn

type NATSConn interface {
	Publish(subject string, data []byte) error
	QueueSubscribe(subj, queue string, cb nats.MsgHandler) (*nats.Subscription, error)
}

NATSConn is an interface that represents a NATS connection.

type Worker

type Worker struct {
	NATS       NATSConn
	Subject    string
	Handler    Handler
	PublicKeys map[string]*ecdsa.PublicKey
	PrivateKey *ecdsa.PrivateKey
	SenderID   string
}

Worker subscribes to a pool subject and handles jobs.

func (*Worker) Start

func (w *Worker) Start() error

Start begins consuming and handling JobRequests. It blocks until the subscription is closed.