gridscale.cluster.HeadNode.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cluster_3 Show documentation

cluster

The newest version!

package gridscale.cluster

import java.io.ByteArrayInputStream

import gridscale._
import ssh._
import local._

object HeadNode:
  given (using SSH): SSHHeadNode = SSHHeadNode()
  def local = LocalHeadNode()
  def ssh(using SSH) = SSHHeadNode()

sealed trait HeadNode:
  def execute(cmd: String): ExecutionResult
  def write(bytes: Array[Byte], path: String): Unit
  def read(path: String): String
  def rmFile(path: String): Unit

case class SSHHeadNode()(using SSH) extends HeadNode:
  override def execute(cmd: String) = gridscale.ssh.run(cmd)
  override def write(bytes: Array[Byte], path: String) = ssh.writeFile(() ⇒ new ByteArrayInputStream(bytes), path)
  override def read(path: String) = ssh.readFile(path, scala.io.Source.fromInputStream(_).mkString)
  override def rmFile(path: String): Unit = ssh.rmFile(path)

case class LocalHeadNode() extends HeadNode:
  override def execute(cmd: String) = Local.execute(cmd)
  override def write(bytes: Array[Byte], path: String) = Local.writeBytes(bytes, path)
  override def read(path: String) = Local.readFile(path, scala.io.Source.fromInputStream(_).mkString)
  override def rmFile(path: String): Unit = Local.rmFile(path)