Minfra demo (GoogleCloudPlatform#2233)

djmailhot · andrewsg · commit 6561c3d4b7b0 · 2019-06-25T16:26:43.000-07:00
* Add load generation to UI of minfra demo webserver

* Add comments to minfra demo for simulating high CPU load

* Fix linting for minfra demo webserver
diff --git a/compute/managed-instances/demo/app.py b/compute/managed-instances/demo/app.py
@@ -20,25 +20,29 @@
 """
 
 from flask import Flask, make_response, render_template
+from random import random
 from re import sub
 from requests import get
 from socket import gethostname
+from multiprocessing import Process
 
 PORT_NUMBER = 80
 
 app = Flask(__name__)
-healthy = True
+_is_healthy = True
+_worker = None
 
 
 @app.route('/')
 def index():
     """Returns the demo UI."""
-    global healthy
+    global _is_healthy
     return render_template('index.html',
                            hostname=gethostname(),
                            zone=_get_zone(),
                            template=_get_template(),
-                           healthy=healthy)
+                           healthy=_is_healthy,
+                           working=_is_working())
 
 
 @app.route('/health')
@@ -48,21 +52,23 @@ def health():
     Returns:
         HTTP status 200 if 'healthy', HTTP status 500 if 'unhealthy'
     """
-    global healthy
-    template = render_template('health.html', healthy=healthy)
-    return make_response(template, 200 if healthy else 500)
+    global _is_healthy
+    template = render_template('health.html', healthy=_is_healthy)
+    return make_response(template, 200 if _is_healthy else 500)
 
 
 @app.route('/makeHealthy')
 def make_healthy():
     """Sets the server to simulate a 'healthy' status."""
-    global healthy
-    healthy = True
+    global _is_healthy
+    _is_healthy = True
+
     template = render_template('index.html',
                                hostname=gethostname(),
                                zone=_get_zone(),
                                template=_get_template(),
-                               healthy=True)
+                               healthy=True,
+                               working=_is_working())
     response = make_response(template, 302)
     response.headers['Location'] = '/'
     return response
@@ -71,13 +77,53 @@ def make_healthy():
 @app.route('/makeUnhealthy')
 def make_unhealthy():
     """Sets the server to simulate an 'unhealthy' status."""
-    global healthy
-    healthy = False
+    global _is_healthy
+    _is_healthy = False
+
+    template = render_template('index.html',
+                               hostname=gethostname(),
+                               zone=_get_zone(),
+                               template=_get_template(),
+                               healthy=False,
+                               working=_is_working())
+    response = make_response(template, 302)
+    response.headers['Location'] = '/'
+    return response
+
+
+@app.route('/startLoad')
+def start_load():
+    """Sets the server to simulate high CPU load."""
+    global _worker, _is_healthy
+    if not _is_working():
+        _worker = Process(target=_burn_cpu)
+        _worker.start()
+
+    template = render_template('index.html',
+                               hostname=gethostname(),
+                               zone=_get_zone(),
+                               template=_get_template(),
+                               healthy=_is_healthy,
+                               working=True)
+    response = make_response(template, 302)
+    response.headers['Location'] = '/'
+    return response
+
+
+@app.route('/stopLoad')
+def stop_load():
+    """Sets the server to stop simulating CPU load."""
+    global _worker, _is_healthy
+    if _is_working():
+        _worker.terminate()
+        _worker = None
+
     template = render_template('index.html',
                                hostname=gethostname(),
                                zone=_get_zone(),
                                template=_get_template(),
-                               healthy=False)
+                               healthy=_is_healthy,
+                               working=False)
     response = make_response(template, 302)
     response.headers['Location'] = '/'
     return response
@@ -116,5 +162,17 @@ def _get_template():
         return ''
 
 
+def _is_working():
+    """Returns TRUE if the server is currently simulating CPU load."""
+    global _worker
+    return _worker is not None and _worker.is_alive()
+
+
+def _burn_cpu():
+    """Burn CPU cycles to simulate high CPU load."""
+    while True:
+        random()*random()
+
+
 if __name__ == "__main__":
     app.run(debug=False, port=PORT_NUMBER)
diff --git a/compute/managed-instances/demo/templates/index.html b/compute/managed-instances/demo/templates/index.html
@@ -21,6 +21,14 @@
                 <td>Template:</td>
                 <td><b>{{ template }}</b></td>
             </tr>
+            <tr>
+                <td>Current load:</td>
+                {% if working %}
+                <td><span class="btn red">high</span></td>
+                {% else %}
+                <td><span class="btn green">none</span></td>
+                {% endif %}
+            </tr>
             <tr>
                 <td>Health status:</td>
                 {% if healthy %}
@@ -39,6 +47,12 @@
                     {% endif %}
 
                     <a class="btn blue" href="/health">Check health</a>
+
+                    {% if working %}
+                    <a class="btn blue" href="/stopLoad">Stop load</a>
+                    {% else %}
+                    <a class="btn blue" href="/startLoad">Start load</a>
+                    {% endif %}
                 </td>
             </tr>
         </tbody>
