Prometheus监控自研组件之数据库TeleDB

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from sqlalchemy.ext.automap import automap_base
from sqlalchemy.orm import Session, sessionmaker
from sqlalchemy import create_engine, MetaData
from urllib import parse
import prometheus_client
from flask import Flask
from prometheus_client import make_wsgi_app
from prometheus_client import Gauge, start_http_server, Counter
from flask import Response, Flask
from prometheus_client.core import InfoMetricFamily, GaugeMetricFamily, REGISTRY,CollectorRegistry

def get_data():
  URL="mysql+pymysql://teledb:%s@135.x.x.x:9014/teledb?charset=utf8" % parse.quote_plus('xxxxxxx')
  print (URL)
  metadata = MetaData()
  engine = create_engine(URL)
  session_factory = sessionmaker(bind=engine)
  session = session_factory()
  metadata.reflect(engine, only=['mysql_status'])
  Base = automap_base(metadata=metadata)
  Base.prepare()
  status_gz = Base.classes.mysql_status
  query = session.query(status_gz)
  list = []
  for row in query:
    dict = {"host":row.host,"port":row.port,"tags":row.tags,"threads":row.threads_running}
    list.append(dict)
  print (list)

  return list   #此处返回一个带有监控数据的列表

app = Flask(__name__)
@app.route("/metrics")
def requests_count():
  data_list=get_data()

  REGISTRY = CollectorRegistry(auto_describe=False)
  status = Gauge("CRM_teledb_runing_threads", "monitor running status.",["host","port","tags"], registry=REGISTRY)#定义Prometheus的自定义指标，打标签，这里给这个指标起名
                                                                                                                                                                    #JF_teledb_runing_threads。
  for data in data_list:
    host ="".join( data.get("host"))
    port = "".join( data.get("port"))
    tags = "".join( data.get("tags"))
    threads = data.get("threads")
    status.labels(host, port,tags).set(threads)
  return Response(prometheus_client.generate_latest(REGISTRY),
                    mimetype="text/plain")
if __name__ == '__main__':
 app.run(host="0.0.0.0",port=8181,debug=True)#把这个web服务器跑在8181端口