slurmcontrol

type SlurmControlInterface interface {
	// GetNodeNames returns the list Slurm nodes by name.
	GetNodeNames(ctx context.Context) ([]string, error)
	// NodeExists returns true if the Slurm node exists, false if not found.
	NodeExists(ctx context.Context, node *corev1.Node) (bool, error)
	// MakeNodeDrain handles adding the DRAIN state to the Slurm node.
	MakeNodeDrain(ctx context.Context, node *corev1.Node, reason string) error
	// MakeNodeUndrain handles removing the DRAIN state from the Slurm node.
	MakeNodeUndrain(ctx context.Context, node *corev1.Node, reason string) error
	// IsNodeDrain checks if the slurm node has the DRAIN state.
	IsNodeDrain(ctx context.Context, node *corev1.Node) (bool, error)
	// IsNodeDrained checks if the slurm node is DRAINED and eligible for removal.
	IsNodeDrained(ctx context.Context, node *corev1.Node) (bool, error)
	// AddNode registers a Kubernetes node in Slurm with the correct CPUs and memory.
	AddNode(ctx context.Context, node *corev1.Node, nodeInfo *nodeinfo.NodeInfo) error
	// NodeNeedsRecreate returns true if the Slurm node exists and its cpu, memory, or gres
	// differ from the desired values (from the Kubernetes node and nodeInfo). Such a node
	// must be drained, removed, and re-added to apply the change.
	NodeNeedsRecreate(ctx context.Context, node *corev1.Node, nodeInfo *nodeinfo.NodeInfo) (bool, error)
	// RemoveNode removes a Kubernetes node from Slurm.
	RemoveNode(ctx context.Context, node *corev1.Node) error
}