add active flow parameter

1 files changed, 2 insertions, 1 deletions

diff --git a/app.py b/app.py
index 67f9d2c..c545d9b 100644
--- a/app.py
+++ b/app.py
@@ -84,6 +84,7 @@ def run_traffic_env():
             yaml_name = json_data.get("yaml_name") + ".yaml"
             m = json_data.get("m")
             d = json_data.get("d")
+            active_flows = json_data.get("active_flows")
             clients_start_ip = json_data.get("clients_start_ip")
             clients_end_ip = json_data.get("clients_end_ip")
             servers_start_ip = json_data.get("servers_start_ip")
@@ -93,7 +94,7 @@ def run_traffic_env():
                 "traffic_playback_workpath": "/opt/traffic_replay"
             }
             tr = trex_replay.TrafficReplay(path_dict)
-            result = tr.trex_playback(yaml_name, pcap_name, m, d, clients_start_ip, clients_end_ip, servers_start_ip, servers_end_ip)
+            result = tr.trex_playback(yaml_name, pcap_name, active_flows, m, d, clients_start_ip, clients_end_ip, servers_start_ip, servers_end_ip)
         # if flag == False and (type == "curl" or type == "wget" or type == "nslookup"):
         #     print("Execute {} operation.".format(type))
         #     command = json_data.get("command")