tasksetting.go

package models

import (
	"context"
	"fmt"
	"os"
	"os/exec"
	"strconv"
	"time"
)

// TaskSetting is the setting data model of A01Task
type TaskSetting struct {
	Version     string            `json:"ver,omitempty"`
	Execution   map[string]string `json:"execution,omitempty"`
	Classifier  map[string]string `json:"classifier,omitempty"`
	Miscellanea map[string]string `json:"misc,omitempty"`
}

// GetIdentifier returns the unique identifier of the task setting
func (setting *TaskSetting) GetIdentifier() string {
	return setting.Classifier["identifier"]
}

// Execute runs the command and returns the execution results
func (setting *TaskSetting) Execute() (result string, duration int, output []byte) {
	shellExec := "/bin/bash"
	if _, err := os.Stat("/bin/bash"); os.IsNotExist(err) {
		shellExec = "/bin/sh"
	}

	timeout := time.Hour * 2
	ctx, cancel := context.WithTimeout(context.Background(), timeout)
	defer cancel()

	execution := []string{"-c", setting.Execution["command"]}
	cmd := exec.CommandContext(ctx, shellExec, execution...)

	begin := time.Now()
	output, err := cmd.CombinedOutput()
	elapsed := time.Since(begin)

	duration = int(elapsed.Seconds())

	if err == nil {
		result = "Passed"
	} else {
		if elapsed >= timeout {
			result = "Timeout"
		} else {
			result = "Failed"
		}
	}

	return
}

// CreateCompletedTask returns an uncommitted Task instance which represents a completed task
func (setting *TaskSetting) CreateCompletedTask(result string, duration int, podName string, runID string) *TaskResult {
	nRunID, _ := strconv.Atoi(runID)

	task := TaskResult{
		Name:          fmt.Sprintf("Test: %s", setting.GetIdentifier()),
		Duration:      duration,
		Result:        result,
		ResultDetails: map[string]interface{}{"agent": podName},
		RunID:         nRunID,
		Settings:      *setting,
		Status:        "Completed",
	}

	return &task
}

// CreateUncompletedTask returns an uncommitted Task instance which represents an incomplete task.
func (setting *TaskSetting) CreateUncompletedTask(podName string, runID string, errorMsg string) *TaskResult {
	nRunID, _ := strconv.Atoi(runID)

	task := TaskResult{
		Name:   fmt.Sprintf("Test: %s", setting.GetIdentifier()),
		Result: "Error",
		ResultDetails: map[string]interface{}{
			"agent": podName,
			"error": errorMsg,
		},
		RunID:    nRunID,
		Settings: *setting,
		Status:   "Error",
	}

	return &task
}