cmd/stepd: Be able to trim multiple things

This commit is contained in:
Sasha Koshka 2024-12-12 18:11:47 -05:00
parent df7c49247b
commit 60c52ee4ce

View File

@ -225,7 +225,9 @@ func main () {
// set up the trimming routine
trimmer := trimmerRoutine {
Trimmer: &handler,
Trimmers: []step.Trimmer {
&handler,
},
}
// set up the routine manager
@ -266,7 +268,7 @@ func (this *httpServerRoutine) Run (ctx context.Context) error {
}
type trimmerRoutine struct {
Trimmer step.Trimmer
Trimmers []step.Trimmer
}
func (this *trimmerRoutine) Run (ctx context.Context) error {
@ -275,7 +277,9 @@ func (this *trimmerRoutine) Run (ctx context.Context) error {
for {
select {
case <- ticker.C:
this.Trimmer.Trim()
for _, trimmer := range this.Trimmers {
trimmer.Trim()
}
case <- ctx.Done():
return ctx.Err()
}