[local] Improve container shutdown for local experiments

Author: mcopik

scripts/clean.sh

@@ -10,10 +10,12 @@ pkill -9 -f proc_analyzer.py
 # stop all docker containers
 # TODO: create common ancestor for all runtimes
 echo "Stopping Docker containers..."
-active_containers=$(docker ps --filter ancestor=sebs-local-python -q)
-if [[ -n "${active_containers}" ]]; then
-  docker stop -t0 ${active_containers} > /dev/null
-fi
+for name in "nodejs.13.6" "python.3.6"; do
+  active_containers=$(docker ps --filter ancestor=sebs.run.local.${name} -q)
+  if [[ -n "${active_containers}" ]]; then
+    docker stop -t0 ${active_containers} > /dev/null
+  fi
+done
 echo "Stopping minio storage containers..."
 active_containers=$(docker ps --filter ancestor=minio/minio -q)
 if [[ -n "${active_containers}" ]]; then

scripts/run_experiments.py

@@ -297,6 +297,11 @@ def run_experiment_disk_io(input_config):
                     help='Cache directory')
 parser.add_argument('--update', action='store_true', default=False,
                     help='Update function code in cache and deployment.')
+parser.add_argument('--shutdown-containers', action='store_true',
+                    help='Shutdown containers after experiments.')
+parser.add_argument('--no-shutdown-containers', action='store_false',
+                    help='Shutdown containers after experiments.')
+parser.set_defaults(shutdown_containers=True)
 parser.add_argument('--verbose', action='store', default=False, type=bool,
                     help='Verbose output')
 
@@ -703,7 +708,8 @@ def create_function(self, code_package: CodePackage, experiment_config :dict):
                 json.dump(docker_stats, out_f, indent=2)
 
             # 10. Kill docker instance
-            container.stop()
+            #if args.shutdown_containers:
+            #    container.stop()
 
             # 11. Find experiment JSONs and include in summary
             result_path = os.path.join(dest_dir, 'results')