web-dev-qa-db-fra.com

Golang: implémenter un cron / exécuter des tâches à un moment précis

J'ai cherché des exemples sur la façon d'implémenter une fonction qui vous permet d'exécuter des tâches à un certain moment dans Go, mais je n'ai rien trouvé.

J'en ai implémenté un moi-même et je le partage dans les réponses, afin que d'autres personnes puissent avoir une référence pour leur propre implémentation.

crongoscheduled-taskscrontabticker
25
Daniele B

Il s'agit d'une implémentation générale, qui vous permet de définir:

  • période d'intervalle
  • heure à cocher
  • minute pour cocher
  • deuxième à cocher

MISE À JOUR: (la fuite de mémoire a été corrigée)

import (
"fmt"
"time"
)

const INTERVAL_PERIOD time.Duration = 24 * time.Hour

const HOUR_TO_TICK int = 23
const MINUTE_TO_TICK int = 00
const SECOND_TO_TICK int = 03

type jobTicker struct {
    timer *time.Timer
}

func runningRoutine() {
    jobTicker := &jobTicker{}
    jobTicker.updateTimer()
    for {
        <-jobTicker.timer.C
        fmt.Println(time.Now(), "- just ticked")
        jobTicker.updateTimer()
    }
}

func (t *jobTicker) updateTimer() {
    nextTick := time.Date(time.Now().Year(), time.Now().Month(), 
    time.Now().Day(), HOUR_TO_TICK, MINUTE_TO_TICK, SECOND_TO_TICK, 0, time.Local)
    if !nextTick.After(time.Now()) {
        nextTick = nextTick.Add(INTERVAL_PERIOD)
    }
    fmt.Println(nextTick, "- next tick")
    diff := nextTick.Sub(time.Now())
    if t.timer == nil {
        t.timer = time.NewTimer(diff)
    } else {
        t.timer.Reset(diff)
    }
}
27
Daniele B

la réponse fournie par @Daniele B n'est pas assez bonne, comme le dit @Caleb, que l'implémentation fuit la mémoire, car chaque fois que nous créons un nouveau ticker, l'ancien ne sera jamais publié.

donc j'enveloppe le time.timer, et le réinitialiser à chaque fois, un exemple ici:

package main

import (
    "fmt"
    "time"
)

const INTERVAL_PERIOD time.Duration = 24 * time.Hour

const HOUR_TO_TICK int = 23
const MINUTE_TO_TICK int = 21
const SECOND_TO_TICK int = 03

type jobTicker struct {
    t *time.Timer
}

func getNextTickDuration() time.Duration {
    now := time.Now()
    nextTick := time.Date(now.Year(), now.Month(), now.Day(), HOUR_TO_TICK, MINUTE_TO_TICK, SECOND_TO_TICK, 0, time.Local)
    if nextTick.Before(now) {
        nextTick = nextTick.Add(INTERVAL_PERIOD)
    }
    return nextTick.Sub(time.Now())
}

func NewJobTicker() jobTicker {
    fmt.Println("new tick here")
    return jobTicker{time.NewTimer(getNextTickDuration())}
}

func (jt jobTicker) updateJobTicker() {
    fmt.Println("next tick here")
    jt.t.Reset(getNextTickDuration())
}

func main() {
    jt := NewJobTicker()
    for {
        <-jt.t.C
        fmt.Println(time.Now(), "- just ticked")
        jt.updateJobTicker()
    }
}
17
simon_xia

Au cas où quelqu'un tomberait sur cette question à la recherche d'une solution rapide. J'ai trouvé une bibliothèque soignée qui facilite vraiment la planification des travaux.

Lien: https://github.com/jasonlvhit/gocron

L'API est assez simple:

import (
    "fmt"
    "github.com/jasonlvhit/gocron"
)

func task() {
    fmt.Println("Task is being performed.")
}

func main() {
    s := gocron.NewScheduler()
    s.Every(2).Hours().Do(task)
    <- s.Start()
}
13
Emmanuel Ay

J'ai créé un package qui prend en charge la syntaxe crontab si vous le connaissez, par exemple:

ctab := crontab.New()
ctab.AddJob("*/5 * * * *", myFunc)
ctab.AddJob("0 0 * * *", myFunc2)

Lien du package: https://github.com/mileusna/crontab

2
mileusna

Il s'agit d'une autre implémentation générale sans besoin d'une bibliothèque tierce.

Exécutez un func une fois par jour à midi.

  • Période: time.Hour * 24
  • Décalage: time.Hour * 12

Exécutez un func deux fois par jour à 03:40 (00:00 + 03:40) et 15:40 (12:00 + 03:40).

  • Période: time.Hour * 12
  • Décalage: time.Hour * 3 + time.Minute * 40

Mise à jour (2020-01-28):

Changements:

  • context.Context peut être utilisé pour l'annulation, le rend testable.
  • time.Ticker supprime la nécessité de calculer l'heure de la prochaine exécution.
package main

import (
    "context"
    "time"
)

// Schedule calls function `f` with a period `p` offsetted by `o`.
func Schedule(ctx context.Context, p time.Duration, o time.Duration, f func(time.Time)) {
    // Position the first execution
    first := time.Now().Truncate(p).Add(o)
    if first.Before(time.Now()) {
        first = first.Add(p)
    }
    firstC := time.After(first.Sub(time.Now()))

    // Receiving from a nil channel blocks forever
    t := &time.Ticker{C: nil}

    for {
        select {
        case v := <-firstC:
            // The ticker has to be started before f as it can take some time to finish
            t = time.NewTicker(p)
            f(v)
        case v := <-t.C:
            f(v)
        case <-ctx.Done():
            t.Stop()
            return
        }
    }

}

Original:

package main

import (
    "time"
)

// Repeat calls function `f` with a period `d` offsetted by `o`.
func Repeat(d time.Duration, o time.Duration, f func(time.Time)) {
    next := time.Now().Truncate(d).Add(o)
    if next.Before(time.Now()) {
        next = next.Add(d)
    }

    t := time.NewTimer(next.Sub(time.Now()))

    for {
        v := <-t.C
        next = next.Add(d)
        t.Reset(next.Sub(time.Now()))
        f(v)
    }
}
2
Nergal