/* * Copyright 2012-2015 org.opencloudb. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.opencloudb.response; import java.nio.ByteBuffer; import org.opencloudb.MycatServer; import org.opencloudb.config.Fields; import org.opencloudb.manager.ManagerConnection; import org.opencloudb.mysql.PacketUtil; import org.opencloudb.net.mysql.EOFPacket; import org.opencloudb.net.mysql.FieldPacket; import org.opencloudb.net.mysql.ResultSetHeaderPacket; import org.opencloudb.net.mysql.RowDataPacket; import org.opencloudb.statistic.SQLRecord; import org.opencloudb.util.IntegerUtil; import org.opencloudb.util.LongUtil; import org.opencloudb.util.StringUtil; /** * 查询执行时间超过设定阈值的SQL * * @author mycat */ public final class ShowSQLSlow { private static final int FIELD_COUNT = 7; private static final ResultSetHeaderPacket header = PacketUtil.getHeader(FIELD_COUNT); private static final FieldPacket[] fields = new FieldPacket[FIELD_COUNT]; private static final EOFPacket eof = new EOFPacket(); static { int i = 0; byte packetId = 0; header.packetId = ++packetId; fields[i] = PacketUtil.getField("HOST", Fields.FIELD_TYPE_VAR_STRING); fields[i++].packetId = ++packetId; fields[i] = PacketUtil.getField("SCHEMA", Fields.FIELD_TYPE_LONG); fields[i++].packetId = ++packetId; fields[i] = PacketUtil.getField("DATASOURCE", Fields.FIELD_TYPE_VAR_STRING); fields[i++].packetId = ++packetId; fields[i] = PacketUtil.getField("INDEX", Fields.FIELD_TYPE_LONG); fields[i++].packetId = ++packetId; fields[i] = PacketUtil.getField("START_TIME", Fields.FIELD_TYPE_LONGLONG); fields[i++].packetId = ++packetId; fields[i] = PacketUtil.getField("EXECUTE_TIME", Fields.FIELD_TYPE_LONGLONG); fields[i++].packetId = ++packetId; fields[i] = PacketUtil.getField("SQL", Fields.FIELD_TYPE_VAR_STRING); fields[i++].packetId = ++packetId; eof.packetId = ++packetId; } public static void execute(ManagerConnection c) { ByteBuffer buffer = c.allocate(); // write header buffer = header.write(buffer, c); // write fields for (FieldPacket field : fields) { buffer = field.write(buffer, c); } // write eof buffer = eof.write(buffer, c); // write rows byte packetId = eof.packetId; SQLRecord[] records = MycatServer.getInstance().getSqlRecorder().getRecords(); for (int i = records.length - 1; i >= 0; i--) { if (records[i] != null) { RowDataPacket row = getRow(records[i], c.getCharset()); row.packetId = ++packetId; buffer = row.write(buffer, c); } } // write last eof EOFPacket lastEof = new EOFPacket(); lastEof.packetId = ++packetId; buffer = lastEof.write(buffer, c); // write buffer c.write(buffer); } private static RowDataPacket getRow(SQLRecord sql, String charset) { RowDataPacket row = new RowDataPacket(FIELD_COUNT); row.add(StringUtil.encode(sql.host, charset)); row.add(StringUtil.encode(sql.schema, charset)); row.add(StringUtil.encode(sql.dataNode, charset)); row.add(IntegerUtil.toBytes(sql.dataNodeIndex)); row.add(LongUtil.toBytes(sql.startTime)); row.add(LongUtil.toBytes(sql.executeTime)); row.add(StringUtil.encode(sql.statement, charset)); return row; } }