18 lines
1.2 KiB
Python
18 lines
1.2 KiB
Python
import pymysql,pathlib,re,json
|
|
|
|
def connect():
|
|
env={}
|
|
for name in ['base.env','stat-writer.env']:
|
|
for line in pathlib.Path('/opt/lingniu-go-native/env/'+name).read_text().splitlines():
|
|
if '=' in line and not line.startswith('#'):
|
|
k,v=line.split('=',1);env[k]=v.strip().strip(chr(34)).strip(chr(39))
|
|
m=re.fullmatch(r'([^:]+):(.*?)@tcp\(([^:]+):(\d+)\)/([^?]+).*',env['MYSQL_DSN'])
|
|
return pymysql.connect(user=m[1],password=m[2],host=m[3],port=int(m[4]),database=m[5],charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor,autocommit=False)
|
|
if __name__=='__main__':
|
|
c=connect()
|
|
with c.cursor() as q:
|
|
q.execute('START TRANSACTION READ ONLY')
|
|
for sql in ["SELECT DATABASE() AS db, VERSION() AS version", "SELECT table_name,engine,table_rows,ROUND((data_length+index_length)/1024/1024,1) AS mb FROM information_schema.tables WHERE table_schema=DATABASE() ORDER BY data_length DESC", "SELECT protocol,source_ip,COUNT(*) AS n,COUNT(DISTINCT vin) AS vins FROM vehicle_daily_mileage_source WHERE source_ip IN ('legacy-mysql.lingniu-prod','manual-lingniu-prod-day-mileage') GROUP BY protocol,source_ip"]:
|
|
q.execute(sql);print(json.dumps(q.fetchall(),default=str))
|
|
c.rollback();c.close()
|