package org.opencloudb.util;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.List;
import org.opencloudb.net.mysql.FieldPacket;
import org.opencloudb.net.mysql.RowDataPacket;
/**
*
* @author struct
*
*/
public class ResultSetUtil {
public static int toFlag(ResultSetMetaData metaData, int column)
throws SQLException {
int flags = 0;
if (metaData.isNullable(column) == 1) {
flags |= 0001;
}
if (metaData.isSigned(column)) {
flags |= 0020;
}
if (metaData.isAutoIncrement(column)) {
flags |= 0200;
}
return flags;
}
public static void resultSetToPacket(String charset, Connection source,
List<FieldPacket> fieldPks, ResultSet rs,
List<RowDataPacket> rowsPkg) throws SQLException {
ResultSetMetaData metaData = rs.getMetaData();
int colunmCount = metaData.getColumnCount();
if (colunmCount > 0) {
for (int i = 0; i < colunmCount; i++) {
int j = i + 1;
FieldPacket fieldPacket = new FieldPacket();
fieldPacket.orgName = StringUtil.encode(
metaData.getColumnLabel(j), charset);
fieldPacket.name = StringUtil.encode(
metaData.getColumnName(j), charset);
fieldPacket.orgTable = StringUtil.encode(
metaData.getTableName(j), charset);
fieldPacket.table = StringUtil.encode(
metaData.getTableName(j), charset);
fieldPacket.db = StringUtil.encode(
metaData.getSchemaName(j), charset);
fieldPacket.length = metaData.getColumnDisplaySize(j);
fieldPacket.flags = toFlag(metaData, j);
fieldPacket.decimals = (byte) metaData.getScale(j);
int javaType = MysqlDefs.javaTypeDetect(
metaData.getColumnType(j), fieldPacket.decimals);
fieldPacket.type = (byte) (MysqlDefs
.javaTypeMysql(javaType) & 0xff);
fieldPks.add(fieldPacket);
}
}
while (rs.next()) {
RowDataPacket row = new RowDataPacket(fieldPks.size());
for (int i = 0; i < colunmCount; i++) {
int j = i + 1;
// BindValue bindValue = new BindValue();
// bindValue.bufferType = fieldPackets[i].type;
// bindValue.bindLength = fieldPackets[i].length;
// bindValue.scale = fieldPackets[i].decimals;
// bindValue.charset = fieldPackets[i].charsetName;
// PacketUtil.resultToBindValue(bindValue, j, rs,
// fieldPackets[i]);
row.add(StringUtil.encode(rs.getString(j), charset));
}
rowsPkg.add(row);
}
}
public static byte[] fromHex(String hexString) {
String[] hex = hexString.split(" ");
byte[] b = new byte[hex.length];
for (int i = 0; i < hex.length; i++) {
b[i] = (byte) (Integer.parseInt(hex[i], 16) & 0xff);
}
return b;
}
public static void main(String[] args) throws Exception {
// byte[] byt = fromHex("20 00 00 02 03 64 65 66 00 00 00 0A 40 40 73 71 6C 5F 6D 6F 64 65 00 0C 21 00 BA 00 00 00 FD 01 00 1F 00 00");
// MysqlPacketBuffer buffer = new MysqlPacketBuffer(byt);
// /*
// * ResultSetHeaderPacket packet = new ResultSetHeaderPacket();
// * packet.init(buffer);
// */
// FieldPacket[] fields = new FieldPacket[(int) 1];
// for (int i = 0; i < 1; i++) {
// fields[i] = new FieldPacket();
// fields[i].init(buffer);
// }
// System.out.println(1 | 0200);
}
}