Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.github.netty.protocol.mysql.server.ServerResultsetDecoder Maven / Gradle / Ivy
package com.github.netty.protocol.mysql.server;
import com.github.netty.protocol.mysql.*;
import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.EnumSet;
import java.util.List;
import java.util.Set;
/**
*
*/
public class ServerResultsetDecoder extends AbstractPacketDecoder implements ServerDecoder {
private State state = State.COLUMN_COUNT;
private List columnDefinitions;
private Session session;
public ServerResultsetDecoder(Session session, int maxPacketSize) {
super(maxPacketSize);
this.session = session;
}
@Override
protected void decodePacket(ChannelHandlerContext ctx, int sequenceId, ByteBuf packet, List out) {
EnumSet capabilities = session.getFrontendCapabilities();
MysqlCharacterSet serverCharset = session.getServerCharset();
switch (state) {
case ERROR:
case COMPLETE:
case COLUMN_COUNT:
handleColumnCount(sequenceId, packet, out, capabilities, serverCharset);
break;
case COLUMN_DEFINITION:
handleColumnDefinition(sequenceId, packet, out, capabilities, serverCharset);
break;
case ROW:
handleRow(sequenceId, packet, out, capabilities, serverCharset);
break;
default: {
break;
}
}
}
private void handleColumnCount(int sequenceId, ByteBuf packet, List out,
Set capabilities, MysqlCharacterSet serverCharset) {
int header = packet.readUnsignedByte();
if (header == RESPONSE_ERROR) {
state = State.ERROR;
out.add(decodeErrorResponse(sequenceId, packet, serverCharset));
} else if (header == RESPONSE_OK) {
state = State.COMPLETE;
out.add(decodeOkResponse(sequenceId, packet, capabilities, serverCharset));
} else {
state = State.COLUMN_DEFINITION;
out.add(decodeFieldCount(sequenceId, packet, header));
}
}
private ServerColumnCountPacket decodeFieldCount(int sequenceId, ByteBuf packet, int header) {
int currentResultSetFieldCount = (int) CodecUtils.readLengthEncodedInteger(packet, header);
if (currentResultSetFieldCount < 0) {
throw new IllegalStateException("Field count is too large to handle");
}
columnDefinitions = new ArrayList<>(currentResultSetFieldCount);
return new ServerColumnCountPacket(sequenceId, currentResultSetFieldCount);
}
private void handleColumnDefinition(int sequenceId, ByteBuf packet, List out,
Set capabilities, MysqlCharacterSet serverCharset) {
int header = packet.readUnsignedByte();
if (header == RESPONSE_EOF) {
state = State.ROW;
out.add(decodeEofResponse(sequenceId, packet, capabilities));
} else {
ServerColumnDefinitionPacket columnDefinition = decodeColumnDefinition(sequenceId, packet, header, serverCharset);
columnDefinitions.add(columnDefinition);
out.add(columnDefinition);
}
}
private ServerColumnDefinitionPacket decodeColumnDefinition(int sequenceId, ByteBuf packet, int header, MysqlCharacterSet serverCharset) {
Charset charset = serverCharset.getCharset();
ServerColumnDefinitionPacket.Builder builder = ServerColumnDefinitionPacket
.builder()
.sequenceId(sequenceId)
.catalog(CodecUtils.readLengthEncodedString(packet, header, charset))
.schema(CodecUtils.readLengthEncodedString(packet, charset))
.table(CodecUtils.readLengthEncodedString(packet, charset))
.orgTable(CodecUtils.readLengthEncodedString(packet, charset))
.name(CodecUtils.readLengthEncodedString(packet, charset))
.orgName(CodecUtils.readLengthEncodedString(packet, charset));
packet.readByte();
builder.characterSet(MysqlCharacterSet.findById(packet.readShortLE(), serverCharset))
.columnLength(packet.readUnsignedIntLE())
.type(ColumnType.lookup(packet.readUnsignedByte()))
.addFlags(CodecUtils.readShortEnumSet(packet, ColumnFlag.class))
.decimals(packet.readUnsignedByte());
packet.skipBytes(2);
return builder.build();
}
private void handleRow(int sequenceId, ByteBuf packet, List out, Set capabilities, MysqlCharacterSet serverCharset) {
int header = packet.readByte() & 0xff;
switch (header) {
case RESPONSE_ERROR:
state = State.ERROR;
out.add(decodeErrorResponse(sequenceId, packet, serverCharset));
break;
case RESPONSE_EOF:
state = State.COMPLETE;
out.add(decodeEofResponse(sequenceId, packet, capabilities));
break;
case RESPONSE_OK:
state = State.COMPLETE;
decodeOkResponse(sequenceId, packet, capabilities, serverCharset);
break;
default:
decodeRow(sequenceId, packet, header, out);
}
}
private void decodeRow(int sequenceId, ByteBuf packet, int firstByte, List out) {
int size = columnDefinitions.size();
String[] values = new String[size];
values[0] = CodecUtils.readLengthEncodedString(packet, firstByte, columnDefinitions.get(0).getCharacterSet().getCharset());
for (int i = 1; i < size; i++) {
values[i] = CodecUtils.readLengthEncodedString(packet, columnDefinitions.get(i).getCharacterSet().getCharset());
}
out.add(new ServerResultsetRowPacket(sequenceId, values));
}
enum State {
COLUMN_COUNT,
COLUMN_DEFINITION,
COMPLETE,
ERROR,
ROW
}
}
