diff --git a/backend/src/utilities/database.py b/backend/src/utilities/database.py index 7831ecb..77df52f 100644 --- a/backend/src/utilities/database.py +++ b/backend/src/utilities/database.py @@ -163,6 +163,7 @@ class DatabasePool: correlation_key, packet_id, packet_uid, + flow_id, correlation_source, skb_mark, capture_iface, @@ -201,8 +202,8 @@ class DatabasePool: raw ) VALUES( $1,$2,$3,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18,$19,$20, - $21,$22,$23,$24,$25,$26,$27,$28,$29,$30,$31,$32,$33,$34,$35,$36,$37::jsonb, - $38::jsonb,$39::jsonb,$40 + $21,$22,$23,$24,$25,$26,$27,$28,$29,$30,$31,$32,$33,$34,$35,$36,$37,$38::jsonb, + $39::jsonb,$40::jsonb,$41 ) ON CONFLICT (correlation_key) DO UPDATE SET updated_at = NOW(), @@ -213,6 +214,7 @@ class DatabasePool: END, packet_id = COALESCE(EXCLUDED.packet_id, packets.packet_id), packet_uid = COALESCE(EXCLUDED.packet_uid, packets.packet_uid), + flow_id = COALESCE(EXCLUDED.flow_id, packets.flow_id), correlation_source = COALESCE(EXCLUDED.correlation_source, packets.correlation_source), skb_mark = COALESCE(EXCLUDED.skb_mark, packets.skb_mark), capture_iface = COALESCE(EXCLUDED.capture_iface, packets.capture_iface), @@ -263,6 +265,7 @@ class DatabasePool: pkt_info["correlation_key"], pkt_info.get("packet_id"), pkt_info.get("packet_uid"), + pkt_info.get("flow_id"), pkt_info.get("correlation_source"), pkt_info.get("skb_mark"), pkt_info.get("capture_iface"),