feat: add executor (#3)
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
Adds an executor which can process and dispatch events to a set of workers. Co-authored-by: kjuulh <contact@kjuulh.io> Co-committed-by: kjuulh <contact@kjuulh.io>
This commit is contained in:
parent
86a32513cc
commit
cda23ebd12
@ -3,6 +3,7 @@ package app
|
|||||||
import (
|
import (
|
||||||
"log/slog"
|
"log/slog"
|
||||||
|
|
||||||
|
"git.front.kjuulh.io/kjuulh/orbis/internal/executor"
|
||||||
"git.front.kjuulh.io/kjuulh/orbis/internal/scheduler"
|
"git.front.kjuulh.io/kjuulh/orbis/internal/scheduler"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -21,5 +22,9 @@ func (a *App) Logger() *slog.Logger {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) Scheduler() *scheduler.Scheduler {
|
func (a *App) Scheduler() *scheduler.Scheduler {
|
||||||
return scheduler.NewScheduler(a.logger, Postgres())
|
return scheduler.NewScheduler(a.logger.With("component", "scheduler"), Postgres(), a.Executor())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *App) Executor() *executor.Executor {
|
||||||
|
return executor.NewExecutor(a.logger.With("component", "executor"))
|
||||||
}
|
}
|
||||||
|
22
internal/executor/executor.go
Normal file
22
internal/executor/executor.go
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
package executor
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log/slog"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Executor struct {
|
||||||
|
logger *slog.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewExecutor(logger *slog.Logger) *Executor {
|
||||||
|
return &Executor{
|
||||||
|
logger: logger,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *Executor) DispatchEvents(ctx context.Context) error {
|
||||||
|
e.logger.InfoContext(ctx, "dispatching events")
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
@ -8,18 +8,21 @@ import (
|
|||||||
"math/rand"
|
"math/rand"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"git.front.kjuulh.io/kjuulh/orbis/internal/executor"
|
||||||
"github.com/jackc/pgx/v5"
|
"github.com/jackc/pgx/v5"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Scheduler struct {
|
type Scheduler struct {
|
||||||
logger *slog.Logger
|
logger *slog.Logger
|
||||||
db *pgx.Conn
|
db *pgx.Conn
|
||||||
|
executor *executor.Executor
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewScheduler(logger *slog.Logger, db *pgx.Conn) *Scheduler {
|
func NewScheduler(logger *slog.Logger, db *pgx.Conn, executor *executor.Executor) *Scheduler {
|
||||||
return &Scheduler{
|
return &Scheduler{
|
||||||
logger: logger,
|
logger: logger,
|
||||||
db: db,
|
db: db,
|
||||||
|
executor: executor,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -34,12 +37,14 @@ func (s *Scheduler) Execute(ctx context.Context) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ticker := time.NewTicker(5 * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
case <-ctx.Done():
|
case <-ctx.Done():
|
||||||
s.logger.Info("gracefully shutting down elected scheduler")
|
s.logger.Info("gracefully shutting down elected scheduler")
|
||||||
return nil
|
return nil
|
||||||
default:
|
case <-ticker.C:
|
||||||
if err := s.process(ctx); err != nil {
|
if err := s.process(ctx); err != nil {
|
||||||
return fmt.Errorf("scheduler failed: %w", err)
|
return fmt.Errorf("scheduler failed: %w", err)
|
||||||
}
|
}
|
||||||
@ -52,7 +57,6 @@ func (s *Scheduler) acquireLeader(ctx context.Context) (bool, error) {
|
|||||||
select {
|
select {
|
||||||
case <-ctx.Done():
|
case <-ctx.Done():
|
||||||
return false, nil
|
return false, nil
|
||||||
|
|
||||||
default:
|
default:
|
||||||
var acquiredLock bool
|
var acquiredLock bool
|
||||||
if err := s.db.QueryRow(ctx, "SELECT pg_try_advisory_lock(1234)").Scan(&acquiredLock); err != nil {
|
if err := s.db.QueryRow(ctx, "SELECT pg_try_advisory_lock(1234)").Scan(&acquiredLock); err != nil {
|
||||||
@ -77,10 +81,9 @@ func (s *Scheduler) acquireLeader(ctx context.Context) (bool, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Scheduler) process(ctx context.Context) error {
|
func (s *Scheduler) process(ctx context.Context) error {
|
||||||
s.logger.Debug("scheduler processing items")
|
if err := s.executor.DispatchEvents(ctx); err != nil {
|
||||||
|
return fmt.Errorf("failed to dispatch events: %w", err)
|
||||||
// FIXME: simulate work
|
}
|
||||||
time.Sleep(time.Second * 2)
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user