use of org.voltdb.VoltTable.ColumnInfo in project voltdb by VoltDB.
the class GenerateCPPTestFiles method main.
/**
* @param args
*/
public static void main(String[] args) throws Exception {
boolean generateGeoMessages = true;
String clientDataDirName = ".";
long clusterStartTime = CLUSTER_START_TIME;
int clusterRoundTripTime = CLUSTER_ROUND_TRIP_TIME;
long clientData = CLIENT_DATA;
int leaderIPAddr = LEADER_IP_ADDR;
String buildString = BUILD_STRING;
for (int idx = 0; idx < args.length; idx += 1) {
if ("--client-dir".equals(args[idx])) {
idx += 1;
clientDataDirName = args[idx];
} else if ("--clusterStartTime".equals(args[idx])) {
idx += 1;
clusterStartTime = Long.valueOf(args[idx]);
} else if ("--clientData".equals(args[idx])) {
idx += 1;
clientData = Long.valueOf(args[idx]);
} else if ("--leaderIPAddr".equals(args[idx])) {
idx += 1;
leaderIPAddr = Integer.valueOf(args[idx]);
} else if ("--clusterRoundTripTime".equals(args[idx])) {
idx += 1;
clusterRoundTripTime = Integer.valueOf(args[idx]);
} else if ("--no-geo-messages".equals(args[idx])) {
generateGeoMessages = false;
} else {
abend("Unknown command line argument \"%s\"\n", args[idx]);
}
}
// Make the client data directory if necessary.
File clientDataDir = new File(clientDataDirName);
if (clientDataDir.exists() && !clientDataDir.isDirectory()) {
if (!clientDataDir.isDirectory()) {
abend("Client data dir \"%s\" exists but is not a directory.\n", clientDataDirName);
}
} else {
clientDataDir.mkdirs();
}
//
// Capture a HASH_SHA256 style authentication message. We do this by
// creating a fake server, then, in a separate thread, creating an ordinary
// client which connects to the fake server. We read the authentication
// request from the client, save it, send a faked authentication response,
// close the server and join with the created thread.
//
ServerSocketChannel ssc = ServerSocketChannel.open();
ssc.socket().bind(new InetSocketAddress("localhost", FAKE_SERVER_PORT));
ClientConfig config = new ClientConfig("hello", "world", (ClientStatusListenerExt) null, ClientAuthScheme.HASH_SHA256);
final org.voltdb.client.Client client = ClientFactory.createClient(config);
Thread clientThread = new Thread() {
@Override
public void run() {
try {
client.createConnection("localhost", FAKE_SERVER_PORT);
client.close();
} catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
};
clientThread.setDaemon(true);
clientThread.start();
SocketChannel sc = ssc.accept();
sc.socket().setTcpNoDelay(true);
ByteBuffer authReqSHA256 = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
sc.configureBlocking(true);
readMessage(authReqSHA256, sc);
writeDataFile(clientDataDir, "authentication_request_sha256.msg", authReqSHA256);
writeServerAuthenticationResponse(sc, true);
ssc.close();
clientThread.join(0);
//
// Now, create a fake server again, and login with the HASH_SHA1 scheme.
// We save this authentication request as well. The client in the
// separate thread then sends some procedure invocation messages. We
// save all of these in files and then join with the client thread.
//
ssc = ServerSocketChannel.open();
ssc.socket().bind(new InetSocketAddress("localhost", FAKE_SERVER_PORT));
config = new ClientConfig("hello", "world", (ClientStatusListenerExt) null, ClientAuthScheme.HASH_SHA1);
final org.voltdb.client.Client oclient = ClientFactory.createClient(config);
Thread oclientThread = new Thread() {
@Override
public void run() {
NullCallback ncb = new NullCallback();
try {
oclient.createConnection("localhost", FAKE_SERVER_PORT);
oclient.callProcedure("Insert", "Hello", "World", "English");
try {
oclient.callProcedure("Insert", "Hello", "World", "English");
} catch (Exception e) {
}
oclient.callProcedure("Select", "English");
//
// Geo support.
//
// Insert a point and a polygon.
oclient.callProcedure("InsertGeo", 200, GeographyValue.fromWKT(smallPolyTxt), GeographyPointValue.fromWKT(smallPointTxt));
// Insert two nulls for points and polygons.
oclient.callProcedure("InsertGeo", 201, null, null);
// Select one row with a point and a polygon both.
oclient.callProcedure("SelectGeo", 100);
// Select another row with a different point and polygon.
oclient.callProcedure("SelectGeo", 101);
// Select one row with a null polygon and one non-null point.
oclient.callProcedure("SelectGeo", 102);
// Select one row with a non-null polygon and a null point.
oclient.callProcedure("SelectGeo", 103);
// Select one row with two nulls.
oclient.callProcedure("SelectGeo", 104);
} catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
};
oclientThread.setDaemon(true);
oclientThread.start();
sc = ssc.accept();
sc.socket().setTcpNoDelay(true);
ByteBuffer authReqSHA1 = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
sc.configureBlocking(true);
readMessage(authReqSHA1, sc);
writeDataFile(clientDataDir, "authentication_request.msg", authReqSHA1);
writeServerAuthenticationResponse(sc, true);
//
// Read some call procedure messages.
//
// The client engages us in some witty banter, which we don't
// actually care about for the purposes of this program. But
// we need to read past it, and acknowledge it anyway. We are
// acting as a server here. We don't need to change the client
// data at all.
//
ByteBuffer subscription_request = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
readMessage(subscription_request, sc);
writeServerCallResponse(sc, getRequestClientData(subscription_request));
ByteBuffer stats_request = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
readMessage(stats_request, sc);
writeServerCallResponse(sc, getRequestClientData(stats_request));
ByteBuffer syscat_request = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
readMessage(syscat_request, sc);
writeServerCallResponse(sc, getRequestClientData(stats_request));
//
// Now, read the invocation requests from the client. We can't
// actually respond, so we fake up a response. But this is good
// enough for now, and we save the message.
//
String[] vanillaFileNames = new String[] { "invocation_request_success.msg", "invocation_request_fail_cv.msg", "invocation_request_select.msg" };
Map<String, ByteBuffer> vanillaMessages = new HashMap<String, ByteBuffer>();
for (String fileName : vanillaFileNames) {
ByteBuffer responseMessage = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
vanillaMessages.put(fileName, responseMessage);
readMessage(responseMessage, sc);
writeServerCallResponse(sc, getRequestClientData(responseMessage));
// Set the client data. The value here is not important, but it
// needs to be shared between this and the client unit tests.
setRequestClientData(responseMessage, clientData);
writeDataFile(clientDataDir, fileName, responseMessage);
}
// Note that these names are somewhat stylized. They name
// the file which holds the request. The response to this
// request will be in a similarly named file, but with _request_
// replaced by _response_. So, make sure there is one _request_
// substring in the file names.
String[] geoFileNames = new String[] { "invocation_request_insert_geo.msg", "invocation_request_insert_geo_nulls.msg", "invocation_request_select_geo_both.msg", "invocation_request_select_geo_both_mid.msg", "invocation_request_select_geo_polynull.msg", "invocation_request_select_geo_ptnull.msg", "invocation_request_select_geo_bothnull.msg" };
Map<String, ByteBuffer> geoMessages = new HashMap<String, ByteBuffer>();
for (String filename : geoFileNames) {
ByteBuffer requestMessage = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
// We need to save these for later.
geoMessages.put(filename, requestMessage);
readMessage(requestMessage, sc);
writeServerCallResponse(sc, getRequestClientData(requestMessage));
setRequestClientData(requestMessage, clientData);
if (generateGeoMessages) {
writeDataFile(clientDataDir, filename, requestMessage);
}
}
oclient.close();
ssc.close();
oclientThread.join();
// Now, connect to a real server. We are going to pretend to be a
// client and write the messages we just read from the client, as we pretended to be
// a server. We will then capture the responses in files.
SocketChannel voltsc = null;
try {
voltsc = SocketChannel.open(new InetSocketAddress("localhost", TRUE_SERVER_PORT));
voltsc.socket().setTcpNoDelay(true);
voltsc.configureBlocking(true);
System.err.printf("Connected.\n");
} catch (IOException ex) {
abend("Can't connect to a server. Is there a VoltDB server running?.\n");
}
// Write the authentication message and then
// read the response. We need the response. The
// Client will engage in witty repartee with the
// server, but we neither see nor care about that.
//
// Note that for each of these responses we need to
// set some parameters, so that they will not depend
// on the particular context we executed. This is the
// cluster start time, the client data, the leader IP
// address and the build string. The client unit tests
// will know these values.
//
ByteBuffer scratch = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
voltsc.write(authReqSHA1);
readMessage(scratch, voltsc);
setClusterStartTimestamp(scratch, clusterStartTime);
setLeaderIPAddr(scratch, leaderIPAddr);
setBuildString(scratch, buildString);
writeDataFile(clientDataDir, "authentication_response.msg", scratch);
for (String filename : vanillaFileNames) {
// Write the three procedure messages.
ByteBuffer requestMessage = vanillaMessages.get(filename);
if (requestMessage == null) {
abend("Cannot find request message for file name \"%s\"\n", filename);
}
voltsc.write(requestMessage);
readMessage(scratch, voltsc);
setResponseClientData(scratch, clientData);
setClusterRoundTripTime(scratch, clusterRoundTripTime);
String responseFileName = filename.replaceAll("_request_", "_response_");
writeDataFile(clientDataDir, responseFileName, scratch);
}
if (generateGeoMessages) {
for (String filename : geoFileNames) {
// Write the three procedure messages.
ByteBuffer requestMessage = geoMessages.get(filename);
if (requestMessage == null) {
abend("Cannot find request message for file name \"%s\"\n", filename);
}
voltsc.write(requestMessage);
readMessage(scratch, voltsc);
setResponseClientData(scratch, clientData);
setClusterRoundTripTime(scratch, clusterRoundTripTime);
String responseFileName = filename.replaceAll("_request_", "_response_");
System.out.printf("Writing Response file \"%s\".\n", responseFileName);
writeDataFile(clientDataDir, responseFileName, scratch);
}
}
voltsc.close();
clientThread.join();
Thread.sleep(3000);
ssc = ServerSocketChannel.open();
ssc.socket().bind(new InetSocketAddress("localhost", FAKE_SERVER_PORT));
clientThread = new Thread() {
@Override
public void run() {
try {
org.voltdb.client.Client newClient = ClientFactory.createClient();
newClient.createConnection("localhost", FAKE_SERVER_PORT);
String[] strings = new String[] { "oh", "noes" };
byte[] bytes = new byte[] { 22, 33, 44 };
short[] shorts = new short[] { 22, 33, 44 };
int[] ints = new int[] { 22, 33, 44 };
long[] longs = new long[] { 22, 33, 44 };
double[] doubles = new double[] { 3, 3.1, 3.14, 3.1459 };
TimestampType[] timestamps = new TimestampType[] { new TimestampType(33), new TimestampType(44) };
BigDecimal[] bds = new BigDecimal[] { new BigDecimal("3"), new BigDecimal("3.14"), new BigDecimal("3.1459") };
try {
newClient.callProcedure("foo", strings, bytes, shorts, ints, longs, doubles, timestamps, bds, null, "ohnoes!", (byte) 22, (short) 22, 22, (long) 22, 3.1459, new TimestampType(33), new BigDecimal("3.1459"));
} catch (Exception e) {
}
} catch (Exception e) {
e.printStackTrace();
}
}
};
clientThread.setDaemon(true);
clientThread.start();
voltsc = ssc.accept();
// Read the authentication message. We don't need it.
readMessage(scratch, voltsc);
writeServerAuthenticationResponse(voltsc, true);
//
// The client engages us in some dialog. We don't need this
// either, but we need to read past it.
//
subscription_request = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
readMessage(subscription_request, voltsc);
writeServerCallResponse(voltsc, getRequestClientData(subscription_request));
stats_request = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
readMessage(stats_request, voltsc);
writeServerCallResponse(voltsc, getRequestClientData(stats_request));
syscat_request = ByteBuffer.allocate(DEFAULT_BUFFER_SIZE);
readMessage(syscat_request, voltsc);
writeServerCallResponse(voltsc, getRequestClientData(stats_request));
// Read the all-types call procedure message.
readMessage(scratch, voltsc);
writeServerCallResponse(voltsc, getRequestClientData(scratch));
setRequestClientData(scratch, clientData);
writeDataFile(clientDataDir, "invocation_request_all_params.msg", scratch);
voltsc.close();
clientThread.join();
//
// Serialize a message and write it.
//
ColumnInfo[] columns = new ColumnInfo[] { new ColumnInfo("column1", VoltType.TINYINT), new ColumnInfo("column2", VoltType.STRING), new ColumnInfo("column3", VoltType.SMALLINT), new ColumnInfo("column4", VoltType.INTEGER), new ColumnInfo("column5", VoltType.BIGINT), new ColumnInfo("column6", VoltType.TIMESTAMP), new ColumnInfo("column7", VoltType.DECIMAL), new ColumnInfo("column8", VoltType.GEOGRAPHY), new ColumnInfo("column9", VoltType.GEOGRAPHY_POINT) };
VoltTable vt = new VoltTable(columns);
GeographyValue poly = GeographyValue.fromWKT(smallPolyTxt);
GeographyPointValue pt = GeographyPointValue.fromWKT(smallPointTxt);
vt.addRow(null, null, null, null, null, null, null, poly, pt);
vt.addRow(0, "", 2, 4, 5, new TimestampType(44), new BigDecimal("3.1459"), poly, pt);
vt.addRow(0, null, 2, 4, 5, null, null, poly, pt);
vt.addRow(null, "woobie", null, null, null, new TimestampType(44), new BigDecimal("3.1459"), poly, pt);
ByteBuffer bb = ByteBuffer.allocate(vt.getSerializedSize());
vt.flattenToBuffer(bb);
FastSerializer fs = new FastSerializer(vt.getSerializedSize());
fs.write(bb);
bb.flip();
writeDataFile(clientDataDir, "serialized_table.bin", bb);
clientThread.join();
}
use of org.voltdb.VoltTable.ColumnInfo in project voltdb by VoltDB.
the class ReplicatedProcedure method run.
public VoltTable run(long id, String desc) {
if (id != 1 || !desc.equals("haha")) {
throw new VoltAbortException();
}
VoltTable result = new VoltTable(new ColumnInfo("txnId", VoltType.BIGINT), new ColumnInfo("timestamp", VoltType.BIGINT));
result.addRow(DeprecatedProcedureAPIAccess.getVoltPrivateRealTransactionId(this), getUniqueId());
// replicated txns get their results replaced by a hash... so stash this here
setAppStatusString(result.toJSONString());
return result;
}
use of org.voltdb.VoltTable.ColumnInfo in project voltdb by VoltDB.
the class TestEmptySchema method testEmptySchema.
public void testEmptySchema() throws Exception {
final Client client = getClient();
// sleep a little so that we have time for the IPC backend to actually be running
// so it can screw us on empty results
Thread.sleep(1000);
// Even running should be an improvement (ENG-4645), but do something just to be sure
// Also, check to be sure we get a full schema for the table and index stats
ColumnInfo[] expectedSchema = new ColumnInfo[13];
expectedSchema[0] = new ColumnInfo("TIMESTAMP", VoltType.BIGINT);
expectedSchema[1] = new ColumnInfo("HOST_ID", VoltType.INTEGER);
expectedSchema[2] = new ColumnInfo("HOSTNAME", VoltType.STRING);
expectedSchema[3] = new ColumnInfo("SITE_ID", VoltType.INTEGER);
expectedSchema[4] = new ColumnInfo("PARTITION_ID", VoltType.BIGINT);
expectedSchema[5] = new ColumnInfo("TABLE_NAME", VoltType.STRING);
expectedSchema[6] = new ColumnInfo("TABLE_TYPE", VoltType.STRING);
expectedSchema[7] = new ColumnInfo("TUPLE_COUNT", VoltType.BIGINT);
expectedSchema[8] = new ColumnInfo("TUPLE_ALLOCATED_MEMORY", VoltType.BIGINT);
expectedSchema[9] = new ColumnInfo("TUPLE_DATA_MEMORY", VoltType.BIGINT);
expectedSchema[10] = new ColumnInfo("STRING_DATA_MEMORY", VoltType.BIGINT);
expectedSchema[11] = new ColumnInfo("TUPLE_LIMIT", VoltType.INTEGER);
expectedSchema[12] = new ColumnInfo("PERCENT_FULL", VoltType.INTEGER);
VoltTable expectedTable = new VoltTable(expectedSchema);
VoltTable[] results = client.callProcedure("@Statistics", "TABLE", 0).getResults();
System.out.println("TABLE RESULTS: " + results[0]);
assertEquals(0, results[0].getRowCount());
assertEquals(expectedSchema.length, results[0].getColumnCount());
validateSchema(results[0], expectedTable);
expectedSchema = new ColumnInfo[12];
expectedSchema[0] = new ColumnInfo("TIMESTAMP", VoltType.BIGINT);
expectedSchema[1] = new ColumnInfo("HOST_ID", VoltType.INTEGER);
expectedSchema[2] = new ColumnInfo("HOSTNAME", VoltType.STRING);
expectedSchema[3] = new ColumnInfo("SITE_ID", VoltType.INTEGER);
expectedSchema[4] = new ColumnInfo("PARTITION_ID", VoltType.BIGINT);
expectedSchema[5] = new ColumnInfo("INDEX_NAME", VoltType.STRING);
expectedSchema[6] = new ColumnInfo("TABLE_NAME", VoltType.STRING);
expectedSchema[7] = new ColumnInfo("INDEX_TYPE", VoltType.STRING);
expectedSchema[8] = new ColumnInfo("IS_UNIQUE", VoltType.TINYINT);
expectedSchema[9] = new ColumnInfo("IS_COUNTABLE", VoltType.TINYINT);
expectedSchema[10] = new ColumnInfo("ENTRY_COUNT", VoltType.BIGINT);
expectedSchema[11] = new ColumnInfo("MEMORY_ESTIMATE", VoltType.BIGINT);
expectedTable = new VoltTable(expectedSchema);
results = client.callProcedure("@Statistics", "INDEX", 0).getResults();
System.out.println("INDEX RESULTS: " + results[0]);
assertEquals(0, results[0].getRowCount());
assertEquals(expectedSchema.length, results[0].getColumnCount());
validateSchema(results[0], expectedTable);
}
use of org.voltdb.VoltTable.ColumnInfo in project voltdb by VoltDB.
the class SiteStatsSource method populateColumnSchema.
@Override
protected void populateColumnSchema(ArrayList<ColumnInfo> columns) {
super.populateColumnSchema(columns);
columns.add(new ColumnInfo(VoltSystemProcedure.CNAME_SITE_ID, VoltSystemProcedure.CTYPE_ID));
}
use of org.voltdb.VoltTable.ColumnInfo in project voltdb by VoltDB.
the class SnapshotScanAgent method constructDigestResultsTable.
private VoltTable constructDigestResultsTable() {
ColumnInfo[] result_columns = new ColumnInfo[6];
int ii = 0;
result_columns[ii++] = new ColumnInfo(VoltSystemProcedure.CNAME_HOST_ID, VoltSystemProcedure.CTYPE_ID);
result_columns[ii++] = new ColumnInfo("PATH", VoltType.STRING);
result_columns[ii++] = new ColumnInfo("NAME", VoltType.STRING);
result_columns[ii++] = new ColumnInfo("TABLES", VoltType.STRING);
result_columns[ii++] = new ColumnInfo("RESULT", VoltType.STRING);
result_columns[ii++] = new ColumnInfo("ERR_MSG", VoltType.STRING);
return new VoltTable(result_columns);
}
Aggregations