/* This file is part of VoltDB.
* Copyright (C) 2008-2017 VoltDB Inc.
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
package txnIdSelfCheck.procedures;
import org.voltdb.SQLStmt;
import org.voltdb.VoltProcedure;
import org.voltdb.VoltTable;
public class CopyLoadPartitionedBase extends VoltProcedure {
public VoltTable[] doWork(SQLStmt select, SQLStmt insert, long cid) {
if (select != null) { // use select / insert
// Get row for cid and copy to new table.
voltQueueSQL(select, cid);
VoltTable[] results = voltExecuteSQL();
VoltTable data = results[0];
if (data.getRowCount() != 1) {
throw new VoltAbortException("Failed to find cid that should exist: cid=" + cid);
}
data.advanceRow();
long rcid = data.getLong(0);
if (rcid != cid) {
throw new VoltAbortException("Failed to find cid does not match. (" + rcid + ":" + cid + ")");
}
long txnid = data.getLong(1);
long rowid = data.getLong(2);
voltQueueSQL(insert, rcid, txnid, rowid);
return voltExecuteSQL();
} else { // use insert into
voltQueueSQL(insert,cid);
VoltTable[] results = voltExecuteSQL();
VoltTable data = results[0];
int cnt = (int) data.fetchRow(0).getLong(0);
if (cnt != 1) {
throw new VoltAbortException("incorrect number of inserted rows=" + cnt + " for cid=" + cid);
}
return results;
}
}
public long run() {
return 0; // never called in base procedure
}
}