Merge pull request #687 from luzonghaoa/patch-1

Update replay.py
awslabs · May 10, 2023 · a797c91 · a797c91
2 parents 11b5785 + a335044
commit a797c91
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/src/SimpleReplay/replay.py b/src/SimpleReplay/replay.py
@@ -397,14 +397,12 @@ def save_query_stats(self, starttime, endtime, xid, query_idx):
                 )
 
     def get_tagged_sql(self, query_text, idx, transaction, connection):
-        if g_config.get("source_tag", None):
-            json_tags = {
-                "xid": transaction.xid,
-                "query_idx": idx,
-                "replay_start": g_replay_timestamp.isoformat(),
-                "source": g_config.get('source_tag', 'SimpleReplay'),
-            }
-
+        json_tags = {
+            "xid": transaction.xid,
+            "query_idx": idx,
+            "replay_start": g_replay_timestamp.isoformat(),
+            "source": g_config.get('source_tag', 'SimpleReplay'),
+        }
 
         return "/* {} */ {}".format(json.dumps(json_tags), query_text)