Support for command flag in services

vertex-center · Oct 7, 2023 · 2e446bb · 2e446bb
1 parent 0d69f8a
commit 2e446bb
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 0 deletions.
diff --git a/adapter/docker_cli.go b/adapter/docker_cli.go
@@ -57,6 +57,7 @@ func (a DockerCliAdapter) CreateContainer(options types.CreateContainerOptions)
 		Tty:          true,
 		AttachStdout: true,
 		AttachStderr: true,
+		Cmd:          options.Cmd,
 	}
 
 	hostConfig := container.HostConfig{

diff --git a/adapter/runner_docker.go b/adapter/runner_docker.go
@@ -210,6 +210,11 @@ func (a RunnerDockerAdapter) Start(inst *types.Instance, setStatus func(status s
 				options.Sysctls = *service.Methods.Docker.Sysctls
 			}
 
+			// cmd
+			if service.Methods.Docker.Cmd != nil {
+				options.Cmd = strings.Split(*service.Methods.Docker.Cmd, " ")
+			}
+
 			if service.Methods.Docker.Dockerfile != nil {
 				options.ImageName = inst.DockerImageVertexName()
 				id, err = a.createContainer(options)

diff --git a/types/docker.go b/types/docker.go
@@ -29,6 +29,7 @@ type CreateContainerOptions struct {
 	Env           []string          `json:"env,omitempty"`
 	CapAdd        []string          `json:"cap_add,omitempty"`
 	Sysctls       map[string]string `json:"sysctls,omitempty"`
+	Cmd           []string          `json:"cmd,omitempty"`
 }
 
 type BuildImageOptions struct {

diff --git a/types/service.go b/types/service.go
@@ -209,6 +209,9 @@ type ServiceMethodDocker struct {
 
 	// Sysctls allows to modify kernel parameters.
 	Sysctls *map[string]string `yaml:"sysctls,omitempty" json:"sysctls,omitempty"`
+
+	// Cmd is the command to run in the container.
+	Cmd *string `yaml:"command,omitempty" json:"command,omitempty"`
 }
 
 type ServiceMethods struct {