use of io.mycat.beans.mysql.packet.ErrorPacketImpl in project Mycat2 by MyCATApache.
the class RowSetQuery method runTextQuery.
public static PromiseInternal<SqlResult<Void>> runTextQuery(String curSql, MySQLClientSession mySQLClientSession, StreamMysqlCollector collectorArg) {
if (mySQLClientSession.getCurNIOHandler() != null) {
throw new IllegalArgumentException();
}
PromiseInternal<SqlResult<Void>> promise = VertxUtil.newPromise();
if (mySQLClientSession.getIOThread() == Thread.currentThread()) {
VertxMycatTextCollector<Object, Object> resultSetHandler = new VertxMycatTextCollector<Object, Object>((Collector) collectorArg);
if (LOGGER.isDebugEnabled()) {
if (curSql.startsWith("XA ROLLBACK")) {
LOGGER.debug("session id:{} sql:{}", mySQLClientSession.sessionId(), curSql, new Throwable());
}
LOGGER.debug("session id:{} sql:{}", mySQLClientSession.sessionId(), curSql);
}
resultSetHandler.request(mySQLClientSession, MySQLCommandType.COM_QUERY, curSql.getBytes(), new ResultSetCallBack<MySQLClientSession>() {
@Override
public void onFinishedSendException(Exception exception, Object sender, Object attr) {
MycatException mycatException = new MycatException(MySQLErrorCode.ER_UNKNOWN_ERROR, exception.getMessage());
LOGGER.error("session id:{} sql:{}", mySQLClientSession.sessionId(), curSql, mycatException);
promise.tryFail(mycatException);
}
@Override
public void onFinishedException(Exception exception, Object sender, Object attr) {
MycatException mycatException = new MycatException(MySQLErrorCode.ER_UNKNOWN_ERROR, exception.getMessage());
LOGGER.error("session id:{} sql:{}", mySQLClientSession.sessionId(), curSql, mycatException);
promise.tryFail(mycatException);
}
@Override
public void onFinished(boolean monopolize, MySQLClientSession mysql, Object sender, Object attr) {
if (LOGGER.isDebugEnabled()) {
LOGGER.debug("onFinished session id:{} sql:{}", mySQLClientSession.sessionId(), curSql);
}
MySqlResult<Void> mySqlResult = new MySqlResult<>(resultSetHandler.getRowCount(), resultSetHandler.getAffectedRows(), resultSetHandler.getLastInsertId(), null, Optional.ofNullable(resultSetHandler.getRowResultDecoder()).map(i -> i.rowDesc).map(i -> i.columnDescriptor()).orElse(Collections.emptyList()));
promise.complete(mySqlResult);
}
@Override
public void onErrorPacket(ErrorPacketImpl errorPacket, boolean monopolize, MySQLClientSession mysql, Object sender, Object attr) {
MycatException mycatException = new MycatException(errorPacket.getErrorCode(), errorPacket.getErrorMessageString());
LOGGER.error("onErrorPacket session id:{} sql:{}", mySQLClientSession.sessionId(), curSql, mycatException);
promise.tryFail(mycatException);
}
});
} else {
mySQLClientSession.getIOThread().addNIOJob(new NIOJob() {
@Override
public void run(ReactorEnvThread reactor) throws Exception {
if (LOGGER.isDebugEnabled()) {
LOGGER.error("nio job session id:{} sql:{}", mySQLClientSession.sessionId(), curSql, new Throwable());
}
runTextQuery(curSql, mySQLClientSession, collectorArg).onComplete(promise);
}
@Override
public void stop(ReactorEnvThread reactor, Exception reason) {
promise.tryFail(reason);
}
@Override
public String message() {
return "proxy query text result set";
}
});
}
return promise;
}
use of io.mycat.beans.mysql.packet.ErrorPacketImpl in project Mycat2 by MyCATApache.
the class MySQLAPIImpl method query.
@Override
public void query(String sql, ResultSetCollector collector, MySQLAPIExceptionCallback exceptionCollector) {
TextResultSetTransforCollector transfor = new TextResultSetTransforCollector(collector);
TextResultSetHandler queryResultSetTask = new TextResultSetHandler(transfor);
queryResultSetTask.request(mySQLClientSession, MySQLCommandType.COM_QUERY, sql.getBytes(), new ResultSetCallBack<MySQLClientSession>() {
@Override
public void onFinishedSendException(Exception exception, Object sender, Object attr) {
exceptionCollector.onException(exception, MySQLAPIImpl.this);
}
@Override
public void onFinishedException(Exception exception, Object sender, Object attr) {
exceptionCollector.onException(exception, MySQLAPIImpl.this);
}
@Override
public void onFinished(boolean monopolize, MySQLClientSession mysql, Object sender, Object attr) {
exceptionCollector.onFinished(monopolize, MySQLAPIImpl.this);
}
@Override
public void onErrorPacket(ErrorPacketImpl errorPacket, boolean monopolize, MySQLClientSession mysql, Object sender, Object attr) {
exceptionCollector.onErrorPacket(errorPacket, monopolize, MySQLAPIImpl.this);
}
});
}
use of io.mycat.beans.mysql.packet.ErrorPacketImpl in project Mycat2 by MyCATApache.
the class ResultSetHandler method onSocketRead.
// /**
// * 该方法可能会被重写
// */
// default void onFinishedCollect(MySQLClientSession mysql, boolean success, String errorMessage) {
// ResultSetCallBack callBack = mysql.getCallBackAndReset();
// assert callBack != null;
// if (success) {
// callBack.onFinishedCollect(mysql, this, true, getResult(), null);
// } else {
// callBack.onFinishedCollect(mysql, this, false, errorMessage, null);
// }
// }
/**
* 读事件处理
*/
@Override
default void onSocketRead(MySQLClientSession mysql) {
assert mysql.getCurNIOHandler() == this;
if (!mysql.checkOpen()) {
ResultSetCallBack callBackAndReset = mysql.getCallBack();
ClosedChannelException closedChannelException = new ClosedChannelException();
onException(mysql, closedChannelException);
callBackAndReset.onFinishedException(closedChannelException, this, null);
return;
}
try {
MySQLPacketResolver resolver = mysql.getBackendPacketResolver();
ProxyBuffer proxyBuffer = mysql.currentProxyBuffer();
proxyBuffer.newBufferIfNeed();
if (!mysql.readFromChannel()) {
return;
}
mysql.setRequestSuccess(true);
int totalPacketEndIndex = proxyBuffer.channelReadEndIndex();
MySQLPacket mySQLPacket = (MySQLPacket) proxyBuffer;
boolean isResponseFinished = false;
ErrorPacketImpl errorPacket = null;
while (mysql.readProxyPayloadFully()) {
MySQLPayloadType type = mysql.getBackendPacketResolver().getMySQLPayloadType();
isResponseFinished = mysql.isResponseFinished();
MySQLPacket payload = mysql.currentProxyPayload();
int startPos = payload.packetReadStartIndex();
int endPos = payload.packetReadEndIndex();
switch(type) {
case REQUEST:
this.onRequest(mySQLPacket, startPos, endPos);
break;
case LOAD_DATA_REQUEST:
this.onLoadDataRequest(mySQLPacket, startPos, endPos);
break;
case REQUEST_COM_QUERY:
this.onRequestComQuery(mySQLPacket, startPos, endPos);
break;
case REQUEST_SEND_LONG_DATA:
this.onPrepareLongData(mySQLPacket, startPos, endPos);
break;
case REQUEST_PREPARE:
this.onReqeustPrepareStatement(mySQLPacket, startPos, endPos);
break;
case REQUEST_COM_STMT_CLOSE:
this.onRequestComStmtClose(mySQLPacket, startPos, endPos);
break;
case FIRST_ERROR:
{
ErrorPacketImpl packet = new ErrorPacketImpl();
errorPacket = packet;
packet.readPayload(mySQLPacket);
this.onFirstError(packet);
break;
}
case FIRST_OK:
MycatMonitor.onResultSetEnd(mysql);
this.onOk(mySQLPacket, startPos, endPos);
break;
case FIRST_EOF:
this.onEof(mySQLPacket, startPos, endPos);
break;
case COLUMN_COUNT:
this.onColumnCount(resolver.getColumnCount());
break;
case COLUMN_DEF:
this.onColumnDef(mySQLPacket, startPos, endPos);
break;
case COLUMN_EOF:
this.onColumnDefEof(mySQLPacket, startPos, endPos);
break;
case TEXT_ROW:
this.onTextRow(mySQLPacket, startPos, endPos);
break;
case BINARY_ROW:
this.onBinaryRow(mySQLPacket, startPos, endPos);
break;
case ROW_EOF:
case ROW_OK:
MycatMonitor.onResultSetEnd(mysql);
this.onRowOk(mySQLPacket, startPos, endPos);
break;
case ROW_ERROR:
ErrorPacketImpl packet = new ErrorPacketImpl();
errorPacket = packet;
packet.readPayload(mySQLPacket);
this.onRowError(packet, startPos, endPos);
break;
case PREPARE_OK:
this.onPrepareOk(resolver);
break;
case PREPARE_OK_PARAMER_DEF:
this.onPrepareOkParameterDef(mySQLPacket, startPos, endPos);
break;
case PREPARE_OK_COLUMN_DEF:
this.onPrepareOkColumnDef(mySQLPacket, startPos, endPos);
break;
case PREPARE_OK_COLUMN_DEF_EOF:
this.onPrepareOkColumnDefEof(resolver);
break;
case PREPARE_OK_PARAMER_DEF_EOF:
this.onPrepareOkParameterDefEof(resolver);
break;
}
mysql.resetCurrentProxyPayload();
proxyBuffer.channelReadEndIndex(totalPacketEndIndex);
if (isResponseFinished) {
break;
}
assert mysql.getCurNIOHandler() == this;
MySQLPacketResolver packetResolver = mysql.getBackendPacketResolver();
mySQLPacket.packetReadStartIndex(packetResolver.getEndPos());
}
if (isResponseFinished) {
ByteBuffer allocate = ByteBuffer.allocate(8192);
if (mysql.channel().read(allocate) > 0) {
throw new IllegalArgumentException();
}
boolean responseFinished = mysql.isResponseFinished();
mysql.getBackendPacketResolver().setState(MySQLPacketResolver.ComQueryState.QUERY_PACKET);
ResultSetCallBack callBackAndReset = mysql.getCallBack();
mysql.setCallBack(null);
onFinishedCollect(mysql);
onClear(mysql);
if (errorPacket == null) {
callBackAndReset.onFinished(mysql.isMonopolized(), mysql, this, null);
} else {
callBackAndReset.onErrorPacket(errorPacket, mysql.isMonopolized(), mysql, this, null);
}
return;
}
} catch (Exception e) {
LOGGER.error("", e);
ResultSetCallBack callBackAndReset = mysql.getCallBack();
Objects.requireNonNull(callBackAndReset);
mysql.setCallBack(null);
if (mysql.isRequestSuccess()) {
MycatMonitor.onResultSetReadException(mysql, e);
onFinishedCollectException(mysql, e);
onException(mysql, e);
callBackAndReset.onFinishedException(e, this, null);
return;
} else {
MycatMonitor.onResultSetWriteException(mysql, e);
onFinishedCollectException(mysql, e);
onException(mysql, e);
callBackAndReset.onFinishedSendException(e, this, null);
return;
}
}
}
Aggregations