utils.go

package taskmanager

import (
	"errors"
	"fmt"
	"github.com/go-co-op/gocron/v2"
	"strings"
	"time"
)

// parseJobDefinition parses gocron.JobDefinition from a string value.
func parseJobDefinition(value string) (gocron.JobDefinition, error) {
	if value == DefinitionRunOnce {
		return gocron.OneTimeJob(gocron.OneTimeJobStartImmediately()), nil
	}

	if duration, err := time.ParseDuration(value); err == nil {
		return gocron.DurationJob(duration), nil
	}

	if valueParts := strings.Split(value, ":"); len(valueParts) == 2 {
		definition := fmt.Sprintf("%s %s * * *",
			strings.TrimSpace(valueParts[0]),
			strings.TrimSpace(valueParts[1]))
		return gocron.CronJob(definition, false), nil
	}

	switch len(strings.Split(value, " ")) {
	case 5:
		return gocron.CronJob(value, false), nil
	case 6:
		return gocron.CronJob(value, true), nil
	}

	return nil, errors.New("invalid task definition")
}