root
3f5b5c1e01
Extract monitoring recovery/inspection workers and schedule dispatch into a dedicated scheduler process. Add worker-generate process for article generation and template-assist queue consumption. Introduce shared/schedule runtime for cron-based worker lifecycle management. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
89 lines
2.0 KiB
Go
89 lines
2.0 KiB
Go
package scheduler
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
"go.uber.org/zap"
|
|
|
|
tenantapp "github.com/geo-platform/tenant-api/internal/tenant/app"
|
|
)
|
|
|
|
const (
|
|
defaultMonitoringLeaseRecoveryInterval = 30 * time.Minute
|
|
defaultMonitoringLeaseRecoveryTimeout = 30 * time.Second
|
|
)
|
|
|
|
type MonitoringLeaseRecoveryWorker struct {
|
|
monitoringPool *pgxpool.Pool
|
|
logger *zap.Logger
|
|
interval time.Duration
|
|
timeout time.Duration
|
|
}
|
|
|
|
func NewMonitoringLeaseRecoveryWorker(monitoringPool *pgxpool.Pool, logger *zap.Logger) *MonitoringLeaseRecoveryWorker {
|
|
return &MonitoringLeaseRecoveryWorker{
|
|
monitoringPool: monitoringPool,
|
|
logger: logger,
|
|
interval: defaultMonitoringLeaseRecoveryInterval,
|
|
timeout: defaultMonitoringLeaseRecoveryTimeout,
|
|
}
|
|
}
|
|
|
|
func (w *MonitoringLeaseRecoveryWorker) Start(ctx context.Context) {
|
|
if w == nil || w.monitoringPool == nil {
|
|
return
|
|
}
|
|
go w.run(ctx)
|
|
}
|
|
|
|
func (w *MonitoringLeaseRecoveryWorker) run(ctx context.Context) {
|
|
w.runOnce(ctx)
|
|
|
|
ticker := time.NewTicker(w.interval)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-ticker.C:
|
|
w.runOnce(ctx)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (w *MonitoringLeaseRecoveryWorker) runOnce(parent context.Context) {
|
|
ctx, cancel := context.WithTimeout(parent, w.timeout)
|
|
defer cancel()
|
|
|
|
tx, err := w.monitoringPool.Begin(ctx)
|
|
if err != nil {
|
|
if w.logger != nil {
|
|
w.logger.Warn("monitoring lease recovery begin failed", zap.Error(err))
|
|
}
|
|
return
|
|
}
|
|
defer tx.Rollback(ctx)
|
|
|
|
expiredCount, err := tenantapp.ExpireMonitoringLeasedTasks(ctx, tx, nil, time.Now().UTC())
|
|
if err != nil {
|
|
if w.logger != nil {
|
|
w.logger.Warn("monitoring lease recovery failed", tenantapp.MonitoringWorkerErrorFields(err)...)
|
|
}
|
|
return
|
|
}
|
|
|
|
if err := tx.Commit(ctx); err != nil {
|
|
if w.logger != nil {
|
|
w.logger.Warn("monitoring lease recovery commit failed", zap.Error(err))
|
|
}
|
|
return
|
|
}
|
|
|
|
if expiredCount > 0 && w.logger != nil {
|
|
w.logger.Info("monitoring lease recovery completed", zap.Int64("expired_task_count", expiredCount))
|
|
}
|
|
}
|