X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/5ef89a24d5fa8bc6926a433e22360a09fdb3154d..b1ffc878a5ea6ba083b8fbc8f20c15c7a6f1e1ec:/lib/crunchrun/executor.go

diff --git a/lib/crunchrun/executor.go b/lib/crunchrun/executor.go
index 787edda010..308b05cdeb 100644
--- a/lib/crunchrun/executor.go
+++ b/lib/crunchrun/executor.go
@@ -4,10 +4,10 @@
 package crunchrun
 
 import (
+	"context"
 	"io"
 
 	"git.arvados.org/arvados.git/sdk/go/arvados"
-	"golang.org/x/net/context"
 )
 
 type bindmount struct {
@@ -16,19 +16,20 @@ type bindmount struct {
 }
 
 type containerSpec struct {
-	Image         string
-	VCPUs         int
-	RAM           int64
-	WorkingDir    string
-	Env           map[string]string
-	BindMounts    map[string]bindmount
-	Command       []string
-	EnableNetwork bool
-	NetworkMode   string // docker network mode, normally "default"
-	CgroupParent  string
-	Stdin         io.Reader
-	Stdout        io.Writer
-	Stderr        io.Writer
+	Image           string
+	VCPUs           int
+	RAM             int64
+	WorkingDir      string
+	Env             map[string]string
+	BindMounts      map[string]bindmount
+	Command         []string
+	EnableNetwork   bool
+	CUDADeviceCount int
+	NetworkMode     string // docker network mode, normally "default"
+	CgroupParent    string
+	Stdin           io.Reader
+	Stdout          io.Writer
+	Stderr          io.Writer
 }
 
 // containerExecutor is an interface to a container runtime
@@ -36,8 +37,8 @@ type containerSpec struct {
 type containerExecutor interface {
 	// ImageLoad loads the image from the given tarball such that
 	// it can be used to create/start a container.
-	LoadImage(imageID string, container arvados.Container, keepMount string,
-		containerClient *arvados.Client, keepClient IKeepClient) error
+	LoadImage(imageID string, imageTarballPath string, container arvados.Container, keepMount string,
+		containerClient *arvados.Client) error
 
 	// Wait for the container process to finish, and return its
 	// exit code. If applicable, also remove the stopped container
@@ -50,12 +51,18 @@ type containerExecutor interface {
 	// Start the container
 	Start() error
 
-	// CID the container will belong to
-	CgroupID() string
+	// Process ID of a process in the container.  Return 0 if
+	// container is finished or no process has started yet.
+	Pid() int
 
 	// Stop the container immediately
 	Stop() error
 
 	// Release resources (temp dirs, stopped containers)
 	Close()
+
+	// Name and version of runtime engine ("docker 20.10.16", "singularity-ce version 3.9.9")
+	Runtime() string
+
+	GatewayTarget
 }