54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
// Copyright (C) 2019 Nicola Murino
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published
|
|
// by the Free Software Foundation, version 3.
|
|
//
|
|
// This program 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 Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package common
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/robfig/cron/v3"
|
|
|
|
"github.com/drakkan/sftpgo/v2/internal/dataprovider"
|
|
"github.com/drakkan/sftpgo/v2/internal/logger"
|
|
"github.com/drakkan/sftpgo/v2/internal/util"
|
|
)
|
|
|
|
var (
|
|
eventScheduler *cron.Cron
|
|
)
|
|
|
|
func stopEventScheduler() {
|
|
if eventScheduler != nil {
|
|
eventScheduler.Stop()
|
|
eventScheduler = nil
|
|
}
|
|
}
|
|
|
|
func startEventScheduler() {
|
|
stopEventScheduler()
|
|
|
|
options := []cron.Option{
|
|
cron.WithLogger(cron.DiscardLogger),
|
|
}
|
|
if !dataprovider.UseLocalTime() {
|
|
eventManagerLog(logger.LevelDebug, "use UTC time for the scheduler")
|
|
options = append(options, cron.WithLocation(time.UTC))
|
|
}
|
|
|
|
eventScheduler = cron.New(options...)
|
|
eventManager.loadRules()
|
|
_, err := eventScheduler.AddFunc("@every 10m", eventManager.loadRules)
|
|
util.PanicOnError(err)
|
|
eventScheduler.Start()
|
|
}
|