47 lines
894 B
Go
47 lines
894 B
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"github.com/tomwright/grace"
|
|
"log"
|
|
"time"
|
|
)
|
|
|
|
// NewCleanupRunner returns a runner that can be used cleanup old diagrams.
|
|
func NewCleanupRunner(generator Generator) grace.Runner {
|
|
return &cleanupService{
|
|
generator: generator,
|
|
runEvery: time.Minute * 5,
|
|
cleanupLast: time.Hour,
|
|
}
|
|
}
|
|
|
|
// cleanupService is a runner that is used cleanup old diagrams.
|
|
type cleanupService struct {
|
|
generator Generator
|
|
runEvery time.Duration
|
|
cleanupLast time.Duration
|
|
}
|
|
|
|
// Run starts the cleanup process.
|
|
func (s *cleanupService) Run(ctx context.Context) error {
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return nil
|
|
default:
|
|
}
|
|
|
|
if err := s.generator.CleanUp(s.cleanupLast); err != nil {
|
|
log.Printf("error when cleaning up: %s", err.Error())
|
|
}
|
|
|
|
select {
|
|
case <-time.After(s.runEvery):
|
|
continue
|
|
case <-ctx.Done():
|
|
return nil
|
|
}
|
|
}
|
|
}
|