Add support for autostart of Docker containers.

golemfactory · Jul 12, 2019 · 40ddf41 · 40ddf41
1 parent 15978f6
commit 40ddf41
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/gu-model/src/dockerman.rs b/gu-model/src/dockerman.rs
@@ -9,6 +9,8 @@ pub struct CreateOptions {
     #[serde(default)]
     #[serde(skip_serializing_if = "Option::is_none")]
     pub net: Option<NetDef>,
+    #[serde(default)] // default is false
+    pub autostart: bool,
 }
 
 impl CreateOptions {

diff --git a/gu-provider/src/dockerman.rs b/gu-provider/src/dockerman.rs
@@ -420,11 +420,14 @@ impl Handler<CreateSession<CreateOptions>> for DockerMan {
                 ActorResponse::r#async(fut::wrap_future(pull_and_create).and_then(
                     move |id, act: &mut DockerMan, _| {
                         if let Some(ref api) = act.docker_api {
-                            let deploy = DockerSession {
+                            let mut deploy = DockerSession {
                                 workspace,
                                 container: api.container(Cow::from(id.clone())),
                                 status: PeerSessionStatus::CREATED,
                             };
+                            if msg.options.autostart {
+                                deploy.do_start();
+                            }
                             act.deploys.insert_deploy(id.clone(), deploy);
                             fut::ok(id)
                         } else {