initial commit

pkg/scheduler_unit.go

@@ -0,0 +1,66 @@
+package gocron_server
+
+import (
+	"os/exec"
+	"strings"
+)
+
+type UnitExecutable interface {
+	Call() (string, error)
+}
+
+type UnitExecCmd struct {
+	name string
+	args []string
+}
+
+func (ue *UnitExecCmd) Call() (string, error) {
+	cmd := exec.Command(ue.name, ue.args...)
+	out, err := cmd.Output()
+	strout := string(out[:])
+
+	if err != nil {
+		return strout, err
+	}
+
+	return strout, nil
+}
+
+func NewUnitExecCmd(command string) *UnitExecCmd {
+	cmdFrags := strings.Split(command, " ")
+	return &UnitExecCmd{name: cmdFrags[0], args: cmdFrags[1:]}
+}
+
+type UnitExecFn struct {
+	fn JobFunc
+}
+
+func (ue *UnitExecFn) Call() (string, error) {
+	return ue.fn()
+}
+
+func NewUnitExecFn(fn JobFunc) *UnitExecFn {
+	return &UnitExecFn{fn}
+}
+
+type SchedulerUnit struct {
+	Name string
+	Exec UnitExecutable
+	Cron string
+}
+
+func NewSchedulerUnit(name, cron string, exec UnitExecutable) *SchedulerUnit {
+	return &SchedulerUnit{
+		Name: name,
+		Exec: exec,
+		Cron: cron,
+	}
+}
+
+func NewManualUnit(name string, exec UnitExecutable) *SchedulerUnit {
+	return &SchedulerUnit{
+		Name: name,
+		Exec: exec,
+		Cron: "",
+	}
+}