From 12e0ae98b66a323b6f76035f6aae13b52c56c8c1 Mon Sep 17 00:00:00 2001
From: Toni Uhlig <matzeton@googlemail.com>
Date: Thu, 27 May 2021 15:03:38 +0200
Subject: Added realtime web based graph example using Plotly/Dash.

Signed-off-by: Toni Uhlig <matzeton@googlemail.com>
---
 examples/README.md                                 |   4 +
 examples/py-flow-dashboard/flow-dash.py            | 113 +++++++++++++++++++++
 examples/py-flow-dashboard/requirements.txt        |   1 +
 .../py-flow-muliprocess/py-flow-multiprocess.py    |   0
 4 files changed, 118 insertions(+)
 create mode 100755 examples/py-flow-dashboard/flow-dash.py
 create mode 100644 examples/py-flow-dashboard/requirements.txt
 mode change 100644 => 100755 examples/py-flow-muliprocess/py-flow-multiprocess.py

(limited to 'examples')

diff --git a/examples/README.md b/examples/README.md
index 22ff8874b..95804aabb 100644
--- a/examples/README.md
+++ b/examples/README.md
@@ -25,6 +25,10 @@ A discontinued tty UI nDPId dashboard. I've figured out that Go + UI is a bad id
 
 Prints prettyfied information about flow events.
 
+## py-flow-dash
+
+A realtime web based graph using Plotly/Dash.
+
 ## py-flow-multiprocess
 
 Simple Python Multiprocess example spawning two worker processes, one connecting to nDPIsrvd and one printing flow id's to STDOUT.
diff --git a/examples/py-flow-dashboard/flow-dash.py b/examples/py-flow-dashboard/flow-dash.py
new file mode 100755
index 000000000..e94048e13
--- /dev/null
+++ b/examples/py-flow-dashboard/flow-dash.py
@@ -0,0 +1,113 @@
+#!/usr/bin/env python3
+
+from collections import deque
+import dash
+from dash.dependencies import Output, Input
+import dash_core_components as dcc
+import dash_html_components as html
+import multiprocessing
+import os
+import plotly
+import plotly.graph_objs as go
+import random
+import sys
+import time
+
+sys.path.append(os.path.dirname(sys.argv[0]) + '/../share/nDPId')
+sys.path.append(os.path.dirname(sys.argv[0]) + '/../usr/share/nDPId')
+try:
+    import nDPIsrvd
+    from nDPIsrvd import nDPIsrvdSocket, TermColor
+except ImportError:
+    sys.path.append(os.path.dirname(sys.argv[0]) + '/../../dependencies')
+    import nDPIsrvd
+    from nDPIsrvd import nDPIsrvdSocket, TermColor
+
+mgr = multiprocessing.Manager()
+
+global shared_flow_dict
+shared_flow_dict = mgr.dict()
+
+X = deque(maxlen = 20)
+X.append(1)
+  
+Y = deque(maxlen = 20)
+Y.append(1)
+  
+app = dash.Dash(__name__)
+  
+app.layout = html.Div(
+    [
+        dcc.Graph(id = 'live-graph', animate = True),
+        dcc.Interval(
+            id = 'graph-update',
+            interval = 1000,
+            n_intervals = 0
+        ),
+    ]
+)
+
+
+@app.callback(
+    Output('live-graph', 'figure'),
+    [ Input('graph-update', 'n_intervals') ]
+)
+def update_graph_scatter(n):
+    X.append(X[-1]+1)
+    Y.append(len(shared_flow_dict))
+  
+    data = plotly.graph_objs.Scatter(
+            x=list(X),
+            y=list(Y),
+            name='Scatter',
+            mode= 'lines+markers'
+    )
+  
+    return {'data': [data],
+            'layout' : go.Layout(xaxis=dict(range=[min(X),max(X)]),yaxis = dict(range = [min(Y),max(Y)]),)}
+
+def web_worker():
+    app.run_server()
+
+    import time
+    while True:
+        s = str()
+        for key in shared_flow_dict.keys():
+            s += '{}, '.format(str(key))
+        time.sleep(1)
+
+def nDPIsrvd_worker_onJsonLineRecvd(json_dict, current_flow, global_user_data):
+    if not 'flow_event_name' in json_dict:
+        return True
+
+    if json_dict['flow_event_name'] == 'new':
+        shared_flow_dict[json_dict['flow_id']] = current_flow
+    elif json_dict['flow_event_name'] == 'idle' or \
+         json_dict['flow_event_name'] == 'end':
+        if json_dict['flow_id'] in shared_flow_dict:
+            del shared_flow_dict[json_dict['flow_id']]
+
+    return True
+
+def nDPIsrvd_worker(address, nDPIsrvd_global_user_data):
+    sys.stderr.write('Recv buffer size: {}\n'.format(nDPIsrvd.NETWORK_BUFFER_MAX_SIZE))
+    sys.stderr.write('Connecting to {} ..\n'.format(address[0]+':'+str(address[1]) if type(address) is tuple else address))
+
+    nsock = nDPIsrvdSocket()
+    nsock.connect(address)
+    nsock.loop(nDPIsrvd_worker_onJsonLineRecvd, nDPIsrvd_global_user_data)
+
+if __name__ == '__main__':
+    argparser = nDPIsrvd.defaultArgumentParser()
+    args = argparser.parse_args()
+    address = nDPIsrvd.validateAddress(args)
+
+    nDPIsrvd_job = multiprocessing.Process(target = nDPIsrvd_worker, args = (address, None))
+    nDPIsrvd_job.start()
+
+    web_job = multiprocessing.Process(target = web_worker, args = ())
+    web_job.start()
+
+    nDPIsrvd_job.join()
+    web_job.terminate()
+    web_job.join()
diff --git a/examples/py-flow-dashboard/requirements.txt b/examples/py-flow-dashboard/requirements.txt
new file mode 100644
index 000000000..a2544f7ec
--- /dev/null
+++ b/examples/py-flow-dashboard/requirements.txt
@@ -0,0 +1 @@
+dash
diff --git a/examples/py-flow-muliprocess/py-flow-multiprocess.py b/examples/py-flow-muliprocess/py-flow-multiprocess.py
old mode 100644
new mode 100755
-- 
cgit v1.2.3