From a70f822e759e6df42ba958873a0d4182767552da Mon Sep 17 00:00:00 2001
From: Noor <noormohammedb@protonmail.com>
Date: Mon, 14 Apr 2025 17:47:02 +0530
Subject: [PATCH] Fix: add restart policy to docker run command

---
 src/container.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/container.rs b/src/container.rs
index 5c4ee6f..7c2fa83 100644
--- a/src/container.rs
+++ b/src/container.rs
@@ -28,7 +28,7 @@ pub fn deploy_enclave(
     // --storage-opt size={disk_mb}m
 
     let docker_deploy_str = format!(
-        r#"docker run -d --name {container_name_uuid} --memory={memory_mb}m --cpus={vcpu} \
+        r#"docker run -d --restart unless-stopped --name {container_name_uuid} --memory={memory_mb}m --cpus={vcpu} \
         -v {enclave_path}:/enclave_package --device /dev/sgx/enclave --device /dev/sgx/provision \
         {port_maping_string} noormohammedb/occlum-enclave:v1 {hratls_pubkey}"#
     );