schedule/task.go

// Package schedule
// file contains all struct and interface define.
package schedule

import (
	"context"
	"log"
	"time"
)

// Task task interface for scheduler task
type Task interface {
	Run(ctx context.Context)
}

// Logger logger interface for scheduler logger
type Logger interface {
	Error(msg string, e any)
	Debugf(msg string, n int32)
	Debug(msg string)
}

// TaskFunc the function of task
type TaskFunc func(ctx context.Context)

// WhenFunc the function define of task constraint
type WhenFunc func(ctx context.Context) bool

type DefaultTask struct {
	fn TaskFunc
}

func NewDefaultTask(fn TaskFunc) *DefaultTask {
	return &DefaultTask{fn: fn}
}

func (d *DefaultTask) Run(ctx context.Context) {
	d.fn(ctx)
}

type NextTick struct {
	Year   int
	Month  int
	Day    int
	Hour   int
	Minute int
	Omit   bool
}

type Limit struct {
	DaysOfWeek []time.Weekday
	StartTime  string
	EndTime    string
	IsBetween  bool
	When       WhenFunc
}

type DefaultLogger struct {
}

func (d *DefaultLogger) Error(msg string, r any) {
	log.Println(msg, r)
}

func (d *DefaultLogger) Debug(msg string) {
	log.Println(msg)
}

func (d *DefaultLogger) Debugf(msg string, i int32) {
	log.Printf(msg, i)
}