Created
March 5, 2017 03:22
-
-
Save limboinf/45efe16aa99ebecb22f156f85e4c1db9 to your computer and use it in GitHub Desktop.
Mysql SLow Query for Flask.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# coding=utf-8 | |
import logging | |
from logging.handlers import RotatingFileHandler | |
from flask import Flask, request, jsonify | |
from flask_sqlalchemy import get_debug_queries | |
from ext import db | |
from users import User | |
app = Flask(__name__) | |
app.config.from_object('config') | |
app.config['DATABASE_QUERY_TIMEOUT'] = 0.0001 | |
app.config['SQLALCHEMY_RECORD_QUERIES'] = True | |
db.init_app(app) | |
formatter = logging.Formatter( | |
"[%(asctime)s] {%(pathname)s:%(lineno)d} %(levelname)s - %(message)s") | |
handler = RotatingFileHandler('slow_query.log', maxBytes=10000, backupCount=10) | |
handler.setLevel(logging.WARN) | |
handler.setFormatter(formatter) | |
app.logger.addHandler(handler) | |
with app.app_context(): | |
db.drop_all() | |
db.create_all() | |
@app.route('/users', methods=['POST']) | |
def users(): | |
username = request.form.get('name') | |
user = User(username) | |
print 'User ID: {}'.format(user.id) | |
db.session.add(user) | |
db.session.commit() | |
return jsonify({'id': user.id}) | |
@app.after_request | |
def after_request(response): | |
for query in get_debug_queries(): | |
if query.duration >= app.config['DATABASE_QUERY_TIMEOUT']: | |
app.logger.warn( | |
('\nContext:{}\nSLOW QUERY: {}\nParameters: {}\n' | |
'Duration: {}\n').format(query.context, query.statement, | |
query.parameters, query.duration)) | |
return response | |
if __name__ == '__main__': | |
app.run(host='0.0.0.0', port=9000) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment