61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
import time
|
|
from flask import Flask, jsonify, request
|
|
from prometheus_client import Counter, Summary, start_http_server, Info
|
|
from timber import axe, LumberJack
|
|
from config import _CLIENT_CONFIG
|
|
|
|
_VERSION = '0.1.5'
|
|
lumber = LumberJack()
|
|
lumber.set_log_level('i')
|
|
app = Flask(__name__)
|
|
|
|
# Create metrics
|
|
REQUEST_COUNT = Counter('power_api_requests_total', 'Total API Requests', ['method', 'endpoint'])
|
|
REQUEST_LATENCY = Summary('power_api_request_latency_seconds', 'API Request Latency', ['method', 'endpoint'])
|
|
version_info = Info('power_api_version', 'Power API Version')
|
|
version_info.info({'power_api_version': _VERSION})
|
|
client_config_info = Info('power_api_client_config_info', 'Power API Client Config')
|
|
client_config_info.info({
|
|
'power_api_config_check_cycles': str(_CLIENT_CONFIG['config_check']['cycles']),
|
|
'power_api_config_dryrun': str(_CLIENT_CONFIG['dryrun']),
|
|
'power_api_retry_attempts': str(_CLIENT_CONFIG['retry']['attempts']),
|
|
'power_api_retry_seconds_healthy': str(_CLIENT_CONFIG['retry']['seconds']['healthy']),
|
|
'power_api_retry_seconds_unhealthy': str(_CLIENT_CONFIG['retry']['seconds']['unhealthy']),
|
|
'power_api_server_addr': str(_CLIENT_CONFIG['server']['address']),
|
|
'power_api_server_port': str(_CLIENT_CONFIG['server']['port'])
|
|
})
|
|
|
|
@app.before_request
|
|
def before_request():
|
|
request.start_time = time.time()
|
|
|
|
@app.after_request
|
|
def after_request(response):
|
|
latency = time.time() - request.start_time
|
|
REQUEST_COUNT.labels(method=request.method, endpoint=request.path).inc()
|
|
REQUEST_LATENCY.labels(method=request.method, endpoint=request.path).observe(latency)
|
|
return response
|
|
|
|
@app.route('/metrics')
|
|
def metrics():
|
|
return str(REQUEST_COUNT.collect()), 200, {'Content-Type': 'text/plain'}
|
|
|
|
@app.route('/power/<hostname>', methods=['GET'])
|
|
def get_status(hostname):
|
|
axe.info(f'Sending power 200 to {hostname}')
|
|
return hostname, 200
|
|
|
|
@app.route('/power/_VERSION', methods=['GET'])
|
|
def get_version():
|
|
return _VERSION, 200
|
|
|
|
@app.route('/power/_CONFIG/client/<hostname>', methods=['GET'])
|
|
def get_client_config(hostname):
|
|
axe.info(f'Config request from {hostname}')
|
|
return jsonify(_CLIENT_CONFIG), 200
|
|
|
|
if __name__ == '__main__':
|
|
start_http_server(port=8000, addr='0.0.0.0')
|
|
app.run(debug=False, host='0.0.0.0')
|
|
# start_http_server(port=8000)
|
|
# app.run(debug=False) |