cluster-dashboard/backend/executor/hostrunner.go

package executor

import (
	"bytes"
	"fmt"
	"os/exec"
	"strings"
)

type HostRunner struct{}

func (HostRunner) SlurmQueue() ([]string, error) {
	panic("todo")
}

func (HostRunner) SlurmJobs() ([]string, error) {
	panic("todo")
}

func (HostRunner) NodeUptime(hostname string) (string, error) {
	panic("todo")
}

func (HostRunner) Temperature(hostname string) (string, error) {
	panic("todo")
}

func (HostRunner) MemoryUsage(hostname string) (string, error) {
	panic("todo")
}

func (HostRunner) StorageUsage(hostname string) (string, error) {
	panic("todo")
}

func (HostRunner) NetworkUpload(hostname string) (string, error) {
	panic("todo")
}

func (HostRunner) NetworkDownload(hostname string) (string, error) {
	panic("todo")
}

//
// Internals
//

// run a shell command and returns its output as string
func run(command string) (string, error) {
	buf := &bytes.Buffer{}

	// ctx := context.Background()
	// ctx, cancel := context.WithTimeout(ctx, 1*time.Second)
	// defer cancel()
	// cmd := exec.CommandContext(ctx, "sh", "-c", command)

	cmd := exec.Command("sh", "-c", command)
	cmd.Stdout = buf

	// BUG: Al momento se questo comando ci mette troppo tutto si blocca, quindi sarebbe meglio
	// usare un context per dare un timeout al comando
	if err := cmd.Run(); err != nil {
		return "", err
	}

	return buf.String(), nil
}

// run a shell command on a given remote via ssh and returns its output as string, does a bit of escaping
func runOnRemote(destination, command string) (string, error) {
	return run(fmt.Sprintf(`ssh %s '%s'`, destination, strings.ReplaceAll(command, `'`, `\'`)))
}