基于小米SAOR实现SQL自助优化接口

Posted 2021-11-12 _雪辉_

#!/usr/bin/python3
# -*- coding: UTF-8 -*-
import pymysql
import os
import json
import time
import logging
import subprocess
from connect_db_forbatch import connect_mysql
from flask_cors import *
os.environ['NLS_LANG'] = 'SIMPLIFIED CHINESE_CHINA.UTF8'
from flask import Flask,request,jsonify
app = Flask(__name__)

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)
logdate = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))
log_path = '/data/mysql/logs/'
log_name = log_path + logdate + '.log'
logfile = log_name
fh = logging.FileHandler(logfile, mode='w')
fh.setLevel(logging.DEBUG)
formatter = logging.Formatter("%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s")
fh.setFormatter(formatter)
logger.addHandler(fh)

@app.route('/optimizesql/', methods=['POST'])
def soar():
    try:
        if not request.data:  #检测是否有数据
            return ('fail')
        params= request.data.decode('utf-8')
        prams = json.loads(params)
        clustername = prams['clustername']
        dbname = prams['dbname']
        sql = prams['sql'].replace('\\\\','').replace('`','\\`')
        #获取mysql实例的元数据信息
        get_ip = "select ip,port from mysql_instance where clustername='%s'" %(clustername)
        host = connect_mysql('127.0.0.1', 33071, 'dbname', 'tablename').select_metadata(get_ip)
        ip = host[0][0]
        port = host[0][1]
        html_cmd = "soar -online-dsn=\\"user:password@%s:%s/%s\\" -test-dsn=\\"user:password@host:port/optimizesql\\" -query=\\"%s\\"  -report-type markdown -log-output=/data/ansible/logs/optimizesql.log" %(ip,port,dbname,sql)
        (status, content) = subprocess.getstatusoutput(html_cmd)
        if status != 0 :
            content = "analyze failed"
        return jsonify(content)
    except Exception as err:
        logger.error('%s',err, exc_info=True)

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5590)