remote_execution_node.go

package main

import (
	"fmt"
	"io"

	"golang.org/x/crypto/ssh"

	"github.com/reconquest/hierr-go"
	"github.com/reconquest/runcmd"
)

type remoteExecutionNode struct {
	node    *distributedLockNode
	command runcmd.CmdWorker

	stdin  io.WriteCloser
	stdout io.WriteCloser
	stderr io.WriteCloser

	exitCode int
}

func (node *remoteExecutionNode) wait() error {
	err := node.command.Wait()
	if err != nil {
		_ = node.stdout.Close()
		_ = node.stderr.Close()
		if sshErrors, ok := err.(*ssh.ExitError); ok {
			node.exitCode = sshErrors.Waitmsg.ExitStatus()

			return fmt.Errorf(
				`%s had failed to evaluate command, `+
					`remote command exited with non-zero code: %d`,
				node.node.String(),
				node.exitCode,
			)
		}

		return hierr.Errorf(
			err,
			`%s failed to finish execution, unexpected error`,
			node.node.String(),
		)
	}

	err = node.stdout.Close()
	if err != nil {
		return hierr.Errorf(
			err,
			`%s can't close stdout`,
			node.node.String(),
		)
	}

	err = node.stderr.Close()
	if err != nil {
		return hierr.Errorf(
			err,
			`%s can't close stderr`,
			node.node.String(),
		)
	}

	return nil
}

func (node *remoteExecutionNode) String() string {
	return node.node.String()
}