Implemented poller keepalive

controller/app/mod_pollers/controllers.py

@@ -5,7 +5,7 @@
 
 mod_pollers = Blueprint('pollers', __name__, url_prefix='/pollers')
 
-pollers = []
+pollers = {}
 
 
 @mod_pollers.route('/', methods=['GET'])
@@ -16,11 +16,14 @@ def get_pollers():
 @mod_pollers.route('/register', methods=['POST'])
 def register():
     if not request.json:
-        print('hit')
         return render_template('pollers/pollers.html', pollers=pollers)
 
-    poller = Poller(request.json['ip'],
-                    request.json['port'],
-                    request.json['name'])
-    pollers.append(poller)
-    return jsonify({'status': 'poller created'}), 201
+    if request.json['name'] not in pollers:
+        poller = Poller(request.json['ip'],
+                        request.json['port'],
+                        request.json['name'])
+        pollers[request.json['name']] = poller
+        return jsonify({'status': 'poller created'}), 201
+    else:
+        pollers[request.json['name']].keepalive()
+        return jsonify({'status': 'poller keepalive'}), 201

controller/app/templates/pollers/pollers.html

@@ -17,13 +17,13 @@ <h2>All Pollers</h2>
                 <th>Online since</th>
                 <th>Last keepalive</th>
             </tr>
-            {% for poller in pollers %}
+            {% for poller_name, poller in pollers.items() %}
                 <tr>
                     <td>{{ poller.name }}</td>
                     <td>{{ poller.ip }}</td>
                     <td>{{ poller.port }}</td>
                     <td>
-                        {% if poller.online %}
+                        {% if poller.is_online %}
                             <span class="label label-success">Online</span>
                         {% else %}
                             <span class="label label-danger">Offline</span>

controller/libs/pollers.py

@@ -17,4 +17,21 @@ def __init__(self, ip, port, name=None):
         self.port = port
         self.name = name
         self.online_since = time()
-        self.last_keepalive = None
+        self.last_keepalive = time()
+        self.keepalive_interval = 120
+        self._is_online = True
+
+    @property
+    def is_online(self):
+        """Check if the poller is still online"""
+        next_keepalive = self.last_keepalive + self.keepalive_interval
+        if next_keepalive >= time():
+            self._is_online = True
+        else:
+            self._is_online = False
+
+        return self._is_online
+
+    def keepalive(self):
+        """Handle received keepalive"""
+        self.last_keepalive = time()