scheduler.go

package main

import (
	"fmt"
	"os"
	"time"

	"github.com/doomsday-project/doomsday/client/doomsday"
	"github.com/olekukonko/tablewriter"
	"github.com/starkandwayne/goutils/ansi"
)

type schedulerCmd struct{}

func (*schedulerCmd) Run() error {
	state, err := client.GetSchedulerState()
	if err != nil {
		return err
	}

	header := tablewriter.NewWriter(os.Stdout)
	header.SetHeader([]string{"RUNNING"})
	header.SetHeaderLine(false)
	header.Render()

	printSchedTaskList(state.Running)

	header = tablewriter.NewWriter(os.Stdout)
	header.SetHeader([]string{"PENDING"})
	header.SetHeaderLine(false)
	header.Render()

	printSchedTaskList(state.Pending)
	return nil
}

func printSchedTaskList(tasks []doomsday.GetSchedulerTask) {
	fmt.Printf("\n")
	table := tablewriter.NewWriter(os.Stdout)
	table.SetBorder(false)
	table.SetRowLine(true)
	table.SetAutoWrapText(false)
	table.SetReflowDuringAutoWrap(false)
	table.SetHeader([]string{"At", "Backend", "Kind", "Reason", "Ready"})
	table.SetAlignment(tablewriter.ALIGN_RIGHT)

	readyStr := ansi.Sprintf("@G{YES}")
	notReadyStr := ansi.Sprintf("@R{NO}")
	now := time.Now()
	for _, task := range tasks {
		timeUntilStr := time.Unix(task.At, 0).Sub(now).Truncate(100 * time.Millisecond).String()
		readyOutStr := notReadyStr
		if task.Ready {
			readyOutStr = readyStr
		}
		table.Append([]string{
			timeUntilStr,
			task.Backend,
			task.Kind,
			task.Reason,
			readyOutStr,
		})
	}
	table.Render()
	fmt.Printf("\n")
}