Hyunsun Moon | f32ae9a | 2020-05-28 13:17:45 -0700 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | |
| 3 | # Copyright 2020-present Open Networking Foundation |
| 4 | # |
| 5 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | # you may not use this file except in compliance with the License. |
| 7 | # You may obtain a copy of the License at |
| 8 | # |
| 9 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 10 | # |
| 11 | # Unless required by applicable law or agreed to in writing, software |
| 12 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | # See the License for the specific language governing permissions and |
| 15 | # limitations under the License. |
| 16 | |
| 17 | import time |
Andy Bavier | 4021a2f | 2020-07-29 12:39:47 -0700 | [diff] [blame^] | 18 | from flask import Flask, jsonify, abort, request, Response |
| 19 | import prometheus_client as prom |
| 20 | |
| 21 | NO_RESULT_THRESHOLD = 720 |
Hyunsun Moon | f32ae9a | 2020-05-28 13:17:45 -0700 | [diff] [blame] | 22 | |
| 23 | app = Flask(__name__) |
| 24 | edges = [ |
| 25 | { |
| 26 | 'name': 'production-edge-example', |
| 27 | 'status': { |
| 28 | 'control_plane': 'connected', |
| 29 | 'user_plane': 'connected' |
| 30 | }, |
| 31 | 'last_update': time.time() |
| 32 | } |
| 33 | ] |
| 34 | |
Andy Bavier | 4021a2f | 2020-07-29 12:39:47 -0700 | [diff] [blame^] | 35 | status_codes = { |
| 36 | "no result": -2, |
| 37 | "error": -1, |
| 38 | "disconnected": 0, |
| 39 | "connecting": 1, |
| 40 | "connected": 2 |
| 41 | } |
| 42 | |
| 43 | cp_status = prom.Gauge("aetheredge_status_control_plane", "Control plane status code", ["name"]) |
| 44 | up_status = prom.Gauge("aetheredge_status_user_plane", "User plane status code", ["name"]) |
| 45 | last_update = prom.Gauge("aetheredge_last_update", "Last reported test result", ["name"]) |
| 46 | |
| 47 | def time_out_stale_results(): |
| 48 | for edge in edges: |
| 49 | time_elapsed = time.time() - edge["last_update"] |
| 50 | if time_elapsed > NO_RESULT_THRESHOLD: |
| 51 | edge['status']['control_plane'] = "no result" |
| 52 | edge['status']['user_plane'] = "no result" |
| 53 | |
| 54 | |
| 55 | @app.route('/edges/metrics', methods=['GET']) |
| 56 | def get_prometheus_metrics(): |
| 57 | res = [] |
| 58 | time_out_stale_results() |
| 59 | for edge in edges: |
| 60 | if edge['name'] == "production-edge-example": |
| 61 | continue |
| 62 | |
| 63 | cp_status.labels(edge['name']).set(status_codes[edge['status']['control_plane']]) |
| 64 | up_status.labels(edge['name']).set(status_codes[edge['status']['user_plane']]) |
| 65 | last_update.labels(edge['name']).set(edge['last_update']) |
| 66 | |
| 67 | res.append(prom.generate_latest(cp_status)) |
| 68 | res.append(prom.generate_latest(up_status)) |
| 69 | res.append(prom.generate_latest(last_update)) |
| 70 | return Response(res, mimetype="text/plain") |
| 71 | |
Hyunsun Moon | f32ae9a | 2020-05-28 13:17:45 -0700 | [diff] [blame] | 72 | |
| 73 | @app.route('/edges/healthz', methods=['GET']) |
| 74 | def get_health(): |
| 75 | return {'message': 'healthy'} |
| 76 | |
| 77 | |
| 78 | @app.route('/edges', methods=['GET']) |
| 79 | def get_edges(): |
Andy Bavier | 4021a2f | 2020-07-29 12:39:47 -0700 | [diff] [blame^] | 80 | time_out_stale_results() |
Hyunsun Moon | f32ae9a | 2020-05-28 13:17:45 -0700 | [diff] [blame] | 81 | return jsonify({'edges': edges}) |
| 82 | |
| 83 | |
| 84 | @app.route('/edges/<string:name>', methods=['GET']) |
| 85 | def get_edge(name): |
Andy Bavier | 4021a2f | 2020-07-29 12:39:47 -0700 | [diff] [blame^] | 86 | time_out_stale_results() |
Hyunsun Moon | f32ae9a | 2020-05-28 13:17:45 -0700 | [diff] [blame] | 87 | edge = [edge for edge in edges if edge['name'] == name] |
| 88 | if len(edge) == 0: |
| 89 | abort(404) |
| 90 | return jsonify({'edge': edge[0]}) |
| 91 | |
| 92 | |
| 93 | @app.route('/edges', methods=['POST']) |
| 94 | def create_or_update_edge(): |
| 95 | if not request.json: |
| 96 | abort(400) |
| 97 | if 'name' not in request.json: |
| 98 | abort(400) |
| 99 | if 'status' not in request.json: |
| 100 | abort(400) |
| 101 | |
| 102 | req_edge = { |
| 103 | 'name': request.json['name'], |
| 104 | 'status': { |
| 105 | 'control_plane': request.json['status']['control_plane'], |
| 106 | 'user_plane': request.json['status']['user_plane'] |
| 107 | }, |
| 108 | 'last_update': time.time() |
| 109 | } |
| 110 | |
| 111 | edge = [edge for edge in edges if edge['name'] == req_edge['name']] |
| 112 | if len(edge) == 0: |
| 113 | print("new edge request " + req_edge['name']) |
| 114 | edges.append(req_edge) |
| 115 | else: |
| 116 | edge[0]['status']['control_plane'] = req_edge['status']['control_plane'] |
| 117 | edge[0]['status']['user_plane'] = req_edge['status']['user_plane'] |
| 118 | edge[0]['last_update'] = req_edge['last_update'] |
| 119 | |
| 120 | return jsonify({'edge': req_edge}), 201 |
| 121 | |
| 122 | |
| 123 | if __name__ == '__main__': |
| 124 | app.run(debug=True, host='0.0.0.0', port=80) |