adding flowcept and IO for swift/t

Author: rafaelfsilva

wfcommons/wfbench/translator/swift_t.py

@@ -76,8 +76,7 @@ def translate(self, output_folder: pathlib.Path) -> None:
         in_count = 0
         self.output_folder = output_folder
         self.cpu_benchmark = output_folder.joinpath("./bin/cpu-benchmark").absolute()
-        self.script = f"string fs = sprintf(flowcept_start, \"{self.workflow.workflow_id}\");\npython_persist(fs);\n\n" if self.workflow.workflow_id else ""
-        self.script += "string root_in_files[];\n"
+        self.script = "string root_in_files[];\n"
 
         for task_name in self.root_task_names:
             task = self.tasks[task_name]
@@ -100,10 +99,6 @@ def translate(self, output_folder: pathlib.Path) -> None:
         for category in self.categories_list:
             self._add_tasks(category)
 
-        # flowcept stop
-        if self.workflow.workflow_id:
-            self.script += "string fss = sprintf(flowcept_stop);\npython_persist(fss);"
-
         run_workflow_code = self._merge_codelines("templates/swift_t_templates/workflow.swift", self.script)
 
         # write benchmark files

wfcommons/wfbench/translator/templates/swift_t_templates/workflow.swift

@@ -4,25 +4,6 @@ import python;
 import string;
 import unix;
 
-string flowcept_start = 
-"""
-workflow_id = "%s"
-from flowcept import Flowcept, FlowceptTask
-flowcept_agent = Flowcept(workflow_id=workflow_id,
-                        bundle_exec_id=workflow_id,
-                        start_persistence=False, save_workflow=False)
-flowcept_agent.start()
-flowcept_task = FlowceptTask(workflow_id=workflow_id, used={
-    "workflow_id": workflow_id
-})
-""";
-
-string flowcept_stop = 
-"""
-flowcept_task.end()
-flowcept_agent.stop()
-""";
-
 string command = 
 """
 import logging
@@ -51,6 +32,17 @@ output_data = {"%s": int(%i)}
 dep = %i
 workflow_id = "%s"
 
+if 'workflow_id':
+    loggin.info("Running with Flowcept.")
+    from flowcept import Flowcept, FlowceptTask
+    fc = Flowcept(workflow_id=args.workflow_id,
+                bundle_exec_id=args.workflow_id,
+                start_persistence=False, save_workflow=False)
+    fc.start()
+    fc_task = FlowceptTask(workflow_id=args.workflow_id, used={
+      'workflow_id': workflow_id
+    })
+
 logging.info(f"Starting {task_name} Benchmark on {socket.gethostname()}")
 
 procs = []
@@ -179,6 +171,10 @@ if cpu_work > 0:
     logging.debug("Completed CPU and Memory Benchmarks!")
     
 logging.info(f"Benchmark {task_name} completed!")
+
+if 'workflow_id':
+    fc_task.end()
+    fc.stop()
 """;
 
 # Generated code goes here