mirror of
https://github.com/hrfee/jfa-go.git
synced 2024-09-19 19:00:11 +00:00
Harvey Tindall
1fa340f096
"import_existing" option in settings enables an every 5-minute daemon which loops through users and imports them to Jellyseerr and copies contact info, if necessary. Also sets new API client flag AutoImportUsers, which decides whether to automatically import non-existent users in it's various methods. also cleaned up the various daemons in the software, most now using the GenericDaemon struct and just providing a new constructor. broken page loop in jellyseerr client also fixed.
66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package main
|
|
|
|
import "time"
|
|
|
|
// https://bbengfort.github.io/snippets/2016/06/26/background-work-goroutines-timer.html THANKS
|
|
|
|
type GenericDaemon struct {
|
|
Stopped bool
|
|
ShutdownChannel chan string
|
|
Interval time.Duration
|
|
period time.Duration
|
|
jobs []func(app *appContext)
|
|
app *appContext
|
|
name string
|
|
}
|
|
|
|
func (d *GenericDaemon) appendJobs(jobs ...func(app *appContext)) {
|
|
d.jobs = append(d.jobs, jobs...)
|
|
}
|
|
|
|
// NewGenericDaemon returns a daemon which can be given jobs that utilize appContext.
|
|
func NewGenericDaemon(interval time.Duration, app *appContext, jobs ...func(app *appContext)) *GenericDaemon {
|
|
d := GenericDaemon{
|
|
Stopped: false,
|
|
ShutdownChannel: make(chan string),
|
|
Interval: interval,
|
|
period: interval,
|
|
app: app,
|
|
name: "Generic Daemon",
|
|
}
|
|
d.jobs = jobs
|
|
return &d
|
|
|
|
}
|
|
|
|
func (d *GenericDaemon) Name(name string) { d.name = name }
|
|
|
|
func (d *GenericDaemon) run() {
|
|
d.app.info.Printf("%s started", d.name)
|
|
for {
|
|
select {
|
|
case <-d.ShutdownChannel:
|
|
d.ShutdownChannel <- "Down"
|
|
return
|
|
case <-time.After(d.period):
|
|
break
|
|
}
|
|
started := time.Now()
|
|
|
|
for _, job := range d.jobs {
|
|
job(d.app)
|
|
}
|
|
|
|
finished := time.Now()
|
|
duration := finished.Sub(started)
|
|
d.period = d.Interval - duration
|
|
}
|
|
}
|
|
|
|
func (d *GenericDaemon) Shutdown() {
|
|
d.Stopped = true
|
|
d.ShutdownChannel <- "Down"
|
|
<-d.ShutdownChannel
|
|
close(d.ShutdownChannel)
|
|
}
|