59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package scheduler
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
tenantapp "github.com/geo-platform/tenant-api/internal/tenant/app"
|
|
)
|
|
|
|
type KnowledgeDeletedCleanupWorker struct {
|
|
service *tenantapp.KnowledgeService
|
|
interval time.Duration
|
|
}
|
|
|
|
func NewKnowledgeDeletedCleanupWorker(service *tenantapp.KnowledgeService) *KnowledgeDeletedCleanupWorker {
|
|
return &KnowledgeDeletedCleanupWorker{
|
|
service: service,
|
|
interval: 15 * time.Second,
|
|
}
|
|
}
|
|
|
|
func (w *KnowledgeDeletedCleanupWorker) Start(ctx context.Context) {
|
|
go w.Run(ctx)
|
|
}
|
|
|
|
func (w *KnowledgeDeletedCleanupWorker) Run(ctx context.Context) {
|
|
if w == nil || w.service == nil {
|
|
return
|
|
}
|
|
w.run(ctx)
|
|
}
|
|
|
|
func (w *KnowledgeDeletedCleanupWorker) run(ctx context.Context) {
|
|
w.service.RunDeletedCleanupSweep()
|
|
|
|
ticker := time.NewTicker(w.interval)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-ticker.C:
|
|
w.service.RunDeletedCleanupSweep()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (w *KnowledgeDeletedCleanupWorker) RunOnce(ctx context.Context) (map[string]any, error) {
|
|
if w == nil || w.service == nil {
|
|
return map[string]any{"skipped": true, "reason": "worker_not_ready"}, nil
|
|
}
|
|
startedAt := time.Now()
|
|
w.service.RunDeletedCleanupSweep()
|
|
return map[string]any{
|
|
"duration_ms": time.Since(startedAt).Milliseconds(),
|
|
}, nil
|
|
}
|