66 lines
1.8 KiB
Go
66 lines
1.8 KiB
Go
package worker
|
|
|
|
/* ***** BEGIN GPL LICENSE BLOCK *****
|
|
*
|
|
* Original Code Copyright (C) 2022 Blender Foundation.
|
|
*
|
|
* This file is part of Flamenco.
|
|
*
|
|
* Flamenco is free software: you can redistribute it and/or modify it under
|
|
* the terms of the GNU General Public License as published by the Free Software
|
|
* Foundation, either version 3 of the License, or (at your option) any later
|
|
* version.
|
|
*
|
|
* Flamenco is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
|
* A PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with
|
|
* Flamenco. If not, see <https://www.gnu.org/licenses/>.
|
|
*
|
|
* ***** END GPL LICENSE BLOCK ***** */
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"gitlab.com/blender/flamenco-ng-poc/pkg/api"
|
|
)
|
|
|
|
const durationSleepCheck = 3 * time.Second
|
|
|
|
func (w *Worker) gotoStateAsleep(ctx context.Context) {
|
|
w.stateMutex.Lock()
|
|
defer w.stateMutex.Unlock()
|
|
|
|
w.state = api.WorkerStatusAsleep
|
|
w.doneWg.Add(2)
|
|
w.ackStateChange(ctx, w.state)
|
|
go w.runStateAsleep(ctx)
|
|
}
|
|
|
|
func (w *Worker) runStateAsleep(ctx context.Context) {
|
|
defer w.doneWg.Done()
|
|
logger := log.With().Str("status", string(w.state)).Logger()
|
|
logger.Info().Msg("sleeping")
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
logger.Debug().Msg("asleep state interrupted by context cancellation")
|
|
return
|
|
case <-w.doneChan:
|
|
logger.Debug().Msg("asleep state interrupted by shutdown")
|
|
return
|
|
case <-time.After(durationSleepCheck):
|
|
newStatus := w.queryManagerForStateChange(ctx)
|
|
if newStatus != nil {
|
|
logger.Debug().Str("newStatus", string(*newStatus)).Msg("asleep state interrupted by state change")
|
|
w.changeState(ctx, *newStatus)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|