+#include "Table.h"
+#include "CloudComm.h"
+#include "SlotBuffer.h"
+#include "NewKey.h"
+#include "Slot.h"
+#include "KeyValue.h"
+#include "Error.h"
+#include "PendingTransaction.h"
+#include "TableStatus.h"
+#include "TransactionStatus.h"
+#include "Transaction.h"
+#include "LastMessage.h"
+#include "Random.h"
+#include "ByteBuffer.h"
+#include "Abort.h"
+#include "CommitPart.h"
+#include "ArbitrationRound.h"
+#include "TransactionPart.h"
+#include "Commit.h"
+#include "RejectedMessage.h"
+#include "SlotIndexer.h"
+#include <stdlib.h>
+
+int compareInt64(const void * a, const void *b) {
+ const int64_t * pa = (const int64_t *) a;
+ const int64_t * pb = (const int64_t *) b;
+ if (*pa < *pb)
+ return -1;
+ else if (*pa > *pb)
+ return 1;
+ else
+ return 0;
+}
+
+Table::Table(IoTString *baseurl, IoTString *password, int64_t _localMachineId, int listeningPort) :
+ buffer(NULL),
+ cloud(new CloudComm(this, baseurl, password, listeningPort)),
+ random(NULL),
+ liveTableStatus(NULL),
+ pendingTransactionBuilder(NULL),
+ lastPendingTransactionSpeculatedOn(NULL),
+ firstPendingTransaction(NULL),
+ numberOfSlots(0),
+ bufferResizeThreshold(0),
+ liveSlotCount(0),
+ oldestLiveSlotSequenceNumver(1),
+ localMachineId(_localMachineId),
+ sequenceNumber(0),
+ localTransactionSequenceNumber(0),
+ lastTransactionSequenceNumberSpeculatedOn(0),
+ oldestTransactionSequenceNumberSpeculatedOn(0),
+ localArbitrationSequenceNumber(0),
+ hadPartialSendToServer(false),
+ attemptedToSendToServer(false),
+ expectedsize(0),
+ didFindTableStatus(false),
+ currMaxSize(0),
+ lastSlotAttemptedToSend(NULL),
+ lastIsNewKey(false),
+ lastNewSize(0),
+ lastTransactionPartsSent(NULL),
+ lastPendingSendArbitrationEntriesToDelete(NULL),
+ lastNewKey(NULL),
+ committedKeyValueTable(NULL),
+ speculatedKeyValueTable(NULL),
+ pendingTransactionSpeculatedKeyValueTable(NULL),
+ liveNewKeyTable(NULL),
+ lastMessageTable(NULL),
+ rejectedMessageWatchVectorTable(NULL),
+ arbitratorTable(NULL),
+ liveAbortTable(NULL),
+ newTransactionParts(NULL),
+ newCommitParts(NULL),
+ lastArbitratedTransactionNumberByArbitratorTable(NULL),
+ liveTransactionBySequenceNumberTable(NULL),
+ liveTransactionByTransactionIdTable(NULL),
+ liveCommitsTable(NULL),
+ liveCommitsByKeyTable(NULL),
+ lastCommitSeenSequenceNumberByArbitratorTable(NULL),
+ rejectedSlotVector(NULL),
+ pendingTransactionQueue(NULL),
+ pendingSendArbitrationRounds(NULL),
+ pendingSendArbitrationEntriesToDelete(NULL),
+ transactionPartsSent(NULL),
+ outstandingTransactionStatus(NULL),
+ liveAbortsGeneratedByLocal(NULL),
+ offlineTransactionsCommittedAndAtServer(NULL),
+ localCommunicationTable(NULL),
+ lastTransactionSeenFromMachineFromServer(NULL),
+ lastArbitrationDataLocalSequenceNumberSeenFromArbitrator(NULL),
+ lastInsertedNewKey(false),
+ lastSeqNumArbOn(0)
+{
+ init();
+}
+Table::Table(CloudComm *_cloud, int64_t _localMachineId) :
+ buffer(NULL),
+ cloud(_cloud),
+ random(NULL),
+ liveTableStatus(NULL),
+ pendingTransactionBuilder(NULL),
+ lastPendingTransactionSpeculatedOn(NULL),
+ firstPendingTransaction(NULL),
+ numberOfSlots(0),
+ bufferResizeThreshold(0),
+ liveSlotCount(0),
+ oldestLiveSlotSequenceNumver(1),
+ localMachineId(_localMachineId),
+ sequenceNumber(0),
+ localTransactionSequenceNumber(0),
+ lastTransactionSequenceNumberSpeculatedOn(0),
+ oldestTransactionSequenceNumberSpeculatedOn(0),
+ localArbitrationSequenceNumber(0),
+ hadPartialSendToServer(false),
+ attemptedToSendToServer(false),
+ expectedsize(0),
+ didFindTableStatus(false),
+ currMaxSize(0),
+ lastSlotAttemptedToSend(NULL),
+ lastIsNewKey(false),
+ lastNewSize(0),
+ lastTransactionPartsSent(NULL),
+ lastPendingSendArbitrationEntriesToDelete(NULL),
+ lastNewKey(NULL),
+ committedKeyValueTable(NULL),
+ speculatedKeyValueTable(NULL),
+ pendingTransactionSpeculatedKeyValueTable(NULL),
+ liveNewKeyTable(NULL),
+ lastMessageTable(NULL),
+ rejectedMessageWatchVectorTable(NULL),
+ arbitratorTable(NULL),
+ liveAbortTable(NULL),
+ newTransactionParts(NULL),
+ newCommitParts(NULL),
+ lastArbitratedTransactionNumberByArbitratorTable(NULL),
+ liveTransactionBySequenceNumberTable(NULL),
+ liveTransactionByTransactionIdTable(NULL),
+ liveCommitsTable(NULL),
+ liveCommitsByKeyTable(NULL),
+ lastCommitSeenSequenceNumberByArbitratorTable(NULL),
+ rejectedSlotVector(NULL),
+ pendingTransactionQueue(NULL),
+ pendingSendArbitrationRounds(NULL),
+ pendingSendArbitrationEntriesToDelete(NULL),
+ transactionPartsSent(NULL),
+ outstandingTransactionStatus(NULL),
+ liveAbortsGeneratedByLocal(NULL),
+ offlineTransactionsCommittedAndAtServer(NULL),
+ localCommunicationTable(NULL),
+ lastTransactionSeenFromMachineFromServer(NULL),
+ lastArbitrationDataLocalSequenceNumberSeenFromArbitrator(NULL),
+ lastInsertedNewKey(false),
+ lastSeqNumArbOn(0)
+{
+ init();
+}
/**
- * IoTTable data structure. Provides client interface.
- * @author Brian Demsky
- * @version 1.0
+ * Init all the stuff needed for for table usage
*/
+void Table::init() {
+ // Init helper objects
+ random = new Random();
+ buffer = new SlotBuffer();
+
+ // init data structs
+ committedKeyValueTable = new Hashtable<IoTString *, KeyValue *>();
+ speculatedKeyValueTable = new Hashtable<IoTString *, KeyValue *>();
+ pendingTransactionSpeculatedKeyValueTable = new Hashtable<IoTString *, KeyValue *>();
+ liveNewKeyTable = new Hashtable<IoTString *, NewKey *>();
+ lastMessageTable = new Hashtable<int64_t, Pair<int64_t, Liveness *> * >();
+ rejectedMessageWatchVectorTable = new Hashtable<int64_t, Hashset<RejectedMessage *> * >();
+ arbitratorTable = new Hashtable<IoTString *, int64_t>();
+ liveAbortTable = new Hashtable<Pair<int64_t, int64_t> *, Abort *, uintptr_t, 0, pairHashFunction, pairEquals>();
+ newTransactionParts = new Hashtable<int64_t, Hashtable<Pair<int64_t, int32_t> *, TransactionPart *, uintptr_t, 0, pairHashFunction, pairEquals> *>();
+ newCommitParts = new Hashtable<int64_t, Hashtable<Pair<int64_t, int32_t> *, CommitPart *, uintptr_t, 0, pairHashFunction, pairEquals> *>();
+ lastArbitratedTransactionNumberByArbitratorTable = new Hashtable<int64_t, int64_t>();
+ liveTransactionBySequenceNumberTable = new Hashtable<int64_t, Transaction *>();
+ liveTransactionByTransactionIdTable = new Hashtable<Pair<int64_t, int64_t> *, Transaction *, uintptr_t, 0, pairHashFunction, pairEquals>();
+ liveCommitsTable = new Hashtable<int64_t, Hashtable<int64_t, Commit *> * >();
+ liveCommitsByKeyTable = new Hashtable<IoTString *, Commit *>();
+ lastCommitSeenSequenceNumberByArbitratorTable = new Hashtable<int64_t, int64_t>();
+ rejectedSlotVector = new Vector<int64_t>();
+ pendingTransactionQueue = new Vector<Transaction *>();
+ pendingSendArbitrationEntriesToDelete = new Vector<Entry *>();
+ transactionPartsSent = new Hashtable<Transaction *, Vector<int32_t> *>();
+ outstandingTransactionStatus = new Hashtable<int64_t, TransactionStatus *>();
+ liveAbortsGeneratedByLocal = new Hashtable<int64_t, Abort *>();
+ offlineTransactionsCommittedAndAtServer = new Hashset<Pair<int64_t, int64_t> *, uintptr_t, 0, pairHashFunction, pairEquals>();
+ localCommunicationTable = new Hashtable<int64_t, Pair<IoTString *, int32_t> *>();
+ lastTransactionSeenFromMachineFromServer = new Hashtable<int64_t, int64_t>();
+ pendingSendArbitrationRounds = new Vector<ArbitrationRound *>();
+ lastArbitrationDataLocalSequenceNumberSeenFromArbitrator = new Hashtable<int64_t, int64_t>();
+
+ // Other init stuff
+ numberOfSlots = buffer->capacity();
+ setResizeThreshold();
+}
-final class Table {
-
- /* Constants */
- static final int FREE_SLOTS = 2; // Number of slots that should be kept free // 10
- static final int SKIP_THRESHOLD = 10;
- static final double RESIZE_MULTIPLE = 1.2;
- static final double RESIZE_THRESHOLD = 0.75;
- static final int REJECTED_THRESHOLD = 5;
-
- /* Helper Objects */
- SlotBuffer buffer = NULL;
- CloudComm cloud = NULL;
- Random random = NULL;
- TableStatus liveTableStatus = NULL;
- PendingTransaction pendingTransactionBuilder = NULL; // Pending Transaction used in building a Pending Transaction
- Transaction lastPendingTransactionSpeculatedOn = NULL; // Last transaction that was speculated on from the pending transaction
- Transaction firstPendingTransaction = NULL; // first transaction in the pending transaction list
-
- /* Variables */
- int numberOfSlots = 0; // Number of slots stored in buffer
- int bufferResizeThreshold = 0; // Threshold on the number of live slots before a resize is needed
- int64_t liveSlotCount = 0; // Number of currently live slots
- int64_t oldestLiveSlotSequenceNumver = 0; // Smallest sequence number of the slot with a live entry
- int64_t localMachineId = 0; // Machine ID of this client device
- int64_t sequenceNumber = 0; // Largest sequence number a client has received
- int64_t localSequenceNumber = 0;
-
- // int smallestTableStatusSeen = -1; // Smallest Table Status that was seen in the latest slots sent from the server
- // int largestTableStatusSeen = -1; // Largest Table Status that was seen in the latest slots sent from the server
- int64_t localTransactionSequenceNumber = 0; // Local sequence number counter for transactions
- int64_t lastTransactionSequenceNumberSpeculatedOn = -1; // the last transaction that was speculated on
- int64_t oldestTransactionSequenceNumberSpeculatedOn = -1; // the oldest transaction that was speculated on
- int64_t localArbitrationSequenceNumber = 0;
- bool hadPartialSendToServer = false;
- bool attemptedToSendToServer = false;
- int64_t expectedsize;
- bool didFindTableStatus = false;
- int64_t currMaxSize = 0;
-
- Slot lastSlotAttemptedToSend = NULL;
- bool lastIsNewKey = false;
- int lastNewSize = 0;
- Hashtable<Transaction, List<Integer>> lastTransactionPartsSent = NULL;
- List<Entry> lastPendingSendArbitrationEntriesToDelete = NULL;
- NewKey lastNewKey = NULL;
-
-
- /* Data Structures */
- Hashtable<IoTString, KeyValue> committedKeyValueTable = NULL; // Table of committed key value pairs
- Hashtable<IoTString, KeyValue> speculatedKeyValueTable = NULL; // Table of speculated key value pairs, if there is a speculative value
- Hashtable<IoTString, KeyValue> pendingTransactionSpeculatedKeyValueTable = NULL; // Table of speculated key value pairs, if there is a speculative value from the pending transactions
- Hashtable<IoTString, NewKey> liveNewKeyTable = NULL; // Table of live new keys
- Hashtable<Long, Pair<Long, Liveness>> lastMessageTable = NULL; // Last message sent by a client machine id -> (Seq Num, Slot or LastMessage);
- Hashtable<Long, HashSet<RejectedMessage>> rejectedMessageWatchListTable = NULL; // Table of machine Ids and the set of rejected messages they have not seen yet
- Hashtable<IoTString, Long> arbitratorTable = NULL; // Table of keys and their arbitrators
- Hashtable<Pair<int64_t, int64_t>, Abort> liveAbortTable = NULL; // Table live abort messages
- Hashtable<Long, Hashtable<Pair<Long, Integer>, TransactionPart>> newTransactionParts = NULL; // transaction parts that are seen in this latest round of slots from the server
- Hashtable<Long, Hashtable<Pair<Long, Integer>, CommitPart>> newCommitParts = NULL; // commit parts that are seen in this latest round of slots from the server
- Hashtable<int64_t, int64_t> lastArbitratedTransactionNumberByArbitratorTable = NULL; // Last transaction sequence number that an arbitrator arbitrated on
- Hashtable<Long, Transaction> liveTransactionBySequenceNumberTable = NULL; // live transaction grouped by the sequence number
- Hashtable<Pair<int64_t, int64_t>, Transaction> liveTransactionByTransactionIdTable = NULL; // live transaction grouped by the transaction ID
- Hashtable<Long, Hashtable<Long, Commit>> liveCommitsTable = NULL;
- Hashtable<IoTString, Commit> liveCommitsByKeyTable = NULL;
- Hashtable<int64_t, int64_t> lastCommitSeenSequenceNumberByArbitratorTable = NULL;
- Vector<Long> rejectedSlotList = NULL; // List of rejected slots that have yet to be sent to the server
- List<Transaction> pendingTransactionQueue = NULL;
- List<ArbitrationRound> pendingSendArbitrationRounds = NULL;
- List<Entry> pendingSendArbitrationEntriesToDelete = NULL;
- Hashtable<Transaction, List<Integer>> transactionPartsSent = NULL;
- Hashtable<Long, TransactionStatus> outstandingTransactionStatus = NULL;
- Hashtable<Long, Abort> liveAbortsGeneratedByLocal = NULL;
- Set<Pair<int64_t, int64_t>> offlineTransactionsCommittedAndAtServer = NULL;
- Hashtable<Long, Pair<String, Integer>> localCommunicationTable = NULL;
- Hashtable<int64_t, int64_t> lastTransactionSeenFromMachineFromServer = NULL;
- Hashtable<int64_t, int64_t> lastArbitrationDataLocalSequenceNumberSeenFromArbitrator = NULL;
-
-
- Table(String baseurl, String password, int64_t _localMachineId, int listeningPort) {
- localMachineId = _localMachineId;
- cloud = new CloudComm(this, baseurl, password, listeningPort);
-
- init();
- }
-
- Table(CloudComm _cloud, int64_t _localMachineId) {
- localMachineId = _localMachineId;
- cloud = _cloud;
-
- init();
- }
-
- /**
- * Init all the stuff needed for for table usage
- */
- void init() {
-
- // Init helper objects
- random = new Random();
- buffer = new SlotBuffer();
-
- // Set Variables
- oldestLiveSlotSequenceNumver = 1;
-
- // init data structs
- committedKeyValueTable = new Hashtable<IoTString, KeyValue>();
- speculatedKeyValueTable = new Hashtable<IoTString, KeyValue>();
- pendingTransactionSpeculatedKeyValueTable = new Hashtable<IoTString, KeyValue>();
- liveNewKeyTable = new Hashtable<IoTString, NewKey>();
- lastMessageTable = new Hashtable<Long, Pair<Long, Liveness>>();
- rejectedMessageWatchListTable = new Hashtable<Long, HashSet<RejectedMessage>>();
- arbitratorTable = new Hashtable<IoTString, Long>();
- liveAbortTable = new Hashtable<Pair<int64_t, int64_t>, Abort>();
- newTransactionParts = new Hashtable<Long, Hashtable<Pair<Long, Integer>, TransactionPart>>();
- newCommitParts = new Hashtable<Long, Hashtable<Pair<Long, Integer>, CommitPart>>();
- lastArbitratedTransactionNumberByArbitratorTable = new Hashtable<int64_t, int64_t>();
- liveTransactionBySequenceNumberTable = new Hashtable<Long, Transaction>();
- liveTransactionByTransactionIdTable = new Hashtable<Pair<int64_t, int64_t>, Transaction>();
- liveCommitsTable = new Hashtable<Long, Hashtable<Long, Commit>>();
- liveCommitsByKeyTable = new Hashtable<IoTString, Commit>();
- lastCommitSeenSequenceNumberByArbitratorTable = new Hashtable<int64_t, int64_t>();
- rejectedSlotList = new Vector<Long>();
- pendingTransactionQueue = new ArrayList<Transaction>();
- pendingSendArbitrationEntriesToDelete = new ArrayList<Entry>();
- transactionPartsSent = new Hashtable<Transaction, List<Integer>>();
- outstandingTransactionStatus = new Hashtable<Long, TransactionStatus>();
- liveAbortsGeneratedByLocal = new Hashtable<Long, Abort>();
- offlineTransactionsCommittedAndAtServer = new HashSet<Pair<int64_t, int64_t>>();
- localCommunicationTable = new Hashtable<Long, Pair<String, Integer>>();
- lastTransactionSeenFromMachineFromServer = new Hashtable<int64_t, int64_t>();
- pendingSendArbitrationRounds = new ArrayList<ArbitrationRound>();
- lastArbitrationDataLocalSequenceNumberSeenFromArbitrator = new Hashtable<int64_t, int64_t>();
-
-
- // Other init stuff
- numberOfSlots = buffer.capacity();
- setResizeThreshold();
- }
-
- // TODO: delete method
- synchronized void printSlots() {
- int64_t o = buffer.getOldestSeqNum();
- int64_t n = buffer.getNewestSeqNum();
-
- int[] types = new int[10];
-
- int num = 0;
-
- int livec = 0;
- int deadc = 0;
-
- int casdasd = 0;
-
- int liveslo = 0;
-
- for (int64_t i = o; i < (n + 1); i++) {
- Slot s = buffer.getSlot(i);
-
-
- if (s.isLive()) {
- liveslo++;
- }
-
- Vector<Entry> entries = s.getEntries();
-
- for (Entry e : entries) {
- if (e.isLive()) {
- int type = e.getType();
-
-
- if (type == 6) {
- RejectedMessage rej = (RejectedMessage)e;
- casdasd++;
-
- System.out.println(rej.getMachineID());
- }
-
-
- types[type] = types[type] + 1;
- num++;
- livec++;
- } else {
- deadc++;
- }
- }
- }
+/**
+ * Initialize the table by inserting a table status as the first entry
+ * into the table status also initialize the crypto stuff.
+ */
+void Table::initTable() {
+ cloud->initSecurity();
+
+ // Create the first insertion into the block chain which is the table status
+ Slot *s = new Slot(this, 1, localMachineId, localSequenceNumber);
+ localSequenceNumber++;
+ TableStatus *status = new TableStatus(s, numberOfSlots);
+ s->addEntry(status);
+ Array<Slot *> *array = cloud->putSlot(s, numberOfSlots);
+
+ if (array == NULL) {
+ array = new Array<Slot *>(1);
+ array->set(0, s);
+ // update local block chain
+ validateAndUpdate(array, true);
+ } else if (array->length() == 1) {
+ // in case we did push the slot BUT we failed to init it
+ validateAndUpdate(array, true);
+ } else {
+ throw new Error("Error on initialization");
+ }
+}
- for (int i = 0; i < 10; i++) {
- System.out.println(i + " " + types[i]);
- }
- System.out.println("Live count: " + livec);
- System.out.println("Live Slot count: " + liveslo);
+/**
+ * Rebuild the table from scratch by pulling the latest block chain
+ * from the server.
+ */
+void Table::rebuild() {
+ // Just pull the latest slots from the server
+ Array<Slot *> *newslots = cloud->getSlots(sequenceNumber + 1);
+ validateAndUpdate(newslots, true);
+ sendToServer(NULL);
+ updateLiveTransactionsAndStatus();
+}
- System.out.println("Dead count: " + deadc);
- System.out.println("Old: " + o);
- System.out.println("New: " + n);
- System.out.println("Size: " + buffer.size());
- // System.out.println("Commits: " + liveCommitsTable.size());
- System.out.println("pendingTrans: " + pendingTransactionQueue.size());
- System.out.println("Trans Status Out: " + outstandingTransactionStatus.size());
+void Table::addLocalCommunication(int64_t arbitrator, IoTString *hostName, int portNumber) {
+ localCommunicationTable->put(arbitrator, new Pair<IoTString *, int32_t>(hostName, portNumber));
+}
- for (Long k : lastArbitratedTransactionNumberByArbitratorTable.keySet()) {
- System.out.println(k + ": " + lastArbitratedTransactionNumberByArbitratorTable.get(k));
- }
+int64_t Table::getArbitrator(IoTString *key) {
+ return arbitratorTable->get(key);
+}
+void Table::close() {
+ cloud->close();
+}
- for (Long a : liveCommitsTable.keySet()) {
- for (Long b : liveCommitsTable.get(a).keySet()) {
- for (KeyValue kv : liveCommitsTable.get(a).get(b).getKeyValueUpdateSet()) {
- System.out.print(kv + " ");
- }
- System.out.print("|| ");
- }
- System.out.println();
- }
+IoTString *Table::getCommitted(IoTString *key) {
+ KeyValue *kv = committedKeyValueTable->get(key);
+ if (kv != NULL) {
+ return kv->getValue();
+ } else {
+ return NULL;
}
+}
- /**
- * Initialize the table by inserting a table status as the first entry into the table status
- * also initialize the crypto stuff.
- */
- synchronized void initTable() throws ServerException {
- cloud.initSecurity();
-
- // Create the first insertion into the block chain which is the table status
- Slot s = new Slot(this, 1, localMachineId, localSequenceNumber);
- localSequenceNumber++;
- TableStatus status = new TableStatus(s, numberOfSlots);
- s.addEntry(status);
- Slot[] array = cloud.putSlot(s, numberOfSlots);
+IoTString *Table::getSpeculative(IoTString *key) {
+ KeyValue *kv = pendingTransactionSpeculatedKeyValueTable->get(key);
- if (array == NULL) {
- array = new Slot[] {s};
- // update local block chain
- validateAndUpdate(array, true);
- } else if (array.length == 1) {
- // in case we did push the slot BUT we failed to init it
- validateAndUpdate(array, true);
- } else {
- throw new Error("Error on initialization");
- }
+ if (kv == NULL) {
+ kv = speculatedKeyValueTable->get(key);
}
- /**
- * Rebuild the table from scratch by pulling the latest block chain from the server.
- */
- synchronized void rebuild() throws ServerException {
- // Just pull the latest slots from the server
- Slot[] newslots = cloud.getSlots(sequenceNumber + 1);
- validateAndUpdate(newslots, true);
- sendToServer(NULL);
- updateLiveTransactionsAndStatus();
-
+ if (kv == NULL) {
+ kv = committedKeyValueTable->get(key);
}
- // String toString() {
- // String retString = " Committed Table: \n";
- // retString += "---------------------------\n";
- // retString += commitedTable.toString();
-
- // retString += "\n\n";
-
- // retString += " Speculative Table: \n";
- // retString += "---------------------------\n";
- // retString += speculativeTable.toString();
+ if (kv != NULL) {
+ return kv->getValue();
+ } else {
+ return NULL;
+ }
+}
- // return retString;
- // }
+IoTString *Table::getCommittedAtomic(IoTString *key) {
+ KeyValue *kv = committedKeyValueTable->get(key);
- synchronized void addLocalCommunication(int64_t arbitrator, String hostName, int portNumber) {
- localCommunicationTable.put(arbitrator, new Pair<String, Integer>(hostName, portNumber));
+ if (!arbitratorTable->contains(key)) {
+ throw new Error("Key not Found.");
}
- synchronized Long getArbitrator(IoTString key) {
- return arbitratorTable.get(key);
+ // Make sure new key value pair matches the current arbitrator
+ if (!pendingTransactionBuilder->checkArbitrator(arbitratorTable->get(key))) {
+ // TODO: Maybe not throw en error
+ throw new Error("Not all Key Values Match Arbitrator.");
}
- synchronized void close() {
- cloud.close();
+ if (kv != NULL) {
+ pendingTransactionBuilder->addKVGuard(new KeyValue(key, kv->getValue()));
+ return kv->getValue();
+ } else {
+ pendingTransactionBuilder->addKVGuard(new KeyValue(key, NULL));
+ return NULL;
}
+}
- synchronized IoTString getCommitted(IoTString key) {
- KeyValue kv = committedKeyValueTable.get(key);
-
- if (kv != NULL) {
- return kv.getValue();
- } else {
- return NULL;
- }
+IoTString *Table::getSpeculativeAtomic(IoTString *key) {
+ if (!arbitratorTable->contains(key)) {
+ throw new Error("Key not Found.");
}
- synchronized IoTString getSpeculative(IoTString key) {
- KeyValue kv = pendingTransactionSpeculatedKeyValueTable.get(key);
-
- if (kv == NULL) {
- kv = speculatedKeyValueTable.get(key);
- }
-
- if (kv == NULL) {
- kv = committedKeyValueTable.get(key);
- }
-
- if (kv != NULL) {
- return kv.getValue();
- } else {
- return NULL;
- }
+ // Make sure new key value pair matches the current arbitrator
+ if (!pendingTransactionBuilder->checkArbitrator(arbitratorTable->get(key))) {
+ // TODO: Maybe not throw en error
+ throw new Error("Not all Key Values Match Arbitrator.");
}
- synchronized IoTString getCommittedAtomic(IoTString key) {
- KeyValue kv = committedKeyValueTable.get(key);
-
- if (arbitratorTable.get(key) == NULL) {
- throw new Error("Key not Found.");
- }
-
- // Make sure new key value pair matches the current arbitrator
- if (!pendingTransactionBuilder.checkArbitrator(arbitratorTable.get(key))) {
- // TODO: Maybe not throw en error
- throw new Error("Not all Key Values Match Arbitrator.");
- }
+ KeyValue *kv = pendingTransactionSpeculatedKeyValueTable->get(key);
- if (kv != NULL) {
- pendingTransactionBuilder.addKVGuard(new KeyValue(key, kv.getValue()));
- return kv.getValue();
- } else {
- pendingTransactionBuilder.addKVGuard(new KeyValue(key, NULL));
- return NULL;
- }
+ if (kv == NULL) {
+ kv = speculatedKeyValueTable->get(key);
}
- synchronized IoTString getSpeculativeAtomic(IoTString key) {
- if (arbitratorTable.get(key) == NULL) {
- throw new Error("Key not Found.");
- }
-
- // Make sure new key value pair matches the current arbitrator
- if (!pendingTransactionBuilder.checkArbitrator(arbitratorTable.get(key))) {
- // TODO: Maybe not throw en error
- throw new Error("Not all Key Values Match Arbitrator.");
- }
-
- KeyValue kv = pendingTransactionSpeculatedKeyValueTable.get(key);
-
- if (kv == NULL) {
- kv = speculatedKeyValueTable.get(key);
- }
+ if (kv == NULL) {
+ kv = committedKeyValueTable->get(key);
+ }
- if (kv == NULL) {
- kv = committedKeyValueTable.get(key);
- }
+ if (kv != NULL) {
+ pendingTransactionBuilder->addKVGuard(new KeyValue(key, kv->getValue()));
+ return kv->getValue();
+ } else {
+ pendingTransactionBuilder->addKVGuard(new KeyValue(key, NULL));
+ return NULL;
+ }
+}
- if (kv != NULL) {
- pendingTransactionBuilder.addKVGuard(new KeyValue(key, kv.getValue()));
- return kv.getValue();
- } else {
- pendingTransactionBuilder.addKVGuard(new KeyValue(key, NULL));
- return NULL;
+bool Table::update() {
+ try {
+ Array<Slot *> *newSlots = cloud->getSlots(sequenceNumber + 1);
+ validateAndUpdate(newSlots, false);
+ sendToServer(NULL);
+ updateLiveTransactionsAndStatus();
+ return true;
+ } catch (Exception *e) {
+ SetIterator<int64_t, Pair<IoTString *, int32_t> *> *kit = getKeyIterator(localCommunicationTable);
+ while (kit->hasNext()) {
+ int64_t m = kit->next();
+ updateFromLocal(m);
}
+ delete kit;
}
- synchronized bool update() {
- try {
- Slot[] newSlots = cloud.getSlots(sequenceNumber + 1);
- validateAndUpdate(newSlots, false);
- sendToServer(NULL);
-
+ return false;
+}
- updateLiveTransactionsAndStatus();
+bool Table::createNewKey(IoTString *keyName, int64_t machineId) {
+ while (true) {
+ if (!arbitratorTable->contains(keyName)) {
+ // There is already an arbitrator
+ return false;
+ }
+ NewKey *newKey = new NewKey(NULL, keyName, machineId);
+ if (sendToServer(newKey)) {
+ // If successfully inserted
return true;
- } catch (Exception e) {
- // e.printStackTrace();
-
- for (Long m : localCommunicationTable.keySet()) {
- updateFromLocal(m);
- }
}
-
- return false;
}
+}
- synchronized bool createNewKey(IoTString keyName, int64_t machineId) throws ServerException {
- while (true) {
- if (arbitratorTable.get(keyName) != NULL) {
- // There is already an arbitrator
- return false;
- }
+void Table::startTransaction() {
+ // Create a new transaction, invalidates any old pending transactions.
+ pendingTransactionBuilder = new PendingTransaction(localMachineId);
+}
- NewKey newKey = new NewKey(NULL, keyName, machineId);
+void Table::addKV(IoTString *key, IoTString *value) {
- if (sendToServer(newKey)) {
- // If successfully inserted
- return true;
- }
- }
+ // Make sure it is a valid key
+ if (!arbitratorTable->contains(key)) {
+ throw new Error("Key not Found.");
}
- synchronized void startTransaction() {
- // Create a new transaction, invalidates any old pending transactions.
- pendingTransactionBuilder = new PendingTransaction(localMachineId);
+ // Make sure new key value pair matches the current arbitrator
+ if (!pendingTransactionBuilder->checkArbitrator(arbitratorTable->get(key))) {
+ // TODO: Maybe not throw en error
+ throw new Error("Not all Key Values Match Arbitrator.");
}
- synchronized void addKV(IoTString key, IoTString value) {
-
- // Make sure it is a valid key
- if (arbitratorTable.get(key) == NULL) {
- throw new Error("Key not Found.");
- }
-
- // Make sure new key value pair matches the current arbitrator
- if (!pendingTransactionBuilder.checkArbitrator(arbitratorTable.get(key))) {
- // TODO: Maybe not throw en error
- throw new Error("Not all Key Values Match Arbitrator.");
- }
+ // Add the key value to this transaction
+ KeyValue *kv = new KeyValue(key, value);
+ pendingTransactionBuilder->addKV(kv);
+}
- // Add the key value to this transaction
- KeyValue kv = new KeyValue(key, value);
- pendingTransactionBuilder.addKV(kv);
+TransactionStatus *Table::commitTransaction() {
+ if (pendingTransactionBuilder->getKVUpdates()->size() == 0) {
+ // transaction with no updates will have no effect on the system
+ return new TransactionStatus(TransactionStatus_StatusNoEffect, -1);
}
- synchronized TransactionStatus commitTransaction() {
+ // Set the local transaction sequence number and increment
+ pendingTransactionBuilder->setClientLocalSequenceNumber(localTransactionSequenceNumber);
+ localTransactionSequenceNumber++;
- if (pendingTransactionBuilder.getKVUpdates().size() == 0) {
- // transaction with no updates will have no effect on the system
- return new TransactionStatus(TransactionStatus.StatusNoEffect, -1);
- }
-
- // Set the local transaction sequence number and increment
- pendingTransactionBuilder.setClientLocalSequenceNumber(localTransactionSequenceNumber);
- localTransactionSequenceNumber++;
-
- // Create the transaction status
- TransactionStatus transactionStatus = new TransactionStatus(TransactionStatus.StatusPending, pendingTransactionBuilder.getArbitrator());
-
- // Create the new transaction
- Transaction newTransaction = pendingTransactionBuilder.createTransaction();
- newTransaction.setTransactionStatus(transactionStatus);
-
- if (pendingTransactionBuilder.getArbitrator() != localMachineId) {
- // Add it to the queue and invalidate the builder for safety
- pendingTransactionQueue.add(newTransaction);
- } else {
- arbitrateOnLocalTransaction(newTransaction);
- updateLiveStateFromLocal();
- }
+ // Create the transaction status
+ TransactionStatus *transactionStatus = new TransactionStatus(TransactionStatus_StatusPending, pendingTransactionBuilder->getArbitrator());
- pendingTransactionBuilder = new PendingTransaction(localMachineId);
+ // Create the new transaction
+ Transaction *newTransaction = pendingTransactionBuilder->createTransaction();
+ newTransaction->setTransactionStatus(transactionStatus);
- try {
- sendToServer(NULL);
- } catch (ServerException e) {
+ if (pendingTransactionBuilder->getArbitrator() != localMachineId) {
+ // Add it to the queue and invalidate the builder for safety
+ pendingTransactionQueue->add(newTransaction);
+ } else {
+ arbitrateOnLocalTransaction(newTransaction);
+ updateLiveStateFromLocal();
+ }
- Set<Long> arbitratorTriedAndFailed = new HashSet<Long>();
- for (Iterator<Transaction> iter = pendingTransactionQueue.iterator(); iter.hasNext(); ) {
- Transaction transaction = iter.next();
+ pendingTransactionBuilder = new PendingTransaction(localMachineId);
- if (arbitratorTriedAndFailed.contains(transaction.getArbitrator())) {
- // Already contacted this client so ignore all attempts to contact this client
- // to preserve ordering for arbitrator
- continue;
- }
+ try {
+ sendToServer(NULL);
+ } catch (ServerException *e) {
+
+ Hashset<int64_t> *arbitratorTriedAndFailed = new Hashset<int64_t>();
+ uint size = pendingTransactionQueue->size();
+ uint oldindex = 0;
+ for (int iter = 0; iter < size; iter++) {
+ Transaction *transaction = pendingTransactionQueue->get(iter);
+ pendingTransactionQueue->set(oldindex++, pendingTransactionQueue->get(iter));
+
+ if (arbitratorTriedAndFailed->contains(transaction->getArbitrator())) {
+ // Already contacted this client so ignore all attempts to contact this client
+ // to preserve ordering for arbitrator
+ continue;
+ }
- Pair<Boolean, Boolean> sendReturn = sendTransactionToLocal(transaction);
+ Pair<bool, bool> sendReturn = sendTransactionToLocal(transaction);
- if (sendReturn.getFirst()) {
- // Failed to contact over local
- arbitratorTriedAndFailed.add(transaction.getArbitrator());
- } else {
- // Successful contact or should not contact
+ if (sendReturn.getFirst()) {
+ // Failed to contact over local
+ arbitratorTriedAndFailed->add(transaction->getArbitrator());
+ } else {
+ // Successful contact or should not contact
- if (sendReturn.getSecond()) {
- // did arbitrate
- iter.remove();
- }
+ if (sendReturn.getSecond()) {
+ // did arbitrate
+ oldindex--;
}
}
}
-
- updateLiveStateFromLocal();
-
- return transactionStatus;
+ pendingTransactionQueue->setSize(oldindex);
}
- /**
- * Get the machine ID for this client
- */
- int64_t getMachineId() {
- return localMachineId;
- }
+ updateLiveStateFromLocal();
- /**
- * Decrement the number of live slots that we currently have
- */
- void decrementLiveCount() {
- liveSlotCount--;
- }
-
- /**
- * Recalculate the new resize threshold
- */
- void setResizeThreshold() {
- int resizeLower = (int) (RESIZE_THRESHOLD * numberOfSlots);
- bufferResizeThreshold = resizeLower - 1 + random.nextInt(numberOfSlots - resizeLower);
- }
-
- int64_t getLocalSequenceNumber() {
- return localSequenceNumber;
- }
-
-
- bool lastInsertedNewKey = false;
-
- bool sendToServer(NewKey newKey) throws ServerException {
-
- bool fromRetry = false;
-
- try {
- if (hadPartialSendToServer) {
- Slot[] newSlots = cloud.getSlots(sequenceNumber + 1);
- if (newSlots.length == 0) {
- fromRetry = true;
- ThreeTuple<Boolean, Boolean, Slot[]> sendSlotsReturn = sendSlotsToServer(lastSlotAttemptedToSend, lastNewSize, lastIsNewKey);
-
- if (sendSlotsReturn.getFirst()) {
- if (newKey != NULL) {
- if (lastInsertedNewKey && (lastNewKey.getKey() == newKey.getKey()) && (lastNewKey.getMachineID() == newKey.getMachineID())) {
- newKey = NULL;
- }
- }
-
- for (Transaction transaction : lastTransactionPartsSent.keySet()) {
- transaction.resetServerFailure();
-
- // Update which transactions parts still need to be sent
- transaction.removeSentParts(lastTransactionPartsSent.get(transaction));
-
- // Add the transaction status to the outstanding list
- outstandingTransactionStatus.put(transaction.getSequenceNumber(), transaction.getTransactionStatus());
-
- // Update the transaction status
- transaction.getTransactionStatus().setStatus(TransactionStatus.StatusSentPartial);
-
- // Check if all the transaction parts were successfully sent and if so then remove it from pending
- if (transaction.didSendAllParts()) {
- transaction.getTransactionStatus().setStatus(TransactionStatus.StatusSentFully);
- pendingTransactionQueue.remove(transaction);
- }
- }
- } else {
-
- newSlots = sendSlotsReturn.getThird();
+ return transactionStatus;
+}
- bool isInserted = false;
- for (Slot s : newSlots) {
- if ((s.getSequenceNumber() == lastSlotAttemptedToSend.getSequenceNumber()) && (s.getMachineID() == localMachineId)) {
- isInserted = true;
- break;
- }
- }
+/**
+ * Recalculate the new resize threshold
+ */
+void Table::setResizeThreshold() {
+ int resizeLower = (int) (Table_RESIZE_THRESHOLD * numberOfSlots);
+ bufferResizeThreshold = resizeLower - 1 + random->nextInt(numberOfSlots - resizeLower);
+}
- for (Slot s : newSlots) {
- if (isInserted) {
- break;
- }
+int64_t Table::getLocalSequenceNumber() {
+ return localSequenceNumber;
+}
- // Process each entry in the slot
- for (Entry entry : s.getEntries()) {
+bool Table::sendToServer(NewKey *newKey) {
+ bool fromRetry = false;
+ try {
+ if (hadPartialSendToServer) {
+ Array<Slot *> *newSlots = cloud->getSlots(sequenceNumber + 1);
+ if (newSlots->length() == 0) {
+ fromRetry = true;
+ ThreeTuple<bool, bool, Array<Slot *> *> sendSlotsReturn = sendSlotsToServer(lastSlotAttemptedToSend, lastNewSize, lastIsNewKey);
- if (entry.getType() == Entry.TypeLastMessage) {
- LastMessage lastMessage = (LastMessage)entry;
- if ((lastMessage.getMachineID() == localMachineId) && (lastMessage.getSequenceNumber() == lastSlotAttemptedToSend.getSequenceNumber())) {
- isInserted = true;
- break;
- }
- }
- }
+ if (sendSlotsReturn.getFirst()) {
+ if (newKey != NULL) {
+ if (lastInsertedNewKey && (lastNewKey->getKey() == newKey->getKey()) && (lastNewKey->getMachineID() == newKey->getMachineID())) {
+ newKey = NULL;
}
+ }
- if (isInserted) {
- if (newKey != NULL) {
- if (lastInsertedNewKey && (lastNewKey.getKey() == newKey.getKey()) && (lastNewKey.getMachineID() == newKey.getMachineID())) {
- newKey = NULL;
- }
- }
-
- for (Transaction transaction : lastTransactionPartsSent.keySet()) {
- transaction.resetServerFailure();
-
- // Update which transactions parts still need to be sent
- transaction.removeSentParts(lastTransactionPartsSent.get(transaction));
-
- // Add the transaction status to the outstanding list
- outstandingTransactionStatus.put(transaction.getSequenceNumber(), transaction.getTransactionStatus());
-
- // Update the transaction status
- transaction.getTransactionStatus().setStatus(TransactionStatus.StatusSentPartial);
+ SetIterator<Transaction *, Vector<int> *> *trit = getKeyIterator(lastTransactionPartsSent);
+ while (trit->hasNext()) {
+ Transaction *transaction = trit->next();
+ transaction->resetServerFailure();
+ // Update which transactions parts still need to be sent
+ transaction->removeSentParts(lastTransactionPartsSent->get(transaction));
+ // Add the transaction status to the outstanding list
+ outstandingTransactionStatus->put(transaction->getSequenceNumber(), transaction->getTransactionStatus());
- // Check if all the transaction parts were successfully sent and if so then remove it from pending
- if (transaction.didSendAllParts()) {
- transaction.getTransactionStatus().setStatus(TransactionStatus.StatusSentFully);
- pendingTransactionQueue.remove(transaction);
- } else {
- transaction.resetServerFailure();
- // Set the transaction sequence number back to nothing
- if (!transaction.didSendAPartToServer()) {
- transaction.setSequenceNumber(-1);
- }
- }
- }
- }
- }
+ // Update the transaction status
+ transaction->getTransactionStatus()->setStatus(TransactionStatus_StatusSentPartial);
- for (Transaction transaction : lastTransactionPartsSent.keySet()) {
- transaction.resetServerFailure();
- // Set the transaction sequence number back to nothing
- if (!transaction.didSendAPartToServer()) {
- transaction.setSequenceNumber(-1);
+ // Check if all the transaction parts were successfully
+ // sent and if so then remove it from pending
+ if (transaction->didSendAllParts()) {
+ transaction->getTransactionStatus()->setStatus(TransactionStatus_StatusSentFully);
+ pendingTransactionQueue->remove(transaction);
}
}
-
- if (sendSlotsReturn.getThird().length != 0) {
- // insert into the local block chain
- validateAndUpdate(sendSlotsReturn.getThird(), true);
- }
- // continue;
+ delete trit;
} else {
+ newSlots = sendSlotsReturn.getThird();
bool isInserted = false;
- for (Slot s : newSlots) {
- if ((s.getSequenceNumber() == lastSlotAttemptedToSend.getSequenceNumber()) && (s.getMachineID() == localMachineId)) {
+ for (uint si = 0; si < newSlots->length(); si++) {
+ Slot *s = newSlots->get(si);
+ if ((s->getSequenceNumber() == lastSlotAttemptedToSend->getSequenceNumber()) && (s->getMachineID() == localMachineId)) {
isInserted = true;
break;
}
}
- for (Slot s : newSlots) {
+ for (uint si = 0; si < newSlots->length(); si++) {
+ Slot *s = newSlots->get(si);
if (isInserted) {
break;
}
// Process each entry in the slot
- for (Entry entry : s.getEntries()) {
-
- if (entry.getType() == Entry.TypeLastMessage) {
- LastMessage lastMessage = (LastMessage)entry;
- if ((lastMessage.getMachineID() == localMachineId) && (lastMessage.getSequenceNumber() == lastSlotAttemptedToSend.getSequenceNumber())) {
+ Vector<Entry *> *ventries = s->getEntries();
+ uint vesize = ventries->size();
+ for (uint vei = 0; vei < vesize; vei++) {
+ Entry *entry = ventries->get(vei);
+ if (entry->getType() == TypeLastMessage) {
+ LastMessage *lastMessage = (LastMessage *)entry;
+ if ((lastMessage->getMachineID() == localMachineId) && (lastMessage->getSequenceNumber() == lastSlotAttemptedToSend->getSequenceNumber())) {
isInserted = true;
break;
}
if (isInserted) {
if (newKey != NULL) {
- if (lastInsertedNewKey && (lastNewKey.getKey() == newKey.getKey()) && (lastNewKey.getMachineID() == newKey.getMachineID())) {
+ if (lastInsertedNewKey && (lastNewKey->getKey() == newKey->getKey()) && (lastNewKey->getMachineID() == newKey->getMachineID())) {
newKey = NULL;
}
}
- for (Transaction transaction : lastTransactionPartsSent.keySet()) {
- transaction.resetServerFailure();
+ SetIterator<Transaction *, Vector<int> *> *trit = getKeyIterator(lastTransactionPartsSent);
+ while (trit->hasNext()) {
+ Transaction *transaction = trit->next();
+ transaction->resetServerFailure();
// Update which transactions parts still need to be sent
- transaction.removeSentParts(lastTransactionPartsSent.get(transaction));
+ transaction->removeSentParts(lastTransactionPartsSent->get(transaction));
// Add the transaction status to the outstanding list
- outstandingTransactionStatus.put(transaction.getSequenceNumber(), transaction.getTransactionStatus());
+ outstandingTransactionStatus->put(transaction->getSequenceNumber(), transaction->getTransactionStatus());
// Update the transaction status
- transaction.getTransactionStatus().setStatus(TransactionStatus.StatusSentPartial);
+ transaction->getTransactionStatus()->setStatus(TransactionStatus_StatusSentPartial);
// Check if all the transaction parts were successfully sent and if so then remove it from pending
- if (transaction.didSendAllParts()) {
- transaction.getTransactionStatus().setStatus(TransactionStatus.StatusSentFully);
- pendingTransactionQueue.remove(transaction);
+ if (transaction->didSendAllParts()) {
+ transaction->getTransactionStatus()->setStatus(TransactionStatus_StatusSentFully);
+ pendingTransactionQueue->remove(transaction);
} else {
- transaction.resetServerFailure();
+ transaction->resetServerFailure();
// Set the transaction sequence number back to nothing
- if (!transaction.didSendAPartToServer()) {
- transaction.setSequenceNumber(-1);
+ if (!transaction->didSendAPartToServer()) {
+ transaction->setSequenceNumber(-1);
}
}
}
- } else {
- for (Transaction transaction : lastTransactionPartsSent.keySet()) {
- transaction.resetServerFailure();
- // Set the transaction sequence number back to nothing
- if (!transaction.didSendAPartToServer()) {
- transaction.setSequenceNumber(-1);
- }
- }
+ delete trit;
}
-
- // insert into the local block chain
- validateAndUpdate(newSlots, true);
}
- }
- } catch (ServerException e) {
- throw e;
- }
-
-
-
- try {
- // While we have stuff that needs inserting into the block chain
- while ((pendingTransactionQueue.size() > 0) || (pendingSendArbitrationRounds.size() > 0) || (newKey != NULL)) {
-
- fromRetry = false;
-
- if (hadPartialSendToServer) {
- throw new Error("Should Be error free");
- }
-
-
-
- // If there is a new key with same name then end
- if ((newKey != NULL) && (arbitratorTable.get(newKey.getKey()) != NULL)) {
- return false;
- }
-
- // Create the slot
- Slot slot = new Slot(this, sequenceNumber + 1, localMachineId, buffer.getSlot(sequenceNumber).getHMAC(), localSequenceNumber);
- localSequenceNumber++;
-
- // Try to fill the slot with data
- ThreeTuple<Boolean, Integer, Boolean> fillSlotsReturn = fillSlot(slot, false, newKey);
- bool needsResize = fillSlotsReturn.getFirst();
- int newSize = fillSlotsReturn.getSecond();
- Boolean insertedNewKey = fillSlotsReturn.getThird();
-
- if (needsResize) {
- // Reset which transaction to send
- for (Transaction transaction : transactionPartsSent.keySet()) {
- transaction.resetNextPartToSend();
-
- // Set the transaction sequence number back to nothing
- if (!transaction.didSendAPartToServer() && !transaction.getServerFailure()) {
- transaction.setSequenceNumber(-1);
- }
- }
-
- // Clear the sent data since we are trying again
- pendingSendArbitrationEntriesToDelete.clear();
- transactionPartsSent.clear();
-
- // We needed a resize so try again
- fillSlot(slot, true, newKey);
- }
-
- lastSlotAttemptedToSend = slot;
- lastIsNewKey = (newKey != NULL);
- lastInsertedNewKey = insertedNewKey;
- lastNewSize = newSize;
- lastNewKey = newKey;
- lastTransactionPartsSent = new Hashtable<Transaction, List<Integer>>(transactionPartsSent);
- lastPendingSendArbitrationEntriesToDelete = new ArrayList<Entry>(pendingSendArbitrationEntriesToDelete);
-
-
- ThreeTuple<Boolean, Boolean, Slot[]> sendSlotsReturn = sendSlotsToServer(slot, newSize, newKey != NULL);
-
- if (sendSlotsReturn.getFirst()) {
-
- // Did insert into the block chain
-
- if (insertedNewKey) {
- // This slot was what was inserted not a previous slot
-
- // New Key was successfully inserted into the block chain so dont want to insert it again
- newKey = NULL;
- }
-
- // Remove the aborts and commit parts that were sent from the pending to send queue
- for (Iterator<ArbitrationRound> iter = pendingSendArbitrationRounds.iterator(); iter.hasNext(); ) {
- ArbitrationRound round = iter.next();
- round.removeParts(pendingSendArbitrationEntriesToDelete);
-
- if (round.isDoneSending()) {
- // Sent all the parts
- iter.remove();
- }
- }
-
- for (Transaction transaction : transactionPartsSent.keySet()) {
- transaction.resetServerFailure();
-
- // Update which transactions parts still need to be sent
- transaction.removeSentParts(transactionPartsSent.get(transaction));
-
- // Add the transaction status to the outstanding list
- outstandingTransactionStatus.put(transaction.getSequenceNumber(), transaction.getTransactionStatus());
-
- // Update the transaction status
- transaction.getTransactionStatus().setStatus(TransactionStatus.StatusSentPartial);
-
- // Check if all the transaction parts were successfully sent and if so then remove it from pending
- if (transaction.didSendAllParts()) {
- transaction.getTransactionStatus().setStatus(TransactionStatus.StatusSentFully);
- pendingTransactionQueue.remove(transaction);
- }
- }
- } else {
-
- // if (!sendSlotsReturn.getSecond()) {
- // for (Transaction transaction : lastTransactionPartsSent.keySet()) {
- // transaction.resetServerFailure();
- // }
- // } else {
- // for (Transaction transaction : lastTransactionPartsSent.keySet()) {
- // transaction.resetServerFailure();
-
- // // Update which transactions parts still need to be sent
- // transaction.removeSentParts(transactionPartsSent.get(transaction));
- // // Add the transaction status to the outstanding list
- // outstandingTransactionStatus.put(transaction.getSequenceNumber(), transaction.getTransactionStatus());
-
- // // Update the transaction status
- // transaction.getTransactionStatus().setStatus(TransactionStatus.StatusSentPartial);
-
- // // Check if all the transaction parts were successfully sent and if so then remove it from pending
- // if (transaction.didSendAllParts()) {
- // transaction.getTransactionStatus().setStatus(TransactionStatus.StatusSentFully);
- // pendingTransactionQueue.remove(transaction);
-
- // for (KeyValue kv : transaction.getKeyValueUpdateSet()) {
- // System.out.println("Sent: " + kv + " from: " + localMachineId + " Slot:" + lastSlotAttemptedToSend.getSequenceNumber() + " Claimed:" + transaction.getSequenceNumber());
- // }
- // }
- // }
- // }
-
- // Reset which transaction to send
- for (Transaction transaction : transactionPartsSent.keySet()) {
- transaction.resetNextPartToSend();
- // transaction.resetNextPartToSend();
-
- // Set the transaction sequence number back to nothing
- if (!transaction.didSendAPartToServer() && !transaction.getServerFailure()) {
- transaction.setSequenceNumber(-1);
- }
+ SetIterator<Transaction *, Vector<int> *> *trit = getKeyIterator(lastTransactionPartsSent);
+ while (trit->hasNext()) {
+ Transaction *transaction = trit->next();
+ transaction->resetServerFailure();
+ // Set the transaction sequence number back to nothing
+ if (!transaction->didSendAPartToServer()) {
+ transaction->setSequenceNumber(-1);
}
}
+ delete trit;
- // Clear the sent data in preparation for next send
- pendingSendArbitrationEntriesToDelete.clear();
- transactionPartsSent.clear();
-
- if (sendSlotsReturn.getThird().length != 0) {
+ if (sendSlotsReturn.getThird()->length() != 0) {
// insert into the local block chain
validateAndUpdate(sendSlotsReturn.getThird(), true);
}
- }
-
- } catch (ServerException e) {
-
- if (e.getType() != ServerException.TypeInputTimeout) {
- // e.printStackTrace();
-
- // Nothing was able to be sent to the server so just clear these data structures
- for (Transaction transaction : transactionPartsSent.keySet()) {
- transaction.resetNextPartToSend();
-
- // Set the transaction sequence number back to nothing
- if (!transaction.didSendAPartToServer() && !transaction.getServerFailure()) {
- transaction.setSequenceNumber(-1);
- }
- }
- } else {
- // There was a partial send to the server
- hadPartialSendToServer = true;
-
-
- // if (!fromRetry) {
- // lastTransactionPartsSent = new Hashtable<Transaction, List<Integer>>(transactionPartsSent);
- // lastPendingSendArbitrationEntriesToDelete = new ArrayList<Entry>(pendingSendArbitrationEntriesToDelete);
- // }
-
- // Nothing was able to be sent to the server so just clear these data structures
- for (Transaction transaction : transactionPartsSent.keySet()) {
- transaction.resetNextPartToSend();
- transaction.setServerFailure();
- }
- }
-
- pendingSendArbitrationEntriesToDelete.clear();
- transactionPartsSent.clear();
-
- throw e;
- }
-
- return newKey == NULL;
- }
-
- synchronized bool updateFromLocal(int64_t machineId) {
- Pair<String, Integer> localCommunicationInformation = localCommunicationTable.get(machineId);
- if (localCommunicationInformation == NULL) {
- // Cant talk to that device locally so do nothing
- return false;
- }
-
- // Get the size of the send data
- int sendDataSize = sizeof(int32_t) + sizeof(int64_t);
-
- Long lastArbitrationDataLocalSequenceNumber = (int64_t) - 1;
- if (lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.get(machineId) != NULL) {
- lastArbitrationDataLocalSequenceNumber = lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.get(machineId);
- }
-
- char[] sendData = new char[sendDataSize];
- ByteBuffer bbEncode = ByteBuffer.wrap(sendData);
-
- // Encode the data
- bbEncode.putLong(lastArbitrationDataLocalSequenceNumber);
- bbEncode.putInt(0);
-
- // Send by local
- char[] returnData = cloud.sendLocalData(sendData, localSequenceNumber, localCommunicationInformation.getFirst(), localCommunicationInformation.getSecond());
- localSequenceNumber++;
-
- if (returnData == NULL) {
- // Could not contact server
- return false;
- }
-
- // Decode the data
- ByteBuffer bbDecode = ByteBuffer.wrap(returnData);
- int numberOfEntries = bbDecode.getInt();
-
- for (int i = 0; i < numberOfEntries; i++) {
- char type = bbDecode.get();
- if (type == Entry.TypeAbort) {
- Abort abort = (Abort)Abort.decode(NULL, bbDecode);
- processEntry(abort);
- } else if (type == Entry.TypeCommitPart) {
- CommitPart commitPart = (CommitPart)CommitPart.decode(NULL, bbDecode);
- processEntry(commitPart);
- }
- }
-
- updateLiveStateFromLocal();
-
- return true;
- }
-
- Pair<Boolean, Boolean> sendTransactionToLocal(Transaction transaction) {
-
- // Get the devices local communications
- Pair<String, Integer> localCommunicationInformation = localCommunicationTable.get(transaction.getArbitrator());
-
- if (localCommunicationInformation == NULL) {
- // Cant talk to that device locally so do nothing
- return new Pair<Boolean, Boolean>(true, false);
- }
-
- // Get the size of the send data
- int sendDataSize = sizeof(int32_t) + sizeof(int64_t);
- for (TransactionPart part : transaction.getParts().values()) {
- sendDataSize += part.getSize();
- }
-
- Long lastArbitrationDataLocalSequenceNumber = (int64_t) - 1;
- if (lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.get(transaction.getArbitrator()) != NULL) {
- lastArbitrationDataLocalSequenceNumber = lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.get(transaction.getArbitrator());
- }
-
- // Make the send data size
- char[] sendData = new char[sendDataSize];
- ByteBuffer bbEncode = ByteBuffer.wrap(sendData);
-
- // Encode the data
- bbEncode.putLong(lastArbitrationDataLocalSequenceNumber);
- bbEncode.putInt(transaction.getParts().size());
- for (TransactionPart part : transaction.getParts().values()) {
- part.encode(bbEncode);
- }
-
-
- // Send by local
- char[] returnData = cloud.sendLocalData(sendData, localSequenceNumber, localCommunicationInformation.getFirst(), localCommunicationInformation.getSecond());
- localSequenceNumber++;
-
- if (returnData == NULL) {
- // Could not contact server
- return new Pair<Boolean, Boolean>(true, false);
- }
-
- // Decode the data
- ByteBuffer bbDecode = ByteBuffer.wrap(returnData);
- bool didCommit = bbDecode.get() == 1;
- bool couldArbitrate = bbDecode.get() == 1;
- int numberOfEntries = bbDecode.getInt();
- bool foundAbort = false;
-
- for (int i = 0; i < numberOfEntries; i++) {
- char type = bbDecode.get();
- if (type == Entry.TypeAbort) {
- Abort abort = (Abort)Abort.decode(NULL, bbDecode);
-
- if ((abort.getTransactionMachineId() == localMachineId) && (abort.getTransactionClientLocalSequenceNumber() == transaction.getClientLocalSequenceNumber())) {
- foundAbort = true;
- }
-
- processEntry(abort);
- } else if (type == Entry.TypeCommitPart) {
- CommitPart commitPart = (CommitPart)CommitPart.decode(NULL, bbDecode);
- processEntry(commitPart);
- }
- }
-
- updateLiveStateFromLocal();
-
- if (couldArbitrate) {
- TransactionStatus status = transaction.getTransactionStatus();
- if (didCommit) {
- status.setStatus(TransactionStatus.StatusCommitted);
- } else {
- status.setStatus(TransactionStatus.StatusAborted);
- }
- } else {
- TransactionStatus status = transaction.getTransactionStatus();
- if (foundAbort) {
- status.setStatus(TransactionStatus.StatusAborted);
- } else {
- status.setStatus(TransactionStatus.StatusCommitted);
- }
- }
-
- return new Pair<Boolean, Boolean>(false, true);
- }
-
- synchronized char[] acceptDataFromLocal(char[] data) {
-
- // Decode the data
- ByteBuffer bbDecode = ByteBuffer.wrap(data);
- int64_t lastArbitratedSequenceNumberSeen = bbDecode.getLong();
- int numberOfParts = bbDecode.getInt();
-
- // If we did commit a transaction or not
- bool didCommit = false;
- bool couldArbitrate = false;
-
- if (numberOfParts != 0) {
-
- // decode the transaction
- Transaction transaction = new Transaction();
- for (int i = 0; i < numberOfParts; i++) {
- bbDecode.get();
- TransactionPart newPart = (TransactionPart)TransactionPart.decode(NULL, bbDecode);
- transaction.addPartDecode(newPart);
- }
-
- // Arbitrate on transaction and pull relevant return data
- Pair<Boolean, Boolean> localArbitrateReturn = arbitrateOnLocalTransaction(transaction);
- couldArbitrate = localArbitrateReturn.getFirst();
- didCommit = localArbitrateReturn.getSecond();
-
- updateLiveStateFromLocal();
-
- // Transaction was sent to the server so keep track of it to prevent double commit
- if (transaction.getSequenceNumber() != -1) {
- offlineTransactionsCommittedAndAtServer.add(transaction.getId());
- }
- }
-
- // The data to send back
- int returnDataSize = 0;
- List<Entry> unseenArbitrations = new ArrayList<Entry>();
-
- // Get the aborts to send back
- List<Long> abortLocalSequenceNumbers = new ArrayList<Long >(liveAbortsGeneratedByLocal.keySet());
- Collections.sort(abortLocalSequenceNumbers);
- for (Long localSequenceNumber : abortLocalSequenceNumbers) {
- if (localSequenceNumber <= lastArbitratedSequenceNumberSeen) {
- continue;
- }
-
- Abort abort = liveAbortsGeneratedByLocal.get(localSequenceNumber);
- unseenArbitrations.add(abort);
- returnDataSize += abort.getSize();
- }
-
- // Get the commits to send back
- Hashtable<Long, Commit> commitForClientTable = liveCommitsTable.get(localMachineId);
- if (commitForClientTable != NULL) {
- List<Long> commitLocalSequenceNumbers = new ArrayList<Long>(commitForClientTable.keySet());
- Collections.sort(commitLocalSequenceNumbers);
-
- for (Long localSequenceNumber : commitLocalSequenceNumbers) {
- Commit commit = commitForClientTable.get(localSequenceNumber);
-
- if (localSequenceNumber <= lastArbitratedSequenceNumberSeen) {
- continue;
- }
-
- unseenArbitrations.addAll(commit.getParts().values());
-
- for (CommitPart commitPart : commit.getParts().values()) {
- returnDataSize += commitPart.getSize();
- }
- }
- }
-
- // Number of arbitration entries to decode
- returnDataSize += 2 * sizeof(int32_t);
-
- // Boolean of did commit or not
- if (numberOfParts != 0) {
- returnDataSize += sizeof(char);
- }
-
- // Data to send Back
- char[] returnData = new char[returnDataSize];
- ByteBuffer bbEncode = ByteBuffer.wrap(returnData);
-
- if (numberOfParts != 0) {
- if (didCommit) {
- bbEncode.put((char)1);
- } else {
- bbEncode.put((char)0);
- }
- if (couldArbitrate) {
- bbEncode.put((char)1);
+ // continue;
} else {
- bbEncode.put((char)0);
- }
- }
-
- bbEncode.putInt(unseenArbitrations.size());
- for (Entry entry : unseenArbitrations) {
- entry.encode(bbEncode);
- }
-
-
- localSequenceNumber++;
- return returnData;
- }
-
- ThreeTuple<Boolean, Boolean, Slot[]> sendSlotsToServer(Slot slot, int newSize, bool isNewKey) throws ServerException {
-
- bool attemptedToSendToServerTmp = attemptedToSendToServer;
- attemptedToSendToServer = true;
-
- bool inserted = false;
- bool lastTryInserted = false;
-
- Slot[] array = cloud.putSlot(slot, newSize);
- if (array == NULL) {
- array = new Slot[] {slot};
- rejectedSlotList.clear();
- inserted = true;
- } else {
- if (array.length == 0) {
- throw new Error("Server Error: Did not send any slots");
- }
-
- // if (attemptedToSendToServerTmp) {
- if (hadPartialSendToServer) {
-
bool isInserted = false;
- for (Slot s : array) {
- if ((s.getSequenceNumber() == slot.getSequenceNumber()) && (s.getMachineID() == localMachineId)) {
+ for (uint si = 0; si < newSlots->length(); si++) {
+ Slot *s = newSlots->get(si);
+ if ((s->getSequenceNumber() == lastSlotAttemptedToSend->getSequenceNumber()) && (s->getMachineID() == localMachineId)) {
isInserted = true;
break;
}
}
- for (Slot s : array) {
+ for (uint si = 0; si < newSlots->length(); si++) {
+ Slot *s = newSlots->get(si);
if (isInserted) {
break;
}
// Process each entry in the slot
- for (Entry entry : s.getEntries()) {
-
- if (entry.getType() == Entry.TypeLastMessage) {
- LastMessage lastMessage = (LastMessage)entry;
-
- if ((lastMessage.getMachineID() == localMachineId) && (lastMessage.getSequenceNumber() == slot.getSequenceNumber())) {
+ Vector<Entry *> *entries = s->getEntries();
+ uint eSize = entries->size();
+ for(uint ei=0; ei < eSize; ei++) {
+ Entry * entry = entries->get(ei);
+
+ if (entry->getType() == TypeLastMessage) {
+ LastMessage *lastMessage = (LastMessage *)entry;
+ if ((lastMessage->getMachineID() == localMachineId) && (lastMessage->getSequenceNumber() == lastSlotAttemptedToSend->getSequenceNumber())) {
isInserted = true;
break;
}
}
}
- if (!isInserted) {
- rejectedSlotList.add(slot.getSequenceNumber());
- lastTryInserted = false;
- } else {
- lastTryInserted = true;
- }
- } else {
- rejectedSlotList.add(slot.getSequenceNumber());
- lastTryInserted = false;
- }
- }
-
- return new ThreeTuple<Boolean, Boolean, Slot[]>(inserted, lastTryInserted, array);
- }
-
- /**
- * Returns false if a resize was needed
- */
- ThreeTuple<Boolean, Integer, Boolean> fillSlot(Slot slot, bool resize, NewKey newKeyEntry) {
-
-
- int newSize = 0;
- if (liveSlotCount > bufferResizeThreshold) {
- resize = true; //Resize is forced
-
- }
-
- if (resize) {
- newSize = (int) (numberOfSlots * RESIZE_MULTIPLE);
- TableStatus status = new TableStatus(slot, newSize);
- slot.addEntry(status);
- }
-
- // Fill with rejected slots first before doing anything else
- doRejectedMessages(slot);
-
- // Do mandatory rescue of entries
- ThreeTuple<Boolean, Boolean, Long> mandatoryRescueReturn = doMandatoryResuce(slot, resize);
-
- // Extract working variables
- bool needsResize = mandatoryRescueReturn.getFirst();
- bool seenLiveSlot = mandatoryRescueReturn.getSecond();
- int64_t currentRescueSequenceNumber = mandatoryRescueReturn.getThird();
-
- if (needsResize && !resize) {
- // We need to resize but we are not resizing so return false
- return new ThreeTuple<Boolean, Integer, Boolean>(true, NULL, NULL);
- }
-
- bool inserted = false;
- if (newKeyEntry != NULL) {
- newKeyEntry.setSlot(slot);
- if (slot.hasSpace(newKeyEntry)) {
-
- slot.addEntry(newKeyEntry);
- inserted = true;
- }
- }
-
- // Clear the transactions, aborts and commits that were sent previously
- transactionPartsSent.clear();
- pendingSendArbitrationEntriesToDelete.clear();
-
- for (ArbitrationRound round : pendingSendArbitrationRounds) {
- bool isFull = false;
- round.generateParts();
- List<Entry> parts = round.getParts();
-
- // Insert pending arbitration data
- for (Entry arbitrationData : parts) {
-
- // If it is an abort then we need to set some information
- if (arbitrationData instanceof Abort) {
- ((Abort)arbitrationData).setSequenceNumber(slot.getSequenceNumber());
- }
-
- if (!slot.hasSpace(arbitrationData)) {
- // No space so cant do anything else with these data entries
- isFull = true;
- break;
- }
-
- // Add to this current slot and add it to entries to delete
- slot.addEntry(arbitrationData);
- pendingSendArbitrationEntriesToDelete.add(arbitrationData);
- }
-
- if (isFull) {
- break;
- }
- }
-
- if (pendingTransactionQueue.size() > 0) {
-
- Transaction transaction = pendingTransactionQueue.get(0);
+ if (isInserted) {
+ if (newKey != NULL) {
+ if (lastInsertedNewKey && (lastNewKey->getKey() == newKey->getKey()) && (lastNewKey->getMachineID() == newKey->getMachineID())) {
+ newKey = NULL;
+ }
+ }
- // Set the transaction sequence number if it has yet to be inserted into the block chain
- // if ((!transaction.didSendAPartToServer() && !transaction.getServerFailure()) || (transaction.getSequenceNumber() == -1)) {
- // transaction.setSequenceNumber(slot.getSequenceNumber());
- // }
-
- if ((!transaction.didSendAPartToServer()) || (transaction.getSequenceNumber() == -1)) {
- transaction.setSequenceNumber(slot.getSequenceNumber());
- }
+ SetIterator<Transaction *, Vector<int> *> *trit = getKeyIterator(lastTransactionPartsSent);
+ while (trit->hasNext()) {
+ Transaction *transaction = trit->next();
+ transaction->resetServerFailure();
+ // Update which transactions parts still need to be sent
+ transaction->removeSentParts(lastTransactionPartsSent->get(transaction));
- while (true) {
- TransactionPart part = transaction.getNextPartToSend();
+ // Add the transaction status to the outstanding list
+ outstandingTransactionStatus->put(transaction->getSequenceNumber(), transaction->getTransactionStatus());
- if (part == NULL) {
- // Ran out of parts to send for this transaction so move on
- break;
- }
+ // Update the transaction status
+ transaction->getTransactionStatus()->setStatus(TransactionStatus_StatusSentPartial);
- if (slot.hasSpace(part)) {
- slot.addEntry(part);
- List<Integer> partsSent = transactionPartsSent.get(transaction);
- if (partsSent == NULL) {
- partsSent = new ArrayList<Integer>();
- transactionPartsSent.put(transaction, partsSent);
+ // Check if all the transaction parts were successfully sent and if so then remove it from pending
+ if (transaction->didSendAllParts()) {
+ transaction->getTransactionStatus()->setStatus(TransactionStatus_StatusSentFully);
+ pendingTransactionQueue->remove(transaction);
+ } else {
+ transaction->resetServerFailure();
+ // Set the transaction sequence number back to nothing
+ if (!transaction->didSendAPartToServer()) {
+ transaction->setSequenceNumber(-1);
+ }
+ }
}
- partsSent.add(part.getPartNumber());
- transactionPartsSent.put(transaction, partsSent);
+ delete trit;
} else {
- break;
- }
- }
- }
-
- // Fill the remainder of the slot with rescue data
- doOptionalRescue(slot, seenLiveSlot, currentRescueSequenceNumber, resize);
-
- return new ThreeTuple<Boolean, Integer, Boolean>(false, newSize, inserted);
- }
-
- void doRejectedMessages(Slot s) {
- if (! rejectedSlotList.isEmpty()) {
- /* TODO: We should avoid generating a rejected message entry if
- * there is already a sufficient entry in the queue (e.g.,
- * equalsto value of true and same sequence number). */
-
- int64_t old_seqn = rejectedSlotList.firstElement();
- if (rejectedSlotList.size() > REJECTED_THRESHOLD) {
- int64_t new_seqn = rejectedSlotList.lastElement();
- RejectedMessage rm = new RejectedMessage(s, s.getSequenceNumber(), localMachineId, old_seqn, new_seqn, false);
- s.addEntry(rm);
- } else {
- int64_t prev_seqn = -1;
- int i = 0;
- /* Go through list of missing messages */
- for (; i < rejectedSlotList.size(); i++) {
- int64_t curr_seqn = rejectedSlotList.get(i);
- Slot s_msg = buffer.getSlot(curr_seqn);
- if (s_msg != NULL)
- break;
- prev_seqn = curr_seqn;
- }
- /* Generate rejected message entry for missing messages */
- if (prev_seqn != -1) {
- RejectedMessage rm = new RejectedMessage(s, s.getSequenceNumber(), localMachineId, old_seqn, prev_seqn, false);
- s.addEntry(rm);
- }
- /* Generate rejected message entries for present messages */
- for (; i < rejectedSlotList.size(); i++) {
- int64_t curr_seqn = rejectedSlotList.get(i);
- Slot s_msg = buffer.getSlot(curr_seqn);
- int64_t machineid = s_msg.getMachineID();
- RejectedMessage rm = new RejectedMessage(s, s.getSequenceNumber(), machineid, curr_seqn, curr_seqn, true);
- s.addEntry(rm);
- }
- }
- }
- }
-
- ThreeTuple<Boolean, Boolean, Long> doMandatoryResuce(Slot slot, bool resize) {
- int64_t newestSequenceNumber = buffer.getNewestSeqNum();
- int64_t oldestSequenceNumber = buffer.getOldestSeqNum();
- if (oldestLiveSlotSequenceNumver < oldestSequenceNumber) {
- oldestLiveSlotSequenceNumver = oldestSequenceNumber;
- }
-
- int64_t currentSequenceNumber = oldestLiveSlotSequenceNumver;
- bool seenLiveSlot = false;
- int64_t firstIfFull = newestSequenceNumber + 1 - numberOfSlots; // smallest seq number in the buffer if it is full
- int64_t threshold = firstIfFull + FREE_SLOTS; // we want the buffer to be clear of live entries up to this point
-
-
- // Mandatory Rescue
- for (; currentSequenceNumber < threshold; currentSequenceNumber++) {
- Slot previousSlot = buffer.getSlot(currentSequenceNumber);
- // Push slot number forward
- if (! seenLiveSlot) {
- oldestLiveSlotSequenceNumver = currentSequenceNumber;
- }
-
- if (!previousSlot.isLive()) {
- continue;
- }
-
- // We have seen a live slot
- seenLiveSlot = true;
-
- // Get all the live entries for a slot
- Vector<Entry> liveEntries = previousSlot.getLiveEntries(resize);
-
- // Iterate over all the live entries and try to rescue them
- for (Entry liveEntry : liveEntries) {
- if (slot.hasSpace(liveEntry)) {
-
- // Enough space to rescue the entry
- slot.addEntry(liveEntry);
- } else if (currentSequenceNumber == firstIfFull) {
- //if there's no space but the entry is about to fall off the queue
- System.out.println("B"); //?
- return new ThreeTuple<Boolean, Boolean, Long>(true, seenLiveSlot, currentSequenceNumber);
-
- }
- }
- }
-
- // Did not resize
- return new ThreeTuple<Boolean, Boolean, Long>(false, seenLiveSlot, currentSequenceNumber);
- }
-
- void doOptionalRescue(Slot s, bool seenliveslot, int64_t seqn, bool resize) {
- /* now go through live entries from least to greatest sequence number until
- * either all live slots added, or the slot doesn't have enough room
- * for SKIP_THRESHOLD consecutive entries*/
- int skipcount = 0;
- int64_t newestseqnum = buffer.getNewestSeqNum();
- search:
- for (; seqn <= newestseqnum; seqn++) {
- Slot prevslot = buffer.getSlot(seqn);
- //Push slot number forward
- if (!seenliveslot)
- oldestLiveSlotSequenceNumver = seqn;
-
- if (!prevslot.isLive())
- continue;
- seenliveslot = true;
- Vector<Entry> liveentries = prevslot.getLiveEntries(resize);
- for (Entry liveentry : liveentries) {
- if (s.hasSpace(liveentry))
- s.addEntry(liveentry);
- else {
- skipcount++;
- if (skipcount > SKIP_THRESHOLD)
- break search;
+ SetIterator<Transaction *, Vector<int> *> *trit = getKeyIterator(lastTransactionPartsSent);
+ while (trit->hasNext()) {
+ Transaction *transaction = trit->next();
+ transaction->resetServerFailure();
+ // Set the transaction sequence number back to nothing
+ if (!transaction->didSendAPartToServer()) {
+ transaction->setSequenceNumber(-1);
+ }
+ }
+ delete trit;
}
+
+ // insert into the local block chain
+ validateAndUpdate(newSlots, true);
}
}
+ } catch (ServerException *e) {
+ throw e;
}
- /**
- * Checks for malicious activity and updates the local copy of the block chain.
- */
- void validateAndUpdate(Slot[] newSlots, bool acceptUpdatesToLocal) {
- // The cloud communication layer has checked slot HMACs already before decoding
- if (newSlots.length == 0) {
- return;
- }
- // Make sure all slots are newer than the last largest slot this client has seen
- int64_t firstSeqNum = newSlots[0].getSequenceNumber();
- if (firstSeqNum <= sequenceNumber) {
- throw new Error("Server Error: Sent older slots!");
- }
+ try {
+ // While we have stuff that needs inserting into the block chain
+ while ((pendingTransactionQueue->size() > 0) || (pendingSendArbitrationRounds->size() > 0) || (newKey != NULL)) {
- // Create an object that can access both new slots and slots in our local chain
- // without committing slots to our local chain
- SlotIndexer indexer = new SlotIndexer(newSlots, buffer);
+ fromRetry = false;
- // Check that the HMAC chain is not broken
- checkHMACChain(indexer, newSlots);
+ if (hadPartialSendToServer) {
+ throw new Error("Should Be error free");
+ }
- // Set to keep track of messages from clients
- HashSet<Long> machineSet = new HashSet<Long>(lastMessageTable.keySet());
- // Process each slots data
- for (Slot slot : newSlots) {
- processSlot(indexer, slot, acceptUpdatesToLocal, machineSet);
- updateExpectedSize();
- }
+ // If there is a new key with same name then end
+ if ((newKey != NULL) && arbitratorTable->contains(newKey->getKey())) {
+ return false;
+ }
- // If there is a gap, check to see if the server sent us everything.
- if (firstSeqNum != (sequenceNumber + 1)) {
+ // Create the slot
+ Slot *slot = new Slot(this, sequenceNumber + 1, localMachineId, buffer->getSlot(sequenceNumber)->getHMAC(), localSequenceNumber);
+ localSequenceNumber++;
- // Check the size of the slots that were sent down by the server.
- // Can only check the size if there was a gap
- checkNumSlots(newSlots.length);
+ // Try to fill the slot with data
+ ThreeTuple<bool, int32_t, bool> fillSlotsReturn = fillSlot(slot, false, newKey);
+ bool needsResize = fillSlotsReturn.getFirst();
+ int newSize = fillSlotsReturn.getSecond();
+ bool insertedNewKey = fillSlotsReturn.getThird();
- // Since there was a gap every machine must have pushed a slot or must have
- // a last message message. If not then the server is hiding slots
- if (!machineSet.isEmpty()) {
- throw new Error("Missing record for machines: " + machineSet);
- }
- }
+ if (needsResize) {
+ // Reset which transaction to send
+ SetIterator<Transaction *, Vector<int> *> *trit = getKeyIterator(transactionPartsSent);
+ while (trit->hasNext()) {
+ Transaction *transaction = trit->next();
+ transaction->resetNextPartToSend();
- // Update the size of our local block chain.
- commitNewMaxSize();
+ // Set the transaction sequence number back to nothing
+ if (!transaction->didSendAPartToServer() && !transaction->getServerFailure()) {
+ transaction->setSequenceNumber(-1);
+ }
+ }
+ delete trit;
- // Commit new to slots to the local block chain.
- for (Slot slot : newSlots) {
+ // Clear the sent data since we are trying again
+ pendingSendArbitrationEntriesToDelete->clear();
+ transactionPartsSent->clear();
- // Insert this slot into our local block chain copy.
- buffer.putSlot(slot);
+ // We needed a resize so try again
+ fillSlot(slot, true, newKey);
+ }
- // Keep track of how many slots are currently live (have live data in them).
- liveSlotCount++;
- }
+ lastSlotAttemptedToSend = slot;
+ lastIsNewKey = (newKey != NULL);
+ lastInsertedNewKey = insertedNewKey;
+ lastNewSize = newSize;
+ lastNewKey = newKey;
+ lastTransactionPartsSent = transactionPartsSent->clone();
+ lastPendingSendArbitrationEntriesToDelete = new Vector<Entry *>(pendingSendArbitrationEntriesToDelete);
- // Get the sequence number of the latest slot in the system
- sequenceNumber = newSlots[newSlots.length - 1].getSequenceNumber();
+ ThreeTuple<bool, bool, Array<Slot *> *> sendSlotsReturn = sendSlotsToServer(slot, newSize, newKey != NULL);
- updateLiveStateFromServer();
+ if (sendSlotsReturn.getFirst()) {
- // No Need to remember after we pulled from the server
- offlineTransactionsCommittedAndAtServer.clear();
+ // Did insert into the block chain
- // This is invalidated now
- hadPartialSendToServer = false;
- }
+ if (insertedNewKey) {
+ // This slot was what was inserted not a previous slot
- void updateLiveStateFromServer() {
- // Process the new transaction parts
- processNewTransactionParts();
+ // New Key was successfully inserted into the block chain so dont want to insert it again
+ newKey = NULL;
+ }
- // Do arbitration on new transactions that were received
- arbitrateFromServer();
+ // Remove the aborts and commit parts that were sent from the pending to send queue
+ uint size = pendingSendArbitrationRounds->size();
+ uint oldcount = 0;
+ for (uint i = 0; i < size; i++) {
+ ArbitrationRound *round = pendingSendArbitrationRounds->get(i);
+ round->removeParts(pendingSendArbitrationEntriesToDelete);
+
+ if (!round->isDoneSending()) {
+ // Sent all the parts
+ pendingSendArbitrationRounds->set(oldcount++,
+ pendingSendArbitrationRounds->get(i));
+ }
+ }
+ pendingSendArbitrationRounds->setSize(oldcount);
- // Update all the committed keys
- bool didCommitOrSpeculate = updateCommittedTable();
+ SetIterator<Transaction *, Vector<int> *> *trit = getKeyIterator(transactionPartsSent);
+ while (trit->hasNext()) {
+ Transaction *transaction = trit->next();
+ transaction->resetServerFailure();
- // Delete the transactions that are now dead
- updateLiveTransactionsAndStatus();
+ // Update which transactions parts still need to be sent
+ transaction->removeSentParts(transactionPartsSent->get(transaction));
- // Do speculations
- didCommitOrSpeculate |= updateSpeculativeTable(didCommitOrSpeculate);
- updatePendingTransactionSpeculativeTable(didCommitOrSpeculate);
- }
+ // Add the transaction status to the outstanding list
+ outstandingTransactionStatus->put(transaction->getSequenceNumber(), transaction->getTransactionStatus());
- void updateLiveStateFromLocal() {
- // Update all the committed keys
- bool didCommitOrSpeculate = updateCommittedTable();
+ // Update the transaction status
+ transaction->getTransactionStatus()->setStatus(TransactionStatus_StatusSentPartial);
- // Delete the transactions that are now dead
- updateLiveTransactionsAndStatus();
+ // Check if all the transaction parts were successfully sent and if so then remove it from pending
+ if (transaction->didSendAllParts()) {
+ transaction->getTransactionStatus()->setStatus(TransactionStatus_StatusSentFully);
+ pendingTransactionQueue->remove(transaction);
+ }
+ }
+ delete trit;
+ } else {
+ // Reset which transaction to send
+ SetIterator<Transaction *, Vector<int> *> *trit = getKeyIterator(transactionPartsSent);
+ while (trit->hasNext()) {
+ Transaction *transaction = trit->next();
+ transaction->resetNextPartToSend();
- // Do speculations
- didCommitOrSpeculate |= updateSpeculativeTable(didCommitOrSpeculate);
- updatePendingTransactionSpeculativeTable(didCommitOrSpeculate);
- }
+ // Set the transaction sequence number back to nothing
+ if (!transaction->didSendAPartToServer() && !transaction->getServerFailure()) {
+ transaction->setSequenceNumber(-1);
+ }
+ }
+ delete trit;
+ }
- void initExpectedSize(int64_t firstSequenceNumber, int64_t numberOfSlots) {
- // if (didFindTableStatus) {
- // return;
- // }
- int64_t prevslots = firstSequenceNumber;
+ // Clear the sent data in preparation for next send
+ pendingSendArbitrationEntriesToDelete->clear();
+ transactionPartsSent->clear();
+ if (sendSlotsReturn.getThird()->length() != 0) {
+ // insert into the local block chain
+ validateAndUpdate(sendSlotsReturn.getThird(), true);
+ }
+ }
- if (didFindTableStatus) {
- // expectedsize = (prevslots < ((int64_t) numberOfSlots)) ? (int) prevslots : expectedsize;
- // System.out.println("Here2: " + expectedsize + " " + numberOfSlots + " " + prevslots);
+ } catch (ServerException *e) {
+ if (e->getType() != ServerException_TypeInputTimeout) {
+ // Nothing was able to be sent to the server so just clear these data structures
+ SetIterator<Transaction *, Vector<int> *> *trit = getKeyIterator(transactionPartsSent);
+ while (trit->hasNext()) {
+ Transaction *transaction = trit->next();
+ transaction->resetNextPartToSend();
+ // Set the transaction sequence number back to nothing
+ if (!transaction->didSendAPartToServer() && !transaction->getServerFailure()) {
+ transaction->setSequenceNumber(-1);
+ }
+ }
+ delete trit;
} else {
- expectedsize = (prevslots < ((int64_t) numberOfSlots)) ? (int) prevslots : numberOfSlots;
- // System.out.println("Here: " + expectedsize);
+ // There was a partial send to the server
+ hadPartialSendToServer = true;
+
+ // Nothing was able to be sent to the server so just clear these data structures
+ SetIterator<Transaction *, Vector<int> *> *trit = getKeyIterator(transactionPartsSent);
+ while (trit->hasNext()) {
+ Transaction *transaction = trit->next();
+ transaction->resetNextPartToSend();
+ transaction->setServerFailure();
+ }
+ delete trit;
}
- // System.out.println(numberOfSlots);
+ pendingSendArbitrationEntriesToDelete->clear();
+ transactionPartsSent->clear();
- didFindTableStatus = true;
- currMaxSize = numberOfSlots;
+ throw e;
}
- void updateExpectedSize() {
- expectedsize++;
+ return newKey == NULL;
+}
- if (expectedsize > currMaxSize) {
- expectedsize = currMaxSize;
- }
- }
+bool Table::updateFromLocal(int64_t machineId) {
+ if (!localCommunicationTable->contains(machineId))
+ return false;
+ Pair<IoTString *, int32_t> * localCommunicationInformation = localCommunicationTable->get(machineId);
- /**
- * Check the size of the block chain to make sure there are enough slots sent back by the server.
- * This is only called when we have a gap between the slots that we have locally and the slots
- * sent by the server therefore in the slots sent by the server there will be at least 1 Table
- * status message
- */
- void checkNumSlots(int numberOfSlots) {
- if (numberOfSlots != expectedsize) {
- throw new Error("Server Error: Server did not send all slots. Expected: " + expectedsize + " Received:" + numberOfSlots);
- }
- }
+ // Get the size of the send data
+ int sendDataSize = sizeof(int32_t) + sizeof(int64_t);
- void updateCurrMaxSize(int newmaxsize) {
- currMaxSize = newmaxsize;
+ int64_t lastArbitrationDataLocalSequenceNumber = (int64_t) -1;
+ if (lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->contains(machineId)) {
+ lastArbitrationDataLocalSequenceNumber = lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->get(machineId);
}
+ Array<char> *sendData = new Array<char>(sendDataSize);
+ ByteBuffer *bbEncode = ByteBuffer_wrap(sendData);
- /**
- * Update the size of of the local buffer if it is needed.
- */
- void commitNewMaxSize() {
- didFindTableStatus = false;
+ // Encode the data
+ bbEncode->putLong(lastArbitrationDataLocalSequenceNumber);
+ bbEncode->putInt(0);
- // Resize the local slot buffer
- if (numberOfSlots != currMaxSize) {
- buffer.resize((int)currMaxSize);
- }
-
- // Change the number of local slots to the new size
- numberOfSlots = (int)currMaxSize;
+ // Send by local
+ Array<char> *returnData = cloud->sendLocalData(sendData, localSequenceNumber, localCommunicationInformation->getFirst(), localCommunicationInformation->getSecond());
+ localSequenceNumber++;
-
- // Recalculate the resize threshold since the size of the local buffer has changed
- setResizeThreshold();
+ if (returnData == NULL) {
+ // Could not contact server
+ return false;
}
- /**
- * Process the new transaction parts from this latest round of slots received from the server
- */
- void processNewTransactionParts() {
+ // Decode the data
+ ByteBuffer *bbDecode = ByteBuffer_wrap(returnData);
+ int numberOfEntries = bbDecode->getInt();
- if (newTransactionParts.size() == 0) {
- // Nothing new to process
- return;
+ for (int i = 0; i < numberOfEntries; i++) {
+ char type = bbDecode->get();
+ if (type == TypeAbort) {
+ Abort *abort = (Abort *)Abort_decode(NULL, bbDecode);
+ processEntry(abort);
+ } else if (type == TypeCommitPart) {
+ CommitPart *commitPart = (CommitPart *)CommitPart_decode(NULL, bbDecode);
+ processEntry(commitPart);
}
+ }
- // Iterate through all the machine Ids that we received new parts for
- for (Long machineId : newTransactionParts.keySet()) {
- Hashtable<Pair<Long, Integer>, TransactionPart> parts = newTransactionParts.get(machineId);
-
- // Iterate through all the parts for that machine Id
- for (Pair<Long, Integer> partId : parts.keySet()) {
- TransactionPart part = parts.get(partId);
-
- Long lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable.get(part.getArbitratorId());
- if ((lastTransactionNumber != NULL) && (lastTransactionNumber >= part.getSequenceNumber())) {
- // Set dead the transaction part
- part.setDead();
- continue;
- }
+ updateLiveStateFromLocal();
- // Get the transaction object for that sequence number
- Transaction transaction = liveTransactionBySequenceNumberTable.get(part.getSequenceNumber());
+ return true;
+}
- if (transaction == NULL) {
- // This is a new transaction that we dont have so make a new one
- transaction = new Transaction();
+Pair<bool, bool> Table::sendTransactionToLocal(Transaction *transaction) {
- // Insert this new transaction into the live tables
- liveTransactionBySequenceNumberTable.put(part.getSequenceNumber(), transaction);
- liveTransactionByTransactionIdTable.put(part.getTransactionId(), transaction);
- }
+ // Get the devices local communications
+ if (!localCommunicationTable->contains(transaction->getArbitrator()))
+ return Pair<bool, bool>(true, false);
+
+ Pair<IoTString *, int32_t> * localCommunicationInformation = localCommunicationTable->get(transaction->getArbitrator());
- // Add that part to the transaction
- transaction.addPartDecode(part);
- }
+ // Get the size of the send data
+ int sendDataSize = sizeof(int32_t) + sizeof(int64_t);
+ {
+ Vector<TransactionPart *> * tParts = transaction->getParts();
+ uint tPartsSize = tParts->size();
+ for (uint i = 0; i < tPartsSize; i++) {
+ TransactionPart * part = tParts->get(i);
+ sendDataSize += part->getSize();
}
-
- // Clear all the new transaction parts in preparation for the next time the server sends slots
- newTransactionParts.clear();
}
+ int64_t lastArbitrationDataLocalSequenceNumber = (int64_t) -1;
+ if (lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->contains(transaction->getArbitrator())) {
+ lastArbitrationDataLocalSequenceNumber = lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->get(transaction->getArbitrator());
+ }
- int64_t lastSeqNumArbOn = 0;
-
- void arbitrateFromServer() {
+ // Make the send data size
+ Array<char> *sendData = new Array<char>(sendDataSize);
+ ByteBuffer *bbEncode = ByteBuffer_wrap(sendData);
- if (liveTransactionBySequenceNumberTable.size() == 0) {
- // Nothing to arbitrate on so move on
- return;
+ // Encode the data
+ bbEncode->putLong(lastArbitrationDataLocalSequenceNumber);
+ bbEncode->putInt(transaction->getParts()->size());
+ {
+ Vector<TransactionPart *> * tParts = transaction->getParts();
+ uint tPartsSize = tParts->size();
+ for (uint i = 0; i < tPartsSize; i++) {
+ TransactionPart * part = tParts->get(i);
+ part->encode(bbEncode);
}
+ }
- // Get the transaction sequence numbers and sort from oldest to newest
- List<Long> transactionSequenceNumbers = new ArrayList<Long>(liveTransactionBySequenceNumberTable.keySet());
- Collections.sort(transactionSequenceNumbers);
-
- // Collection of key value pairs that are
- Hashtable<IoTString, KeyValue> speculativeTableTmp = new Hashtable<IoTString, KeyValue>();
-
- // The last transaction arbitrated on
- int64_t lastTransactionCommitted = -1;
- Set<Abort> generatedAborts = new HashSet<Abort>();
-
- for (Long transactionSequenceNumber : transactionSequenceNumbers) {
- Transaction transaction = liveTransactionBySequenceNumberTable.get(transactionSequenceNumber);
-
-
-
- // Check if this machine arbitrates for this transaction if not then we cant arbitrate this transaction
- if (transaction.getArbitrator() != localMachineId) {
- continue;
- }
-
- if (transactionSequenceNumber < lastSeqNumArbOn) {
- continue;
- }
-
- if (offlineTransactionsCommittedAndAtServer.contains(transaction.getId())) {
- // We have seen this already locally so dont commit again
- continue;
- }
+ // Send by local
+ Array<char> *returnData = cloud->sendLocalData(sendData, localSequenceNumber, localCommunicationInformation->getFirst(), localCommunicationInformation->getSecond());
+ localSequenceNumber++;
+ if (returnData == NULL) {
+ // Could not contact server
+ return Pair<bool, bool>(true, false);
+ }
- if (!transaction.isComplete()) {
- // Will arbitrate in incorrect order if we continue so just break
- // Most likely this
- break;
- }
+ // Decode the data
+ ByteBuffer *bbDecode = ByteBuffer_wrap(returnData);
+ bool didCommit = bbDecode->get() == 1;
+ bool couldArbitrate = bbDecode->get() == 1;
+ int numberOfEntries = bbDecode->getInt();
+ bool foundAbort = false;
+ for (int i = 0; i < numberOfEntries; i++) {
+ char type = bbDecode->get();
+ if (type == TypeAbort) {
+ Abort *abort = (Abort *)Abort_decode(NULL, bbDecode);
- // update the largest transaction seen by arbitrator from server
- if (lastTransactionSeenFromMachineFromServer.get(transaction.getMachineId()) == NULL) {
- lastTransactionSeenFromMachineFromServer.put(transaction.getMachineId(), transaction.getClientLocalSequenceNumber());
- } else {
- Long lastTransactionSeenFromMachine = lastTransactionSeenFromMachineFromServer.get(transaction.getMachineId());
- if (transaction.getClientLocalSequenceNumber() > lastTransactionSeenFromMachine) {
- lastTransactionSeenFromMachineFromServer.put(transaction.getMachineId(), transaction.getClientLocalSequenceNumber());
- }
+ if ((abort->getTransactionMachineId() == localMachineId) && (abort->getTransactionClientLocalSequenceNumber() == transaction->getClientLocalSequenceNumber())) {
+ foundAbort = true;
}
- if (transaction.evaluateGuard(committedKeyValueTable, speculativeTableTmp, NULL)) {
- // Guard evaluated as true
+ processEntry(abort);
+ } else if (type == TypeCommitPart) {
+ CommitPart *commitPart = (CommitPart *)CommitPart_decode(NULL, bbDecode);
+ processEntry(commitPart);
+ }
+ }
- // Update the local changes so we can make the commit
- for (KeyValue kv : transaction.getKeyValueUpdateSet()) {
- speculativeTableTmp.put(kv.getKey(), kv);
- }
+ updateLiveStateFromLocal();
- // Update what the last transaction committed was for use in batch commit
- lastTransactionCommitted = transactionSequenceNumber;
- } else {
- // Guard evaluated was false so create abort
+ if (couldArbitrate) {
+ TransactionStatus * status = transaction->getTransactionStatus();
+ if (didCommit) {
+ status->setStatus(TransactionStatus_StatusCommitted);
+ } else {
+ status->setStatus(TransactionStatus_StatusAborted);
+ }
+ } else {
+ TransactionStatus * status = transaction->getTransactionStatus();
+ if (foundAbort) {
+ status->setStatus(TransactionStatus_StatusAborted);
+ } else {
+ status->setStatus(TransactionStatus_StatusCommitted);
+ }
+ }
- // Create the abort
- Abort newAbort = new Abort(NULL,
- transaction.getClientLocalSequenceNumber(),
- transaction.getSequenceNumber(),
- transaction.getMachineId(),
- transaction.getArbitrator(),
- localArbitrationSequenceNumber);
- localArbitrationSequenceNumber++;
+ return Pair<bool, bool>(false, true);
+}
- generatedAborts.add(newAbort);
+Array<char> *Table::acceptDataFromLocal(Array<char> *data) {
+ // Decode the data
+ ByteBuffer *bbDecode = ByteBuffer_wrap(data);
+ int64_t lastArbitratedSequenceNumberSeen = bbDecode->getLong();
+ int numberOfParts = bbDecode->getInt();
- // Insert the abort so we can process
- processEntry(newAbort);
- }
+ // If we did commit a transaction or not
+ bool didCommit = false;
+ bool couldArbitrate = false;
- lastSeqNumArbOn = transactionSequenceNumber;
+ if (numberOfParts != 0) {
- // liveTransactionBySequenceNumberTable.remove(transactionSequenceNumber);
+ // decode the transaction
+ Transaction *transaction = new Transaction();
+ for (int i = 0; i < numberOfParts; i++) {
+ bbDecode->get();
+ TransactionPart *newPart = (TransactionPart *)TransactionPart_decode(NULL, bbDecode);
+ transaction->addPartDecode(newPart);
}
- Commit newCommit = NULL;
+ // Arbitrate on transaction and pull relevant return data
+ Pair<bool, bool> localArbitrateReturn = arbitrateOnLocalTransaction(transaction);
+ couldArbitrate = localArbitrateReturn.getFirst();
+ didCommit = localArbitrateReturn.getSecond();
- // If there is something to commit
- if (speculativeTableTmp.size() != 0) {
+ updateLiveStateFromLocal();
- // Create the commit and increment the commit sequence number
- newCommit = new Commit(localArbitrationSequenceNumber, localMachineId, lastTransactionCommitted);
- localArbitrationSequenceNumber++;
+ // Transaction was sent to the server so keep track of it to prevent double commit
+ if (transaction->getSequenceNumber() != -1) {
+ offlineTransactionsCommittedAndAtServer->add(new Pair<int64_t, int64_t>(transaction->getId()));
+ }
+ }
- // Add all the new keys to the commit
- for (KeyValue kv : speculativeTableTmp.values()) {
- newCommit.addKV(kv);
- }
+ // The data to send back
+ int returnDataSize = 0;
+ Vector<Entry *> *unseenArbitrations = new Vector<Entry *>();
+
+ // Get the aborts to send back
+ Vector<int64_t> *abortLocalSequenceNumbers = new Vector<int64_t>();
+ {
+ SetIterator<int64_t, Abort *> *abortit = getKeyIterator(liveAbortsGeneratedByLocal);
+ while(abortit->hasNext())
+ abortLocalSequenceNumbers->add(abortit->next());
+ delete abortit;
+ }
+
+ qsort(abortLocalSequenceNumbers->expose(), abortLocalSequenceNumbers->size(), sizeof(int64_t), compareInt64);
+
+ uint asize = abortLocalSequenceNumbers->size();
+ for(uint i=0; i<asize; i++) {
+ int64_t localSequenceNumber = abortLocalSequenceNumbers->get(i);
+ if (localSequenceNumber <= lastArbitratedSequenceNumberSeen) {
+ continue;
+ }
+
+ Abort *abort = liveAbortsGeneratedByLocal->get(localSequenceNumber);
+ unseenArbitrations->add(abort);
+ returnDataSize += abort->getSize();
+ }
- // create the commit parts
- newCommit.createCommitParts();
+ // Get the commits to send back
+ Hashtable<int64_t, Commit *> *commitForClientTable = liveCommitsTable->get(localMachineId);
+ if (commitForClientTable != NULL) {
+ Vector<int64_t> *commitLocalSequenceNumbers = new Vector<int64_t>();
+ {
+ SetIterator<int64_t, Commit *> *commitit = getKeyIterator(commitForClientTable);
+ while(commitit->hasNext())
+ commitLocalSequenceNumbers->add(commitit->next());
+ delete commitit;
+ }
+ qsort(commitLocalSequenceNumbers->expose(), commitLocalSequenceNumbers->size(), sizeof(int64_t), compareInt64);
- // Append all the commit parts to the end of the pending queue waiting for sending to the server
+ uint clsSize = commitLocalSequenceNumbers->size();
+ for(uint clsi = 0; clsi < clsSize; clsi++) {
+ int64_t localSequenceNumber = commitLocalSequenceNumbers->get(clsi);
+ Commit *commit = commitForClientTable->get(localSequenceNumber);
- // Insert the commit so we can process it
- for (CommitPart commitPart : newCommit.getParts().values()) {
- processEntry(commitPart);
+ if (localSequenceNumber <= lastArbitratedSequenceNumberSeen) {
+ continue;
}
- }
- if ((newCommit != NULL) || (generatedAborts.size() > 0)) {
- ArbitrationRound arbitrationRound = new ArbitrationRound(newCommit, generatedAborts);
- pendingSendArbitrationRounds.add(arbitrationRound);
-
- if (compactArbitrationData()) {
- ArbitrationRound newArbitrationRound = pendingSendArbitrationRounds.get(pendingSendArbitrationRounds.size() - 1);
- if (newArbitrationRound.getCommit() != NULL) {
- for (CommitPart commitPart : newArbitrationRound.getCommit().getParts().values()) {
- processEntry(commitPart);
- }
+ {
+ Vector<CommitPart *> * parts = commit->getParts();
+ uint nParts = parts->size();
+ for(uint i=0; i<nParts; i++) {
+ CommitPart * commitPart = parts->get(i);
+ unseenArbitrations->add(commitPart);
+ returnDataSize += commitPart->getSize();
}
}
}
}
- Pair<Boolean, Boolean> arbitrateOnLocalTransaction(Transaction transaction) {
+ // Number of arbitration entries to decode
+ returnDataSize += 2 * sizeof(int32_t);
- // Check if this machine arbitrates for this transaction if not then we cant arbitrate this transaction
- if (transaction.getArbitrator() != localMachineId) {
- return new Pair<Boolean, Boolean>(false, false);
- }
+ // bool of did commit or not
+ if (numberOfParts != 0) {
+ returnDataSize += sizeof(char);
+ }
- if (!transaction.isComplete()) {
- // Will arbitrate in incorrect order if we continue so just break
- // Most likely this
- return new Pair<Boolean, Boolean>(false, false);
- }
+ // Data to send Back
+ Array<char> *returnData = new Array<char>(returnDataSize);
+ ByteBuffer *bbEncode = ByteBuffer_wrap(returnData);
- if (transaction.getMachineId() != localMachineId) {
- // dont do this check for local transactions
- if (lastTransactionSeenFromMachineFromServer.get(transaction.getMachineId()) != NULL) {
- if (lastTransactionSeenFromMachineFromServer.get(transaction.getMachineId()) > transaction.getClientLocalSequenceNumber()) {
- // We've have already seen this from the server
- return new Pair<Boolean, Boolean>(false, false);
- }
- }
+ if (numberOfParts != 0) {
+ if (didCommit) {
+ bbEncode->put((char)1);
+ } else {
+ bbEncode->put((char)0);
+ }
+ if (couldArbitrate) {
+ bbEncode->put((char)1);
+ } else {
+ bbEncode->put((char)0);
}
+ }
- if (transaction.evaluateGuard(committedKeyValueTable, NULL, NULL)) {
- // Guard evaluated as true
+ bbEncode->putInt(unseenArbitrations->size());
+ uint size = unseenArbitrations->size();
+ for (uint i = 0; i < size; i++) {
+ Entry *entry = unseenArbitrations->get(i);
+ entry->encode(bbEncode);
+ }
- // Create the commit and increment the commit sequence number
- Commit newCommit = new Commit(localArbitrationSequenceNumber, localMachineId, -1);
- localArbitrationSequenceNumber++;
+ localSequenceNumber++;
+ return returnData;
+}
- // Update the local changes so we can make the commit
- for (KeyValue kv : transaction.getKeyValueUpdateSet()) {
- newCommit.addKV(kv);
+ThreeTuple<bool, bool, Array<Slot *> *> Table::sendSlotsToServer(Slot *slot, int newSize, bool isNewKey) {
+ bool attemptedToSendToServerTmp = attemptedToSendToServer;
+ attemptedToSendToServer = true;
+
+ bool inserted = false;
+ bool lastTryInserted = false;
+
+ Array<Slot *> *array = cloud->putSlot(slot, newSize);
+ if (array == NULL) {
+ array = new Array<Slot *>();
+ array->set(0, slot);
+ rejectedSlotVector->clear();
+ inserted = true;
+ } else {
+ if (array->length() == 0) {
+ throw new Error("Server Error: Did not send any slots");
+ }
+
+ // if (attemptedToSendToServerTmp) {
+ if (hadPartialSendToServer) {
+
+ bool isInserted = false;
+ uint size = array->length();
+ for (uint i = 0; i < size; i++) {
+ Slot *s = array->get(i);
+ if ((s->getSequenceNumber() == slot->getSequenceNumber()) && (s->getMachineID() == localMachineId)) {
+ isInserted = true;
+ break;
+ }
}
- // create the commit parts
- newCommit.createCommitParts();
+ for (uint i = 0; i < size; i++) {
+ Slot *s = array->get(i);
+ if (isInserted) {
+ break;
+ }
- // Append all the commit parts to the end of the pending queue waiting for sending to the server
- ArbitrationRound arbitrationRound = new ArbitrationRound(newCommit, new HashSet<Abort>());
- pendingSendArbitrationRounds.add(arbitrationRound);
+ // Process each entry in the slot
+ Vector<Entry *> *entries = s->getEntries();
+ uint eSize = entries->size();
+ for(uint ei=0; ei < eSize; ei++) {
+ Entry * entry = entries->get(ei);
- if (compactArbitrationData()) {
- ArbitrationRound newArbitrationRound = pendingSendArbitrationRounds.get(pendingSendArbitrationRounds.size() - 1);
- for (CommitPart commitPart : newArbitrationRound.getCommit().getParts().values()) {
- processEntry(commitPart);
- }
- } else {
- // Insert the commit so we can process it
- for (CommitPart commitPart : newCommit.getParts().values()) {
- processEntry(commitPart);
- }
- }
+ if (entry->getType() == TypeLastMessage) {
+ LastMessage *lastMessage = (LastMessage *)entry;
- if (transaction.getMachineId() == localMachineId) {
- TransactionStatus status = transaction.getTransactionStatus();
- if (status != NULL) {
- status.setStatus(TransactionStatus.StatusCommitted);
+ if ((lastMessage->getMachineID() == localMachineId) && (lastMessage->getSequenceNumber() == slot->getSequenceNumber())) {
+ isInserted = true;
+ break;
+ }
+ }
}
}
- updateLiveStateFromLocal();
- return new Pair<Boolean, Boolean>(true, true);
+ if (!isInserted) {
+ rejectedSlotVector->add(slot->getSequenceNumber());
+ lastTryInserted = false;
+ } else {
+ lastTryInserted = true;
+ }
} else {
+ rejectedSlotVector->add(slot->getSequenceNumber());
+ lastTryInserted = false;
+ }
+ }
- if (transaction.getMachineId() == localMachineId) {
- // For locally created messages update the status
-
- // Guard evaluated was false so create abort
- TransactionStatus status = transaction.getTransactionStatus();
- if (status != NULL) {
- status.setStatus(TransactionStatus.StatusAborted);
- }
- } else {
- Set addAbortSet = new HashSet<Abort>();
+ return ThreeTuple<bool, bool, Array<Slot *> *>(inserted, lastTryInserted, array);
+}
+/**
+ * Returns false if a resize was needed
+ */
+ThreeTuple<bool, int32_t, bool> Table::fillSlot(Slot *slot, bool resize, NewKey *newKeyEntry) {
+ int newSize = 0;
+ if (liveSlotCount > bufferResizeThreshold) {
+ resize = true;//Resize is forced
+ }
- // Create the abort
- Abort newAbort = new Abort(NULL,
- transaction.getClientLocalSequenceNumber(),
- -1,
- transaction.getMachineId(),
- transaction.getArbitrator(),
- localArbitrationSequenceNumber);
- localArbitrationSequenceNumber++;
+ if (resize) {
+ newSize = (int) (numberOfSlots * Table_RESIZE_MULTIPLE);
+ TableStatus *status = new TableStatus(slot, newSize);
+ slot->addEntry(status);
+ }
- addAbortSet.add(newAbort);
+ // Fill with rejected slots first before doing anything else
+ doRejectedMessages(slot);
+ // Do mandatory rescue of entries
+ ThreeTuple<bool, bool, int64_t> mandatoryRescueReturn = doMandatoryResuce(slot, resize);
- // Append all the commit parts to the end of the pending queue waiting for sending to the server
- ArbitrationRound arbitrationRound = new ArbitrationRound(NULL, addAbortSet);
- pendingSendArbitrationRounds.add(arbitrationRound);
+ // Extract working variables
+ bool needsResize = mandatoryRescueReturn.getFirst();
+ bool seenLiveSlot = mandatoryRescueReturn.getSecond();
+ int64_t currentRescueSequenceNumber = mandatoryRescueReturn.getThird();
- if (compactArbitrationData()) {
- ArbitrationRound newArbitrationRound = pendingSendArbitrationRounds.get(pendingSendArbitrationRounds.size() - 1);
- for (CommitPart commitPart : newArbitrationRound.getCommit().getParts().values()) {
- processEntry(commitPart);
- }
- }
- }
+ if (needsResize && !resize) {
+ // We need to resize but we are not resizing so return false
+ return ThreeTuple<bool, int32_t, bool>(true, NULL, NULL);
+ }
- updateLiveStateFromLocal();
- return new Pair<Boolean, Boolean>(true, false);
+ bool inserted = false;
+ if (newKeyEntry != NULL) {
+ newKeyEntry->setSlot(slot);
+ if (slot->hasSpace(newKeyEntry)) {
+ slot->addEntry(newKeyEntry);
+ inserted = true;
}
}
- /**
- * Compacts the arbitration data my merging commits and aggregating aborts so that a single large push of commits can be done instead of many small updates
- */
- bool compactArbitrationData() {
+ // Clear the transactions, aborts and commits that were sent previously
+ transactionPartsSent->clear();
+ pendingSendArbitrationEntriesToDelete->clear();
+ uint size = pendingSendArbitrationRounds->size();
+ for (uint i = 0; i < size; i++) {
+ ArbitrationRound *round = pendingSendArbitrationRounds->get(i);
+ bool isFull = false;
+ round->generateParts();
+ Vector<Entry *> *parts = round->getParts();
+
+ // Insert pending arbitration data
+ uint vsize = parts->size();
+ for (uint vi = 0; vi < vsize; vi++) {
+ Entry *arbitrationData = parts->get(vi);
+
+ // If it is an abort then we need to set some information
+ if (arbitrationData->getType() == TypeAbort) {
+ ((Abort *)arbitrationData)->setSequenceNumber(slot->getSequenceNumber());
+ }
+
+ if (!slot->hasSpace(arbitrationData)) {
+ // No space so cant do anything else with these data entries
+ isFull = true;
+ break;
+ }
- if (pendingSendArbitrationRounds.size() < 2) {
- // Nothing to compact so do nothing
- return false;
+ // Add to this current slot and add it to entries to delete
+ slot->addEntry(arbitrationData);
+ pendingSendArbitrationEntriesToDelete->add(arbitrationData);
}
- ArbitrationRound lastRound = pendingSendArbitrationRounds.get(pendingSendArbitrationRounds.size() - 1);
- if (lastRound.didSendPart()) {
- return false;
+ if (isFull) {
+ break;
}
+ }
- bool hadCommit = (lastRound.getCommit() == NULL);
- bool gotNewCommit = false;
-
- int numberToDelete = 1;
- while (numberToDelete < pendingSendArbitrationRounds.size()) {
- ArbitrationRound round = pendingSendArbitrationRounds.get(pendingSendArbitrationRounds.size() - numberToDelete - 1);
+ if (pendingTransactionQueue->size() > 0) {
+ Transaction *transaction = pendingTransactionQueue->get(0);
+ // Set the transaction sequence number if it has yet to be inserted into the block chain
+ if ((!transaction->didSendAPartToServer()) || (transaction->getSequenceNumber() == -1)) {
+ transaction->setSequenceNumber(slot->getSequenceNumber());
+ }
- if (round.isFull() || round.didSendPart()) {
- // Stop since there is a part that cannot be compacted and we need to compact in order
+ while (true) {
+ TransactionPart *part = transaction->getNextPartToSend();
+ if (part == NULL) {
+ // Ran out of parts to send for this transaction so move on
break;
}
- if (round.getCommit() == NULL) {
-
- // Try compacting aborts only
- int newSize = round.getCurrentSize() + lastRound.getAbortsCount();
- if (newSize > ArbitrationRound.MAX_PARTS) {
- // Cant compact since it would be too large
- break;
+ if (slot->hasSpace(part)) {
+ slot->addEntry(part);
+ Vector<int32_t> *partsSent = transactionPartsSent->get(transaction);
+ if (partsSent == NULL) {
+ partsSent = new Vector<int32_t>();
+ transactionPartsSent->put(transaction, partsSent);
}
- lastRound.addAborts(round.getAborts());
+ partsSent->add(part->getPartNumber());
+ transactionPartsSent->put(transaction, partsSent);
} else {
+ break;
+ }
+ }
+ }
- // Create a new larger commit
- Commit newCommit = Commit.merge(lastRound.getCommit(), round.getCommit(), localArbitrationSequenceNumber);
- localArbitrationSequenceNumber++;
-
- // Create the commit parts so that we can count them
- newCommit.createCommitParts();
+ // Fill the remainder of the slot with rescue data
+ doOptionalRescue(slot, seenLiveSlot, currentRescueSequenceNumber, resize);
- // Calculate the new size of the parts
- int newSize = newCommit.getNumberOfParts();
- newSize += lastRound.getAbortsCount();
- newSize += round.getAbortsCount();
+ return ThreeTuple<bool, int32_t, bool>(false, newSize, inserted);
+}
- if (newSize > ArbitrationRound.MAX_PARTS) {
- // Cant compact since it would be too large
+void Table::doRejectedMessages(Slot *s) {
+ if (!rejectedSlotVector->isEmpty()) {
+ /* TODO: We should avoid generating a rejected message entry if
+ * there is already a sufficient entry in the queue (e->g->,
+ * equalsto value of true and same sequence number)-> */
+
+ int64_t old_seqn = rejectedSlotVector->get(0);
+ if (rejectedSlotVector->size() > Table_REJECTED_THRESHOLD) {
+ int64_t new_seqn = rejectedSlotVector->lastElement();
+ RejectedMessage *rm = new RejectedMessage(s, s->getSequenceNumber(), localMachineId, old_seqn, new_seqn, false);
+ s->addEntry(rm);
+ } else {
+ int64_t prev_seqn = -1;
+ int i = 0;
+ /* Go through list of missing messages */
+ for (; i < rejectedSlotVector->size(); i++) {
+ int64_t curr_seqn = rejectedSlotVector->get(i);
+ Slot *s_msg = buffer->getSlot(curr_seqn);
+ if (s_msg != NULL)
break;
- }
-
- // Set the new compacted part
- lastRound.setCommit(newCommit);
- lastRound.addAborts(round.getAborts());
- gotNewCommit = true;
+ prev_seqn = curr_seqn;
}
+ /* Generate rejected message entry for missing messages */
+ if (prev_seqn != -1) {
+ RejectedMessage *rm = new RejectedMessage(s, s->getSequenceNumber(), localMachineId, old_seqn, prev_seqn, false);
+ s->addEntry(rm);
+ }
+ /* Generate rejected message entries for present messages */
+ for (; i < rejectedSlotVector->size(); i++) {
+ int64_t curr_seqn = rejectedSlotVector->get(i);
+ Slot *s_msg = buffer->getSlot(curr_seqn);
+ int64_t machineid = s_msg->getMachineID();
+ RejectedMessage *rm = new RejectedMessage(s, s->getSequenceNumber(), machineid, curr_seqn, curr_seqn, true);
+ s->addEntry(rm);
+ }
+ }
+ }
+}
+
+ThreeTuple<bool, bool, int64_t> Table::doMandatoryResuce(Slot *slot, bool resize) {
+ int64_t newestSequenceNumber = buffer->getNewestSeqNum();
+ int64_t oldestSequenceNumber = buffer->getOldestSeqNum();
+ if (oldestLiveSlotSequenceNumver < oldestSequenceNumber) {
+ oldestLiveSlotSequenceNumver = oldestSequenceNumber;
+ }
+
+ int64_t currentSequenceNumber = oldestLiveSlotSequenceNumver;
+ bool seenLiveSlot = false;
+ int64_t firstIfFull = newestSequenceNumber + 1 - numberOfSlots; // smallest seq number in the buffer if it is full
+ int64_t threshold = firstIfFull + Table_FREE_SLOTS; // we want the buffer to be clear of live entries up to this point
- numberToDelete++;
+
+ // Mandatory Rescue
+ for (; currentSequenceNumber < threshold; currentSequenceNumber++) {
+ Slot * previousSlot = buffer->getSlot(currentSequenceNumber);
+ // Push slot number forward
+ if (!seenLiveSlot) {
+ oldestLiveSlotSequenceNumver = currentSequenceNumber;
}
- if (numberToDelete != 1) {
- // If there is a compaction
+ if (!previousSlot->isLive()) {
+ continue;
+ }
- // Delete the previous pieces that are now in the new compacted piece
- if (numberToDelete == pendingSendArbitrationRounds.size()) {
- pendingSendArbitrationRounds.clear();
- } else {
- for (int i = 0; i < numberToDelete; i++) {
- pendingSendArbitrationRounds.remove(pendingSendArbitrationRounds.size() - 1);
- }
- }
+ // We have seen a live slot
+ seenLiveSlot = true;
- // Add the new compacted into the pending to send list
- pendingSendArbitrationRounds.add(lastRound);
+ // Get all the live entries for a slot
+ Vector<Entry *> *liveEntries = previousSlot->getLiveEntries(resize);
- // Should reinsert into the commit processor
- if (hadCommit && gotNewCommit) {
- return true;
+ // Iterate over all the live entries and try to rescue them
+ uint lESize = liveEntries->size();
+ for (uint i=0; i< lESize; i++) {
+ Entry * liveEntry = liveEntries->get(i);
+ if (slot->hasSpace(liveEntry)) {
+ // Enough space to rescue the entry
+ slot->addEntry(liveEntry);
+ } else if (currentSequenceNumber == firstIfFull) {
+ //if there's no space but the entry is about to fall off the queue
+ return ThreeTuple<bool, bool, int64_t>(true, seenLiveSlot, currentSequenceNumber);
}
}
-
- return false;
}
- // bool compactArbitrationData() {
- // return false;
- // }
- /**
- * Update all the commits and the committed tables, sets dead the dead transactions
- */
- bool updateCommittedTable() {
+ // Did not resize
+ return ThreeTuple<bool, bool, int64_t>(false, seenLiveSlot, currentSequenceNumber);
+}
- if (newCommitParts.size() == 0) {
- // Nothing new to process
- return false;
+void Table::doOptionalRescue(Slot *s, bool seenliveslot, int64_t seqn, bool resize) {
+ /* now go through live entries from least to greatest sequence number until
+ * either all live slots added, or the slot doesn't have enough room
+ * for SKIP_THRESHOLD consecutive entries*/
+ int skipcount = 0;
+ int64_t newestseqnum = buffer->getNewestSeqNum();
+search:
+ for (; seqn <= newestseqnum; seqn++) {
+ Slot *prevslot = buffer->getSlot(seqn);
+ //Push slot number forward
+ if (!seenliveslot)
+ oldestLiveSlotSequenceNumver = seqn;
+
+ if (!prevslot->isLive())
+ continue;
+ seenliveslot = true;
+ Vector<Entry *> *liveentries = prevslot->getLiveEntries(resize);
+ uint lESize = liveentries->size();
+ for (uint i=0; i< lESize; i++) {
+ Entry * liveentry = liveentries->get(i);
+ if (s->hasSpace(liveentry))
+ s->addEntry(liveentry);
+ else {
+ skipcount++;
+ if (skipcount > Table_SKIP_THRESHOLD)
+ goto donesearch;
+ }
}
+ }
+ donesearch:
+ ;
+}
- // Iterate through all the machine Ids that we received new parts for
- for (Long machineId : newCommitParts.keySet()) {
- Hashtable<Pair<Long, Integer>, CommitPart> parts = newCommitParts.get(machineId);
+/**
+ * Checks for malicious activity and updates the local copy of the block chain->
+ */
+void Table::validateAndUpdate(Array<Slot *> *newSlots, bool acceptUpdatesToLocal) {
+ // The cloud communication layer has checked slot HMACs already
+ // before decoding
+ if (newSlots->length() == 0) {
+ return;
+ }
+
+ // Make sure all slots are newer than the last largest slot this
+ // client has seen
+ int64_t firstSeqNum = newSlots->get(0)->getSequenceNumber();
+ if (firstSeqNum <= sequenceNumber) {
+ throw new Error("Server Error: Sent older slots!");
+ }
- // Iterate through all the parts for that machine Id
- for (Pair<Long, Integer> partId : parts.keySet()) {
- CommitPart part = parts.get(partId);
+ // Create an object that can access both new slots and slots in our
+ // local chain without committing slots to our local chain
+ SlotIndexer *indexer = new SlotIndexer(newSlots, buffer);
- // Get the transaction object for that sequence number
- Hashtable<Long, Commit> commitForClientTable = liveCommitsTable.get(part.getMachineId());
+ // Check that the HMAC chain is not broken
+ checkHMACChain(indexer, newSlots);
- if (commitForClientTable == NULL) {
- // This is the first commit from this device
- commitForClientTable = new Hashtable<Long, Commit>();
- liveCommitsTable.put(part.getMachineId(), commitForClientTable);
- }
+ // Set to keep track of messages from clients
+ Hashset<int64_t> *machineSet = new Hashset<int64_t>();
+ {
+ SetIterator<int64_t, Pair<int64_t, Liveness *> *> * lmit=getKeyIterator(lastMessageTable);
+ while(lmit->hasNext())
+ machineSet->add(lmit->next());
+ delete lmit;
+ }
- Commit commit = commitForClientTable.get(part.getSequenceNumber());
+ // Process each slots data
+ {
+ uint numSlots = newSlots->length();
+ for(uint i=0; i<numSlots; i++) {
+ Slot *slot = newSlots->get(i);
+ processSlot(indexer, slot, acceptUpdatesToLocal, machineSet);
+ updateExpectedSize();
+ }
+ }
- if (commit == NULL) {
- // This is a new commit that we dont have so make a new one
- commit = new Commit();
+ // If there is a gap, check to see if the server sent us
+ // everything->
+ if (firstSeqNum != (sequenceNumber + 1)) {
- // Insert this new commit into the live tables
- commitForClientTable.put(part.getSequenceNumber(), commit);
- }
+ // Check the size of the slots that were sent down by the server->
+ // Can only check the size if there was a gap
+ checkNumSlots(newSlots->length());
- // Add that part to the commit
- commit.addPartDecode(part);
- }
+ // Since there was a gap every machine must have pushed a slot or
+ // must have a last message message-> If not then the server is
+ // hiding slots
+ if (!machineSet->isEmpty()) {
+ throw new Error("Missing record for machines: ");
}
+ }
- // Clear all the new commits parts in preparation for the next time the server sends slots
- newCommitParts.clear();
+ // Update the size of our local block chain->
+ commitNewMaxSize();
+
+ // Commit new to slots to the local block chain->
+ {
+ uint numSlots = newSlots->length();
+ for(uint i=0; i<numSlots; i++) {
+ Slot *slot = newSlots->get(i);
+
+ // Insert this slot into our local block chain copy->
+ buffer->putSlot(slot);
+
+ // Keep track of how many slots are currently live (have live data
+ // in them)->
+ liveSlotCount++;
+ }
+ }
+ // Get the sequence number of the latest slot in the system
+ sequenceNumber = newSlots->get(newSlots->length() - 1)->getSequenceNumber();
+ updateLiveStateFromServer();
- // If we process a new commit keep track of it for future use
- bool didProcessANewCommit = false;
+ // No Need to remember after we pulled from the server
+ offlineTransactionsCommittedAndAtServer->clear();
- // Process the commits one by one
- for (Long arbitratorId : liveCommitsTable.keySet()) {
+ // This is invalidated now
+ hadPartialSendToServer = false;
+}
- // Get all the commits for a specific arbitrator
- Hashtable<Long, Commit> commitForClientTable = liveCommitsTable.get(arbitratorId);
+void Table::updateLiveStateFromServer() {
+ // Process the new transaction parts
+ processNewTransactionParts();
- // Sort the commits in order
- List<Long> commitSequenceNumbers = new ArrayList<Long>(commitForClientTable.keySet());
- Collections.sort(commitSequenceNumbers);
+ // Do arbitration on new transactions that were received
+ arbitrateFromServer();
- // Get the last commit seen from this arbitrator
- int64_t lastCommitSeenSequenceNumber = -1;
- if (lastCommitSeenSequenceNumberByArbitratorTable.get(arbitratorId) != NULL) {
- lastCommitSeenSequenceNumber = lastCommitSeenSequenceNumberByArbitratorTable.get(arbitratorId);
- }
+ // Update all the committed keys
+ bool didCommitOrSpeculate = updateCommittedTable();
- // Go through each new commit one by one
- for (int i = 0; i < commitSequenceNumbers.size(); i++) {
- Long commitSequenceNumber = commitSequenceNumbers.get(i);
- Commit commit = commitForClientTable.get(commitSequenceNumber);
+ // Delete the transactions that are now dead
+ updateLiveTransactionsAndStatus();
- // Special processing if a commit is not complete
- if (!commit.isComplete()) {
- if (i == (commitSequenceNumbers.size() - 1)) {
- // If there is an incomplete commit and this commit is the latest one seen then this commit cannot be processed and there are no other commits
- break;
- } else {
- // This is a commit that was already dead but parts of it are still in the block chain (not flushed out yet).
- // Delete it and move on
- commit.setDead();
- commitForClientTable.remove(commit.getSequenceNumber());
- continue;
- }
- }
+ // Do speculations
+ didCommitOrSpeculate |= updateSpeculativeTable(didCommitOrSpeculate);
+ updatePendingTransactionSpeculativeTable(didCommitOrSpeculate);
+}
- // Update the last transaction that was updated if we can
- if (commit.getTransactionSequenceNumber() != -1) {
- Long lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable.get(commit.getMachineId());
+void Table::updateLiveStateFromLocal() {
+ // Update all the committed keys
+ bool didCommitOrSpeculate = updateCommittedTable();
- // Update the last transaction sequence number that the arbitrator arbitrated on
- if ((lastTransactionNumber == NULL) || (lastTransactionNumber < commit.getTransactionSequenceNumber())) {
- lastArbitratedTransactionNumberByArbitratorTable.put(commit.getMachineId(), commit.getTransactionSequenceNumber());
- }
- }
+ // Delete the transactions that are now dead
+ updateLiveTransactionsAndStatus();
- // Update the last arbitration data that we have seen so far
- if (lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.get(commit.getMachineId()) != NULL) {
+ // Do speculations
+ didCommitOrSpeculate |= updateSpeculativeTable(didCommitOrSpeculate);
+ updatePendingTransactionSpeculativeTable(didCommitOrSpeculate);
+}
- int64_t lastArbitrationSequenceNumber = lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.get(commit.getMachineId());
- if (commit.getSequenceNumber() > lastArbitrationSequenceNumber) {
- // Is larger
- lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.put(commit.getMachineId(), commit.getSequenceNumber());
- }
- } else {
- // Never seen any data from this arbitrator so record the first one
- lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.put(commit.getMachineId(), commit.getSequenceNumber());
- }
+void Table::initExpectedSize(int64_t firstSequenceNumber, int64_t numberOfSlots) {
+ int64_t prevslots = firstSequenceNumber;
- // We have already seen this commit before so need to do the full processing on this commit
- if (commit.getSequenceNumber() <= lastCommitSeenSequenceNumber) {
+ if (didFindTableStatus) {
+ } else {
+ expectedsize = (prevslots < ((int64_t) numberOfSlots)) ? (int) prevslots : numberOfSlots;
+ }
- // Update the last transaction that was updated if we can
- if (commit.getTransactionSequenceNumber() != -1) {
- Long lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable.get(commit.getMachineId());
+ didFindTableStatus = true;
+ currMaxSize = numberOfSlots;
+}
- // Update the last transaction sequence number that the arbitrator arbitrated on
- if ((lastTransactionNumber == NULL) || (lastTransactionNumber < commit.getTransactionSequenceNumber())) {
- lastArbitratedTransactionNumberByArbitratorTable.put(commit.getMachineId(), commit.getTransactionSequenceNumber());
- }
- }
+void Table::updateExpectedSize() {
+ expectedsize++;
- continue;
- }
+ if (expectedsize > currMaxSize) {
+ expectedsize = currMaxSize;
+ }
+}
- // If we got here then this is a brand new commit and needs full processing
- // Get what commits should be edited, these are the commits that have live values for their keys
- Set<Commit> commitsToEdit = new HashSet<Commit>();
- for (KeyValue kv : commit.getKeyValueUpdateSet()) {
- commitsToEdit.add(liveCommitsByKeyTable.get(kv.getKey()));
- }
- commitsToEdit.remove(NULL); // remove NULL since it could be in this set
+/**
+ * Check the size of the block chain to make sure there are enough
+ * slots sent back by the server-> This is only called when we have a
+ * gap between the slots that we have locally and the slots sent by
+ * the server therefore in the slots sent by the server there will be
+ * at least 1 Table status message
+ */
+void Table::checkNumSlots(int numberOfSlots) {
+ if (numberOfSlots != expectedsize) {
+ throw new Error("Server Error: Server did not send all slots-> Expected: ");
+ }
+}
- // Update each previous commit that needs to be updated
- for (Commit previousCommit : commitsToEdit) {
+/**
+ * Update the size of of the local buffer if it is needed->
+ */
+void Table::commitNewMaxSize() {
+ didFindTableStatus = false;
- // Only bother with live commits (TODO: Maybe remove this check)
- if (previousCommit.isLive()) {
+ // Resize the local slot buffer
+ if (numberOfSlots != currMaxSize) {
+ buffer->resize((int32_t)currMaxSize);
+ }
- // Update which keys in the old commits are still live
- for (KeyValue kv : commit.getKeyValueUpdateSet()) {
- previousCommit.invalidateKey(kv.getKey());
- }
+ // Change the number of local slots to the new size
+ numberOfSlots = (int32_t)currMaxSize;
- // if the commit is now dead then remove it
- if (!previousCommit.isLive()) {
- commitForClientTable.remove(previousCommit);
- }
- }
- }
+ // Recalculate the resize threshold since the size of the local
+ // buffer has changed
+ setResizeThreshold();
+}
- // Update the last seen sequence number from this arbitrator
- if (lastCommitSeenSequenceNumberByArbitratorTable.get(commit.getMachineId()) != NULL) {
- if (commit.getSequenceNumber() > lastCommitSeenSequenceNumberByArbitratorTable.get(commit.getMachineId())) {
- lastCommitSeenSequenceNumberByArbitratorTable.put(commit.getMachineId(), commit.getSequenceNumber());
- }
- } else {
- lastCommitSeenSequenceNumberByArbitratorTable.put(commit.getMachineId(), commit.getSequenceNumber());
- }
+/**
+ * Process the new transaction parts from this latest round of slots
+ * received from the server
+ */
+void Table::processNewTransactionParts() {
- // We processed a new commit that we havent seen before
- didProcessANewCommit = true;
+ if (newTransactionParts->size() == 0) {
+ // Nothing new to process
+ return;
+ }
- // Update the committed table of keys and which commit is using which key
- for (KeyValue kv : commit.getKeyValueUpdateSet()) {
- committedKeyValueTable.put(kv.getKey(), kv);
- liveCommitsByKeyTable.put(kv.getKey(), commit);
+ // Iterate through all the machine Ids that we received new parts
+ // for
+ SetIterator<int64_t, Hashtable<Pair<int64_t, int32_t> *, TransactionPart *, uintptr_t, 0, pairHashFunction, pairEquals> *> * tpit= getKeyIterator(newTransactionParts);
+ while(tpit->hasNext()) {
+ int64_t machineId = tpit->next();
+ Hashtable<Pair<int64_t, int32_t> *, TransactionPart *, uintptr_t, 0, pairHashFunction, pairEquals> *parts = newTransactionParts->get(machineId);
+
+ SetIterator<Pair<int64_t, int32_t> *, TransactionPart *, uintptr_t, 0, pairHashFunction, pairEquals> *ptit = getKeyIterator(parts);
+ // Iterate through all the parts for that machine Id
+ while(ptit->hasNext()) {
+ Pair<int64_t, int32_t> * partId = ptit->next();
+ TransactionPart *part = parts->get(partId);
+
+ if (lastArbitratedTransactionNumberByArbitratorTable->contains(part->getArbitratorId())) {
+ int64_t lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable->get(part->getArbitratorId());
+ if (lastTransactionNumber >= part->getSequenceNumber()) {
+ // Set dead the transaction part
+ part->setDead();
+ continue;
}
}
- }
+
+ // Get the transaction object for that sequence number
+ Transaction *transaction = liveTransactionBySequenceNumberTable->get(part->getSequenceNumber());
- return didProcessANewCommit;
- }
+ if (transaction == NULL) {
+ // This is a new transaction that we dont have so make a new one
+ transaction = new Transaction();
- /**
- * Create the speculative table from transactions that are still live and have come from the cloud
- */
- bool updateSpeculativeTable(bool didProcessNewCommits) {
- if (liveTransactionBySequenceNumberTable.keySet().size() == 0) {
- // There is nothing to speculate on
- return false;
+ // Insert this new transaction into the live tables
+ liveTransactionBySequenceNumberTable->put(part->getSequenceNumber(), transaction);
+ liveTransactionByTransactionIdTable->put(new Pair<int64_t, int64_t>(part->getTransactionId()), transaction);
+ }
+
+ // Add that part to the transaction
+ transaction->addPartDecode(part);
}
+ delete ptit;
+ }
+ delete tpit;
+ // Clear all the new transaction parts in preparation for the next
+ // time the server sends slots
+ newTransactionParts->clear();
+}
- // Create a list of the transaction sequence numbers and sort them from oldest to newest
- List<Long> transactionSequenceNumbersSorted = new ArrayList<Long>(liveTransactionBySequenceNumberTable.keySet());
- Collections.sort(transactionSequenceNumbersSorted);
+void Table::arbitrateFromServer() {
- bool hasGapInTransactionSequenceNumbers = transactionSequenceNumbersSorted.get(0) != oldestTransactionSequenceNumberSpeculatedOn;
+ if (liveTransactionBySequenceNumberTable->size() == 0) {
+ // Nothing to arbitrate on so move on
+ return;
+ }
+ // Get the transaction sequence numbers and sort from oldest to newest
+ Vector<int64_t> *transactionSequenceNumbers = new Vector<int64_t>();
+ {
+ SetIterator<int64_t, Transaction *> * trit = getKeyIterator(liveTransactionBySequenceNumberTable);
+ while(trit->hasNext())
+ transactionSequenceNumbers->add(trit->next());
+ delete trit;
+ }
+ qsort(transactionSequenceNumbers->expose(), transactionSequenceNumbers->size(), sizeof(int64_t), compareInt64);
- if (hasGapInTransactionSequenceNumbers || didProcessNewCommits) {
- // If there is a gap in the transaction sequence numbers then there was a commit or an abort of a transaction
- // OR there was a new commit (Could be from offline commit) so a redo the speculation from scratch
+ // Collection of key value pairs that are
+ Hashtable<IoTString *, KeyValue *> * speculativeTableTmp = new Hashtable<IoTString *, KeyValue *>();
- // Start from scratch
- speculatedKeyValueTable.clear();
- lastTransactionSequenceNumberSpeculatedOn = -1;
- oldestTransactionSequenceNumberSpeculatedOn = -1;
+ // The last transaction arbitrated on
+ int64_t lastTransactionCommitted = -1;
+ Hashset<Abort *> *generatedAborts = new Hashset<Abort *>();
+ uint tsnSize = transactionSequenceNumbers->size();
+ for(uint i=0; i<tsnSize; i++) {
+ int64_t transactionSequenceNumber = transactionSequenceNumbers->get(i);
+ Transaction *transaction = liveTransactionBySequenceNumberTable->get(transactionSequenceNumber);
+ // Check if this machine arbitrates for this transaction if not
+ // then we cant arbitrate this transaction
+ if (transaction->getArbitrator() != localMachineId) {
+ continue;
}
- // Remember the front of the transaction list
- oldestTransactionSequenceNumberSpeculatedOn = transactionSequenceNumbersSorted.get(0);
-
- // Find where to start arbitration from
- int startIndex = transactionSequenceNumbersSorted.indexOf(lastTransactionSequenceNumberSpeculatedOn) + 1;
+ if (transactionSequenceNumber < lastSeqNumArbOn) {
+ continue;
+ }
- if (startIndex >= transactionSequenceNumbersSorted.size()) {
- // Make sure we are not out of bounds
- return false; // did not speculate
+ if (offlineTransactionsCommittedAndAtServer->contains(transaction->getId())) {
+ // We have seen this already locally so dont commit again
+ continue;
}
- Set<Long> incompleteTransactionArbitrator = new HashSet<Long>();
- bool didSkip = true;
- for (int i = startIndex; i < transactionSequenceNumbersSorted.size(); i++) {
- int64_t transactionSequenceNumber = transactionSequenceNumbersSorted.get(i);
- Transaction transaction = liveTransactionBySequenceNumberTable.get(transactionSequenceNumber);
+ if (!transaction->isComplete()) {
+ // Will arbitrate in incorrect order if we continue so just break
+ // Most likely this
+ break;
+ }
- if (!transaction.isComplete()) {
- // If there is an incomplete transaction then there is nothing we can do
- // add this transactions arbitrator to the list of arbitrators we should ignore
- incompleteTransactionArbitrator.add(transaction.getArbitrator());
- didSkip = true;
- continue;
- }
- if (incompleteTransactionArbitrator.contains(transaction.getArbitrator())) {
- continue;
+ // update the largest transaction seen by arbitrator from server
+ if (!lastTransactionSeenFromMachineFromServer->contains(transaction->getMachineId())) {
+ lastTransactionSeenFromMachineFromServer->put(transaction->getMachineId(), transaction->getClientLocalSequenceNumber());
+ } else {
+ int64_t lastTransactionSeenFromMachine = lastTransactionSeenFromMachineFromServer->get(transaction->getMachineId());
+ if (transaction->getClientLocalSequenceNumber() > lastTransactionSeenFromMachine) {
+ lastTransactionSeenFromMachineFromServer->put(transaction->getMachineId(), transaction->getClientLocalSequenceNumber());
}
+ }
- lastTransactionSequenceNumberSpeculatedOn = transactionSequenceNumber;
+ if (transaction->evaluateGuard(committedKeyValueTable, speculativeTableTmp, NULL)) {
+ // Guard evaluated as true
- if (transaction.evaluateGuard(committedKeyValueTable, speculatedKeyValueTable, NULL)) {
- // Guard evaluated to true so update the speculative table
- for (KeyValue kv : transaction.getKeyValueUpdateSet()) {
- speculatedKeyValueTable.put(kv.getKey(), kv);
- }
- }
- }
+ // Update the local changes so we can make the commit
+ SetIterator<KeyValue *, KeyValue *> *kvit = transaction->getKeyValueUpdateSet()->iterator();
+ while (kvit->hasNext()) {
+ KeyValue *kv = kvit->next();
+ speculativeTableTmp->put(kv->getKey(), kv);
+ }
+ delete kvit;
+
+ // Update what the last transaction committed was for use in batch commit
+ lastTransactionCommitted = transactionSequenceNumber;
+ } else {
+ // Guard evaluated was false so create abort
+ // Create the abort
+ Abort *newAbort = new Abort(NULL,
+ transaction->getClientLocalSequenceNumber(),
+ transaction->getSequenceNumber(),
+ transaction->getMachineId(),
+ transaction->getArbitrator(),
+ localArbitrationSequenceNumber);
+ localArbitrationSequenceNumber++;
+ generatedAborts->add(newAbort);
- if (didSkip) {
- // Since there was a skip we need to redo the speculation next time around
- lastTransactionSequenceNumberSpeculatedOn = -1;
- oldestTransactionSequenceNumberSpeculatedOn = -1;
+ // Insert the abort so we can process
+ processEntry(newAbort);
}
- // We did some speculation
- return true;
+ lastSeqNumArbOn = transactionSequenceNumber;
}
- /**
- * Create the pending transaction speculative table from transactions that are still in the pending transaction buffer
- */
- void updatePendingTransactionSpeculativeTable(bool didProcessNewCommitsOrSpeculate) {
- if (pendingTransactionQueue.size() == 0) {
- // There is nothing to speculate on
- return;
- }
+ Commit *newCommit = NULL;
+ // If there is something to commit
+ if (speculativeTableTmp->size() != 0) {
+ // Create the commit and increment the commit sequence number
+ newCommit = new Commit(localArbitrationSequenceNumber, localMachineId, lastTransactionCommitted);
+ localArbitrationSequenceNumber++;
- if (didProcessNewCommitsOrSpeculate || (firstPendingTransaction != pendingTransactionQueue.get(0))) {
- // need to reset on the pending speculation
- lastPendingTransactionSpeculatedOn = NULL;
- firstPendingTransaction = pendingTransactionQueue.get(0);
- pendingTransactionSpeculatedKeyValueTable.clear();
+ // Add all the new keys to the commit
+ for (KeyValue *kv : speculativeTableTmp->values()) {
+ newCommit->addKV(kv);
}
- // Find where to start arbitration from
- int startIndex = pendingTransactionQueue.indexOf(firstPendingTransaction) + 1;
+ // create the commit parts
+ newCommit->createCommitParts();
- if (startIndex >= pendingTransactionQueue.size()) {
- // Make sure we are not out of bounds
- return;
+ // Append all the commit parts to the end of the pending queue
+ // waiting for sending to the server
+ // Insert the commit so we can process it
+ for (CommitPart *commitPart : newCommit->getParts()->values()) {
+ processEntry(commitPart);
}
+ }
- for (int i = startIndex; i < pendingTransactionQueue.size(); i++) {
- Transaction transaction = pendingTransactionQueue.get(i);
-
- lastPendingTransactionSpeculatedOn = transaction;
+ if ((newCommit != NULL) || (generatedAborts->size() > 0)) {
+ ArbitrationRound *arbitrationRound = new ArbitrationRound(newCommit, generatedAborts);
+ pendingSendArbitrationRounds->add(arbitrationRound);
- if (transaction.evaluateGuard(committedKeyValueTable, speculatedKeyValueTable, pendingTransactionSpeculatedKeyValueTable)) {
- // Guard evaluated to true so update the speculative table
- for (KeyValue kv : transaction.getKeyValueUpdateSet()) {
- pendingTransactionSpeculatedKeyValueTable.put(kv.getKey(), kv);
+ if (compactArbitrationData()) {
+ ArbitrationRound *newArbitrationRound = pendingSendArbitrationRounds->get(pendingSendArbitrationRounds->size() - 1);
+ if (newArbitrationRound->getCommit() != NULL) {
+ for (CommitPart *commitPart : newArbitrationRound->getCommit()->getParts()->values()) {
+ processEntry(commitPart);
}
}
}
}
+}
- /**
- * Set dead and remove from the live transaction tables the transactions that are dead
- */
- void updateLiveTransactionsAndStatus() {
+Pair<bool, bool> Table::arbitrateOnLocalTransaction(Transaction *transaction) {
- // Go through each of the transactions
- for (Iterator<Map.Entry<Long, Transaction>> iter = liveTransactionBySequenceNumberTable.entrySet().iterator(); iter.hasNext();) {
- Transaction transaction = iter.next().getValue();
+ // Check if this machine arbitrates for this transaction if not then
+ // we cant arbitrate this transaction
+ if (transaction->getArbitrator() != localMachineId) {
+ return Pair<bool, bool>(false, false);
+ }
- // Check if the transaction is dead
- Long lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable.get(transaction.getArbitrator());
- if ((lastTransactionNumber != NULL) && (lastTransactionNumber >= transaction.getSequenceNumber())) {
+ if (!transaction->isComplete()) {
+ // Will arbitrate in incorrect order if we continue so just break
+ // Most likely this
+ return Pair<bool, bool>(false, false);
+ }
- // Set dead the transaction
- transaction.setDead();
+ if (transaction->getMachineId() != localMachineId) {
+ // dont do this check for local transactions
+ if (lastTransactionSeenFromMachineFromServer->get(transaction->getMachineId()) != NULL) {
+ if (lastTransactionSeenFromMachineFromServer->get(transaction->getMachineId()) > transaction->getClientLocalSequenceNumber()) {
+ // We've have already seen this from the server
+ return Pair<bool, bool>(false, false);
+ }
+ }
+ }
- // Remove the transaction from the live table
- iter.remove();
- liveTransactionByTransactionIdTable.remove(transaction.getId());
+ if (transaction->evaluateGuard(committedKeyValueTable, NULL, NULL)) {
+ // Guard evaluated as true Create the commit and increment the
+ // commit sequence number
+ Commit *newCommit = new Commit(localArbitrationSequenceNumber, localMachineId, -1);
+ localArbitrationSequenceNumber++;
+
+ // Update the local changes so we can make the commit
+ SetIterator<KeyValue *, KeyValue *> *kvit = transaction->getKeyValueUpdateSet()->iterator();
+ while (kvit->hasNext()) {
+ KeyValue *kv = kvit->next();
+ newCommit->addKV(kv);
+ }
+ delete kvit;
+
+ // create the commit parts
+ newCommit->createCommitParts();
+
+ // Append all the commit parts to the end of the pending queue
+ // waiting for sending to the server
+ ArbitrationRound *arbitrationRound = new ArbitrationRound(newCommit, new Hashset<Abort *>());
+ pendingSendArbitrationRounds->add(arbitrationRound);
+
+ if (compactArbitrationData()) {
+ ArbitrationRound *newArbitrationRound = pendingSendArbitrationRounds->get(pendingSendArbitrationRounds->size() - 1);
+ for (CommitPart *commitPart : newArbitrationRound->getCommit()->getParts()->values()) {
+ processEntry(commitPart);
+ }
+ } else {
+ // Insert the commit so we can process it
+ for (CommitPart *commitPart : newCommit->getParts()->values()) {
+ processEntry(commitPart);
}
}
- // Go through each of the transactions
- for (Iterator<Map.Entry<Long, TransactionStatus>> iter = outstandingTransactionStatus.entrySet().iterator(); iter.hasNext();) {
- TransactionStatus status = iter.next().getValue();
+ if (transaction->getMachineId() == localMachineId) {
+ TransactionStatus *status = transaction->getTransactionStatus();
+ if (status != NULL) {
+ status->setStatus(TransactionStatus_StatusCommitted);
+ }
+ }
- // Check if the transaction is dead
- Long lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable.get(status.getTransactionArbitrator());
- if ((lastTransactionNumber != NULL) && (lastTransactionNumber >= status.getTransactionSequenceNumber())) {
+ updateLiveStateFromLocal();
+ return Pair<bool, bool>(true, true);
+ } else {
+ if (transaction->getMachineId() == localMachineId) {
+ // For locally created messages update the status
+ // Guard evaluated was false so create abort
+ TransactionStatus * status = transaction->getTransactionStatus();
+ if (status != NULL) {
+ status->setStatus(TransactionStatus_StatusAborted);
+ }
+ } else {
+ Hashset<Abort *> *addAbortSet = new Hashset<Abort * >();
+
+ // Create the abort
+ Abort *newAbort = new Abort(NULL,
+ transaction->getClientLocalSequenceNumber(),
+ -1,
+ transaction->getMachineId(),
+ transaction->getArbitrator(),
+ localArbitrationSequenceNumber);
+ localArbitrationSequenceNumber++;
+ addAbortSet->add(newAbort);
- // Set committed
- status.setStatus(TransactionStatus.StatusCommitted);
+ // Append all the commit parts to the end of the pending queue
+ // waiting for sending to the server
+ ArbitrationRound *arbitrationRound = new ArbitrationRound(NULL, addAbortSet);
+ pendingSendArbitrationRounds->add(arbitrationRound);
- // Remove
- iter.remove();
+ if (compactArbitrationData()) {
+ ArbitrationRound *newArbitrationRound = pendingSendArbitrationRounds->get(pendingSendArbitrationRounds->size() - 1);
+ for (CommitPart *commitPart : newArbitrationRound->getCommit()->getParts()->values()) {
+ processEntry(commitPart);
+ }
}
}
+
+ updateLiveStateFromLocal();
+ return Pair<bool, bool>(true, false);
+ }
+}
+
+/**
+ * Compacts the arbitration data my merging commits and aggregating
+ * aborts so that a single large push of commits can be done instead
+ * of many small updates
+ */
+bool Table::compactArbitrationData() {
+ if (pendingSendArbitrationRounds->size() < 2) {
+ // Nothing to compact so do nothing
+ return false;
}
- /**
- * Process this slot, entry by entry. Also update the latest message sent by slot
- */
- void processSlot(SlotIndexer indexer, Slot slot, bool acceptUpdatesToLocal, HashSet<Long> machineSet) {
+ ArbitrationRound *lastRound = pendingSendArbitrationRounds->get(pendingSendArbitrationRounds->size() - 1);
+ if (lastRound->getDidSendPart()) {
+ return false;
+ }
- // Update the last message seen
- updateLastMessage(slot.getMachineID(), slot.getSequenceNumber(), slot, acceptUpdatesToLocal, machineSet);
+ bool hadCommit = (lastRound->getCommit() == NULL);
+ bool gotNewCommit = false;
- // Process each entry in the slot
- for (Entry entry : slot.getEntries()) {
- switch (entry.getType()) {
+ int numberToDelete = 1;
+ while (numberToDelete < pendingSendArbitrationRounds->size()) {
+ ArbitrationRound *round = pendingSendArbitrationRounds->get(pendingSendArbitrationRounds->size() - numberToDelete - 1);
- case Entry.TypeCommitPart:
- processEntry((CommitPart)entry);
- break;
+ if (round->isFull() || round->getDidSendPart()) {
+ // Stop since there is a part that cannot be compacted and we
+ // need to compact in order
+ break;
+ }
- case Entry.TypeAbort:
- processEntry((Abort)entry);
+ if (round->getCommit() == NULL) {
+ // Try compacting aborts only
+ int newSize = round->getCurrentSize() + lastRound->getAbortsCount();
+ if (newSize > ArbitrationRound_MAX_PARTS) {
+ // Cant compact since it would be too large
break;
+ }
+ lastRound->addAborts(round->getAborts());
+ } else {
+ // Create a new larger commit
+ Commit * newCommit = Commit_merge(lastRound->getCommit(), round->getCommit(), localArbitrationSequenceNumber);
+ localArbitrationSequenceNumber++;
- case Entry.TypeTransactionPart:
- processEntry((TransactionPart)entry);
- break;
+ // Create the commit parts so that we can count them
+ newCommit->createCommitParts();
- case Entry.TypeNewKey:
- processEntry((NewKey)entry);
- break;
+ // Calculate the new size of the parts
+ int newSize = newCommit->getNumberOfParts();
+ newSize += lastRound->getAbortsCount();
+ newSize += round->getAbortsCount();
- case Entry.TypeLastMessage:
- processEntry((LastMessage)entry, machineSet);
+ if (newSize > ArbitrationRound_MAX_PARTS) {
+ // Cant compact since it would be too large
break;
+ }
- case Entry.TypeRejectedMessage:
- processEntry((RejectedMessage)entry, indexer);
- break;
+ // Set the new compacted part
+ lastRound->setCommit(newCommit);
+ lastRound->addAborts(round->getAborts());
+ gotNewCommit = true;
+ }
- case Entry.TypeTableStatus:
- processEntry((TableStatus)entry, slot.getSequenceNumber());
- break;
+ numberToDelete++;
+ }
- default:
- throw new Error("Unrecognized type: " + entry.getType());
+ if (numberToDelete != 1) {
+ // If there is a compaction
+ // Delete the previous pieces that are now in the new compacted piece
+ if (numberToDelete == pendingSendArbitrationRounds->size()) {
+ pendingSendArbitrationRounds->clear();
+ } else {
+ for (int i = 0; i < numberToDelete; i++) {
+ pendingSendArbitrationRounds->removeIndex(pendingSendArbitrationRounds->size() - 1);
}
}
+
+ // Add the new compacted into the pending to send list
+ pendingSendArbitrationRounds->add(lastRound);
+
+ // Should reinsert into the commit processor
+ if (hadCommit && gotNewCommit) {
+ return true;
+ }
}
- /**
- * Update the last message that was sent for a machine Id
- */
- void processEntry(LastMessage entry, HashSet<Long> machineSet) {
- // Update what the last message received by a machine was
- updateLastMessage(entry.getMachineID(), entry.getSequenceNumber(), entry, false, machineSet);
+ return false;
+}
+
+/**
+ * Update all the commits and the committed tables, sets dead the dead
+ * transactions
+ */
+bool Table::updateCommittedTable() {
+
+ if (newCommitParts->size() == 0) {
+ // Nothing new to process
+ return false;
}
- /**
- * Add the new key to the arbitrators table and update the set of live new keys (in case of a rescued new key message)
- */
- void processEntry(NewKey entry) {
+ // Iterate through all the machine Ids that we received new parts for
+ for (int64_t machineId : newCommitParts->keySet()) {
+ Hashtable<Pair<int64_t, int32_t> *, CommitPart *> *parts = newCommitParts->get(machineId);
+
+ // Iterate through all the parts for that machine Id
+ for (Pair<int64_t, int32_t> partId : parts->keySet()) {
+ CommitPart *part = parts->get(partId);
+
+ // Get the transaction object for that sequence number
+ Hashtable<int64_t, Commit *> *commitForClientTable = liveCommitsTable->get(part->getMachineId());
+
+ if (commitForClientTable == NULL) {
+ // This is the first commit from this device
+ commitForClientTable = new Hashtable<int64_t, Commit *>();
+ liveCommitsTable->put(part->getMachineId(), commitForClientTable);
+ }
+
+ Commit *commit = commitForClientTable->get(part->getSequenceNumber());
+
+ if (commit == NULL) {
+ // This is a new commit that we dont have so make a new one
+ commit = new Commit();
- // Update the arbitrator table with the new key information
- arbitratorTable.put(entry.getKey(), entry.getMachineID());
+ // Insert this new commit into the live tables
+ commitForClientTable->put(part->getSequenceNumber(), commit);
+ }
- // Update what the latest live new key is
- NewKey oldNewKey = liveNewKeyTable.put(entry.getKey(), entry);
- if (oldNewKey != NULL) {
- // Delete the old new key messages
- oldNewKey.setDead();
+ // Add that part to the commit
+ commit->addPartDecode(part);
}
}
- /**
- * Process new table status entries and set dead the old ones as new ones come in.
- * keeps track of the largest and smallest table status seen in this current round
- * of updating the local copy of the block chain
- */
- void processEntry(TableStatus entry, int64_t seq) {
- int newNumSlots = entry.getMaxSlots();
- updateCurrMaxSize(newNumSlots);
+ // Clear all the new commits parts in preparation for the next time
+ // the server sends slots
+ newCommitParts->clear();
- initExpectedSize(seq, newNumSlots);
+ // If we process a new commit keep track of it for future use
+ bool didProcessANewCommit = false;
- if (liveTableStatus != NULL) {
- // We have a larger table status so the old table status is no int64_ter alive
- liveTableStatus.setDead();
- }
+ // Process the commits one by one
+ for (int64_T arbitratorId : liveCommitsTable->keySet()) {
- // Make this new table status the latest alive table status
- liveTableStatus = entry;
- }
+ // Get all the commits for a specific arbitrator
+ Hashtable<int64_t, Commit *> *commitForClientTable = liveCommitsTable->get(arbitratorId);
+
+ // Sort the commits in order
+ Vector<int64_t> *commitSequenceNumbers = new Vector<int64_t>(commitForClientTable->keySet());
+ qsort(commitSequenceNumbers->expose(), commitSequenceNumbers->size(), sizeof(int64_t), compareInt64);
- /**
- * Check old messages to see if there is a block chain violation. Also
- */
- void processEntry(RejectedMessage entry, SlotIndexer indexer) {
- int64_t oldSeqNum = entry.getOldSeqNum();
- int64_t newSeqNum = entry.getNewSeqNum();
- bool isequal = entry.getEqual();
- int64_t machineId = entry.getMachineID();
- int64_t seq = entry.getSequenceNumber();
+ // Get the last commit seen from this arbitrator
+ int64_t lastCommitSeenSequenceNumber = -1;
+ if (lastCommitSeenSequenceNumberByArbitratorTable->get(arbitratorId) != NULL) {
+ lastCommitSeenSequenceNumber = lastCommitSeenSequenceNumberByArbitratorTable->get(arbitratorId);
+ }
+ // Go through each new commit one by one
+ for (int i = 0; i < commitSequenceNumbers->size(); i++) {
+ int64_t commitSequenceNumber = commitSequenceNumbers->get(i);
+ Commit *commit = commitForClientTable->get(commitSequenceNumber);
- // Check if we have messages that were supposed to be rejected in our local block chain
- for (int64_t seqNum = oldSeqNum; seqNum <= newSeqNum; seqNum++) {
+ // Special processing if a commit is not complete
+ if (!commit->isComplete()) {
+ if (i == (commitSequenceNumbers->size() - 1)) {
+ // If there is an incomplete commit and this commit is the
+ // latest one seen then this commit cannot be processed and
+ // there are no other commits
+ break;
+ } else {
+ // This is a commit that was already dead but parts of it
+ // are still in the block chain (not flushed out yet)->
+ // Delete it and move on
+ commit->setDead();
+ commitForClientTable->remove(commit->getSequenceNumber());
+ continue;
+ }
+ }
- // Get the slot
- Slot slot = indexer.getSlot(seqNum);
+ // Update the last transaction that was updated if we can
+ if (commit->getTransactionSequenceNumber() != -1) {
+ int64_t lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable->get(commit->getMachineId());
- if (slot != NULL) {
- // If we have this slot make sure that it was not supposed to be a rejected slot
+ // Update the last transaction sequence number that the arbitrator arbitrated on
+ if ((lastTransactionNumber == NULL) || (lastTransactionNumber < commit->getTransactionSequenceNumber())) {
+ lastArbitratedTransactionNumberByArbitratorTable->put(commit->getMachineId(), commit->getTransactionSequenceNumber());
+ }
+ }
- int64_t slotMachineId = slot.getMachineID();
- if (isequal != (slotMachineId == machineId)) {
- throw new Error("Server Error: Trying to insert rejected message for slot " + seqNum);
+ // Update the last arbitration data that we have seen so far
+ if (lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->contains(commit->getMachineId())) {
+ int64_t lastArbitrationSequenceNumber = lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->get(commit->getMachineId());
+ if (commit->getSequenceNumber() > lastArbitrationSequenceNumber) {
+ // Is larger
+ lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->put(commit->getMachineId(), commit->getSequenceNumber());
}
+ } else {
+ // Never seen any data from this arbitrator so record the first one
+ lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->put(commit->getMachineId(), commit->getSequenceNumber());
}
- }
+ // We have already seen this commit before so need to do the
+ // full processing on this commit
+ if (commit->getSequenceNumber() <= lastCommitSeenSequenceNumber) {
- // Create a list of clients to watch until they see this rejected message entry.
- HashSet<Long> deviceWatchSet = new HashSet<Long>();
- for (Map.Entry<Long, Pair<Long, Liveness>> lastMessageEntry : lastMessageTable.entrySet()) {
+ // Update the last transaction that was updated if we can
+ if (commit->getTransactionSequenceNumber() != -1) {
+ int64_t lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable->get(commit->getMachineId());
- // Machine ID for the last message entry
- int64_t lastMessageEntryMachineId = lastMessageEntry.getKey();
+ // Update the last transaction sequence number that the arbitrator arbitrated on
+ if ((lastTransactionNumber == NULL) || (lastTransactionNumber < commit->getTransactionSequenceNumber())) {
+ lastArbitratedTransactionNumberByArbitratorTable->put(commit->getMachineId(), commit->getTransactionSequenceNumber());
+ }
+ }
- // We've seen it, don't need to continue to watch. Our next
- // message will implicitly acknowledge it.
- if (lastMessageEntryMachineId == localMachineId) {
continue;
}
- Pair<Long, Liveness> lastMessageValue = lastMessageEntry.getValue();
- int64_t entrySequenceNumber = lastMessageValue.getFirst();
+ // If we got here then this is a brand new commit and needs full
+ // processing
+ // Get what commits should be edited, these are the commits that
+ // have live values for their keys
+ Hashset<Commit *> *commitsToEdit = new Hashset<Commit *>();
+ {
+ SetIterator<KeyValue *, KeyValue *, uintptr_t, 0, hashKeyValue, equalsKeyValue> *kvit = commit->getKeyValueUpdateSet()->iterator();
+ while (kvit->hasNext()) {
+ KeyValue *kv = kvit->next();
+ commitsToEdit->add(liveCommitsByKeyTable->get(kv->getKey()));
+ }
+ delete kvit;
+ }
+ commitsToEdit->remove(NULL); // remove NULL since it could be in this set
+
+ // Update each previous commit that needs to be updated
+ for (Commit *previousCommit : commitsToEdit) {
- if (entrySequenceNumber < seq) {
+ // Only bother with live commits (TODO: Maybe remove this check)
+ if (previousCommit->isLive()) {
- // Add this rejected message to the set of messages that this machine ID did not see yet
- addWatchList(lastMessageEntryMachineId, entry);
+ // Update which keys in the old commits are still live
+ {
+ SetIterator<KeyValue *, KeyValue *, uintptr_t, 0, hashKeyValue, equalsKeyValue> *kvit = commit->getKeyValueUpdateSet()->iterator();
+ while (kvit->hasNext()) {
+ KeyValue *kv = kvit->next();
+ previousCommit->invalidateKey(kv->getKey());
+ }
+ delete kvit;
+ }
+
+ // if the commit is now dead then remove it
+ if (!previousCommit->isLive()) {
+ commitForClientTable->remove(previousCommit);
+ }
+ }
+ }
- // This client did not see this rejected message yet so add it to the watch set to monitor
- deviceWatchSet.add(lastMessageEntryMachineId);
+ // Update the last seen sequence number from this arbitrator
+ if (lastCommitSeenSequenceNumberByArbitratorTable->get(commit->getMachineId()) != NULL) {
+ if (commit->getSequenceNumber() > lastCommitSeenSequenceNumberByArbitratorTable->get(commit->getMachineId())) {
+ lastCommitSeenSequenceNumberByArbitratorTable->put(commit->getMachineId(), commit->getSequenceNumber());
+ }
+ } else {
+ lastCommitSeenSequenceNumberByArbitratorTable->put(commit->getMachineId(), commit->getSequenceNumber());
}
- }
- if (deviceWatchSet.isEmpty()) {
- // This rejected message has been seen by all the clients so
- entry.setDead();
- } else {
- // We need to watch this rejected message
- entry.setWatchSet(deviceWatchSet);
+ // We processed a new commit that we havent seen before
+ didProcessANewCommit = true;
+
+ // Update the committed table of keys and which commit is using which key
+ {
+ SetIterator<KeyValue *, KeyValue *, uintptr_t, 0, hashKeyValue, equalsKeyValue> *kvit = commit->getKeyValueUpdateSet()->iterator();
+ while (kvit->hasNext()) {
+ KeyValue *kv = kvit->next();
+ committedKeyValueTable->put(kv->getKey(), kv);
+ liveCommitsByKeyTable->put(kv->getKey(), commit);
+ }
+ delete kvit;
+ }
}
}
- /**
- * Check if this abort is live, if not then save it so we can kill it later.
- * update the last transaction number that was arbitrated on.
- */
- void processEntry(Abort entry) {
+ return didProcessANewCommit;
+}
+/**
+ * Create the speculative table from transactions that are still live
+ * and have come from the cloud
+ */
+bool Table::updateSpeculativeTable(bool didProcessNewCommits) {
+ if (liveTransactionBySequenceNumberTable->keySet()->size() == 0) {
+ // There is nothing to speculate on
+ return false;
+ }
- if (entry.getTransactionSequenceNumber() != -1) {
- // update the transaction status if it was sent to the server
- TransactionStatus status = outstandingTransactionStatus.remove(entry.getTransactionSequenceNumber());
- if (status != NULL) {
- status.setStatus(TransactionStatus.StatusAborted);
- }
- }
+ // Create a list of the transaction sequence numbers and sort them
+ // from oldest to newest
+ Vector<int64_t> *transactionSequenceNumbersSorted = new Vector<int64_t>(liveTransactionBySequenceNumberTable->keySet());
+ qsort(transactionSequenceNumberSorted->expose(), transactionSequenceNumbersSorted->size(), sizeof(int64_t), compareInt64);
- // Abort has not been seen by the client it is for yet so we need to keep track of it
- Abort previouslySeenAbort = liveAbortTable.put(entry.getAbortId(), entry);
- if (previouslySeenAbort != NULL) {
- previouslySeenAbort.setDead(); // Delete old version of the abort since we got a rescued newer version
- }
+ bool hasGapInTransactionSequenceNumbers = transactionSequenceNumbersSorted->get(0) != oldestTransactionSequenceNumberSpeculatedOn;
- if (entry.getTransactionArbitrator() == localMachineId) {
- liveAbortsGeneratedByLocal.put(entry.getArbitratorLocalSequenceNumber(), entry);
- }
- if ((entry.getSequenceNumber() != -1) && (lastMessageTable.get(entry.getTransactionMachineId()).getFirst() >= entry.getSequenceNumber())) {
+ if (hasGapInTransactionSequenceNumbers || didProcessNewCommits) {
+ // If there is a gap in the transaction sequence numbers then
+ // there was a commit or an abort of a transaction OR there was a
+ // new commit (Could be from offline commit) so a redo the
+ // speculation from scratch
- // The machine already saw this so it is dead
- entry.setDead();
- liveAbortTable.remove(entry.getAbortId());
+ // Start from scratch
+ speculatedKeyValueTable->clear();
+ lastTransactionSequenceNumberSpeculatedOn = -1;
+ oldestTransactionSequenceNumberSpeculatedOn = -1;
- if (entry.getTransactionArbitrator() == localMachineId) {
- liveAbortsGeneratedByLocal.remove(entry.getArbitratorLocalSequenceNumber());
- }
+ }
- return;
- }
+ // Remember the front of the transaction list
+ oldestTransactionSequenceNumberSpeculatedOn = transactionSequenceNumbersSorted->get(0);
+ // Find where to start arbitration from
+ int startIndex = transactionSequenceNumbersSorted->indexOf(lastTransactionSequenceNumberSpeculatedOn) + 1;
+ if (startIndex >= transactionSequenceNumbersSorted->size()) {
+ // Make sure we are not out of bounds
+ return false; // did not speculate
+ }
+ Hashset<int64_t> *incompleteTransactionArbitrator = new Hashset<int64_t>();
+ bool didSkip = true;
- // Update the last arbitration data that we have seen so far
- if (lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.get(entry.getTransactionArbitrator()) != NULL) {
+ for (int i = startIndex; i < transactionSequenceNumbersSorted->size(); i++) {
+ int64_t transactionSequenceNumber = transactionSequenceNumbersSorted->get(i);
+ Transaction *transaction = liveTransactionBySequenceNumberTable->get(transactionSequenceNumber);
- int64_t lastArbitrationSequenceNumber = lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.get(entry.getTransactionArbitrator());
- if (entry.getSequenceNumber() > lastArbitrationSequenceNumber) {
- // Is larger
- lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.put(entry.getTransactionArbitrator(), entry.getSequenceNumber());
- }
- } else {
- // Never seen any data from this arbitrator so record the first one
- lastArbitrationDataLocalSequenceNumberSeenFromArbitrator.put(entry.getTransactionArbitrator(), entry.getSequenceNumber());
+ if (!transaction->isComplete()) {
+ // If there is an incomplete transaction then there is nothing
+ // we can do add this transactions arbitrator to the list of
+ // arbitrators we should ignore
+ incompleteTransactionArbitrator->add(transaction->getArbitrator());
+ didSkip = true;
+ continue;
}
-
- // Set dead a transaction if we can
- Transaction transactionToSetDead = liveTransactionByTransactionIdTable.remove(new Pair<int64_t, int64_t>(entry.getTransactionMachineId(), entry.getTransactionClientLocalSequenceNumber()));
- if (transactionToSetDead != NULL) {
- liveTransactionBySequenceNumberTable.remove(transactionToSetDead.getSequenceNumber());
+ if (incompleteTransactionArbitrator->contains(transaction->getArbitrator())) {
+ continue;
}
- // Update the last transaction sequence number that the arbitrator arbitrated on
- Long lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable.get(entry.getTransactionArbitrator());
- if ((lastTransactionNumber == NULL) || (lastTransactionNumber < entry.getTransactionSequenceNumber())) {
+ lastTransactionSequenceNumberSpeculatedOn = transactionSequenceNumber;
- // Is a valid one
- if (entry.getTransactionSequenceNumber() != -1) {
- lastArbitratedTransactionNumberByArbitratorTable.put(entry.getTransactionArbitrator(), entry.getTransactionSequenceNumber());
+ if (transaction->evaluateGuard(committedKeyValueTable, speculatedKeyValueTable, NULL)) {
+ // Guard evaluated to true so update the speculative table
+ {
+ SetIterator<KeyValue *, KeyValue *> *kvit = transaction->getKeyValueUpdateSet()->iterator();
+ while (kvit->hasNext()) {
+ KeyValue *kv = kvit->next();
+ speculatedKeyValueTable->put(kv->getKey(), kv);
+ }
+ delete kvit;
}
}
}
- /**
- * Set dead the transaction part if that transaction is dead and keep track of all new parts
- */
- void processEntry(TransactionPart entry) {
- // Check if we have already seen this transaction and set it dead OR if it is not alive
- Long lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable.get(entry.getArbitratorId());
- if ((lastTransactionNumber != NULL) && (lastTransactionNumber >= entry.getSequenceNumber())) {
- // This transaction is dead, it was already committed or aborted
- entry.setDead();
- return;
- }
+ if (didSkip) {
+ // Since there was a skip we need to redo the speculation next time around
+ lastTransactionSequenceNumberSpeculatedOn = -1;
+ oldestTransactionSequenceNumberSpeculatedOn = -1;
+ }
- // This part is still alive
- Hashtable<Pair<Long, Integer>, TransactionPart> transactionPart = newTransactionParts.get(entry.getMachineId());
+ // We did some speculation
+ return true;
+}
- if (transactionPart == NULL) {
- // Dont have a table for this machine Id yet so make one
- transactionPart = new Hashtable<Pair<Long, Integer>, TransactionPart>();
- newTransactionParts.put(entry.getMachineId(), transactionPart);
- }
+/**
+ * Create the pending transaction speculative table from transactions
+ * that are still in the pending transaction buffer
+ */
+void Table::updatePendingTransactionSpeculativeTable(bool didProcessNewCommitsOrSpeculate) {
+ if (pendingTransactionQueue->size() == 0) {
+ // There is nothing to speculate on
+ return;
+ }
- // Update the part and set dead ones we have already seen (got a rescued version)
- TransactionPart previouslySeenPart = transactionPart.put(entry.getPartId(), entry);
- if (previouslySeenPart != NULL) {
- previouslySeenPart.setDead();
- }
+ if (didProcessNewCommitsOrSpeculate || (firstPendingTransaction != pendingTransactionQueue->get(0))) {
+ // need to reset on the pending speculation
+ lastPendingTransactionSpeculatedOn = NULL;
+ firstPendingTransaction = pendingTransactionQueue->get(0);
+ pendingTransactionSpeculatedKeyValueTable->clear();
}
- /**
- * Process new commit entries and save them for future use. Delete duplicates
- */
- void processEntry(CommitPart entry) {
+ // Find where to start arbitration from
+ int startIndex = pendingTransactionQueue->indexOf(firstPendingTransaction) + 1;
+
+ if (startIndex >= pendingTransactionQueue->size()) {
+ // Make sure we are not out of bounds
+ return;
+ }
+ for (int i = startIndex; i < pendingTransactionQueue->size(); i++) {
+ Transaction *transaction = pendingTransactionQueue->get(i);
- // Update the last transaction that was updated if we can
- if (entry.getTransactionSequenceNumber() != -1) {
- Long lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable.get(entry.getMachineId());
+ lastPendingTransactionSpeculatedOn = transaction;
- // Update the last transaction sequence number that the arbitrator arbitrated on
- if ((lastTransactionNumber == NULL) || (lastTransactionNumber < entry.getTransactionSequenceNumber())) {
- lastArbitratedTransactionNumberByArbitratorTable.put(entry.getMachineId(), entry.getTransactionSequenceNumber());
+ if (transaction->evaluateGuard(committedKeyValueTable, speculatedKeyValueTable, pendingTransactionSpeculatedKeyValueTable)) {
+ // Guard evaluated to true so update the speculative table
+ SetIterator<KeyValue *, KeyValue *> *kvit = transaction->getKeyValueUpdateSet()->iterator();
+ while (kvit->hasNext()) {
+ KeyValue *kv = kvit->next();
+ pendingTransactionSpeculatedKeyValueTable->put(kv->getKey(), kv);
}
+ delete kvit;
}
+ }
+}
+/**
+ * Set dead and remove from the live transaction tables the
+ * transactions that are dead
+ */
+void Table::updateLiveTransactionsAndStatus() {
+ // Go through each of the transactions
+ for (Iterator<Map->Entry<int64_t, Transaction> > *iter = liveTransactionBySequenceNumberTable->entrySet()->iterator(); iter->hasNext();) {
+ Transaction *transaction = iter->next()->getValue();
+ // Check if the transaction is dead
+ int64_t lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable->get(transaction->getArbitrator());
+ if ((lastTransactionNumber != NULL) && (lastTransactionNumber >= transaction->getSequenceNumber())) {
- Hashtable<Pair<Long, Integer>, CommitPart> commitPart = newCommitParts.get(entry.getMachineId());
-
- if (commitPart == NULL) {
- // Don't have a table for this machine Id yet so make one
- commitPart = new Hashtable<Pair<Long, Integer>, CommitPart>();
- newCommitParts.put(entry.getMachineId(), commitPart);
- }
+ // Set dead the transaction
+ transaction->setDead();
- // Update the part and set dead ones we have already seen (got a rescued version)
- CommitPart previouslySeenPart = commitPart.put(entry.getPartId(), entry);
- if (previouslySeenPart != NULL) {
- previouslySeenPart.setDead();
+ // Remove the transaction from the live table
+ iter->remove();
+ liveTransactionByTransactionIdTable->remove(transaction->getId());
}
}
- /**
- * Update the last message seen table. Update and set dead the appropriate RejectedMessages as clients see them.
- * Updates the live aborts, removes those that are dead and sets them dead.
- * Check that the last message seen is correct and that there is no mismatch of our own last message or that
- * other clients have not had a rollback on the last message.
- */
- void updateLastMessage(int64_t machineId, int64_t seqNum, Liveness liveness, bool acceptUpdatesToLocal, HashSet<Long> machineSet) {
+ // Go through each of the transactions
+ for (Iterator<Map->Entry<int64_t, TransactionStatus *> > *iter = outstandingTransactionStatus->entrySet()->iterator(); iter->hasNext();) {
+ TransactionStatus *status = iter->next()->getValue();
+
+ // Check if the transaction is dead
+ int64_t lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable->get(status->getTransactionArbitrator());
+ if ((lastTransactionNumber != NULL) && (lastTransactionNumber >= status->getTransactionSequenceNumber())) {
- // We have seen this machine ID
- machineSet.remove(machineId);
+ // Set committed
+ status->setStatus(TransactionStatus_StatusCommitted);
- // Get the set of rejected messages that this machine Id is has not seen yet
- HashSet<RejectedMessage> watchset = rejectedMessageWatchListTable.get(machineId);
+ // Remove
+ iter->remove();
+ }
+ }
+}
- // If there is a rejected message that this machine Id has not seen yet
- if (watchset != NULL) {
+/**
+ * Process this slot, entry by entry-> Also update the latest message sent by slot
+ */
+void Table::processSlot(SlotIndexer *indexer, Slot *slot, bool acceptUpdatesToLocal, Hashset<int64_t> *machineSet) {
+
+ // Update the last message seen
+ updateLastMessage(slot->getMachineID(), slot->getSequenceNumber(), slot, acceptUpdatesToLocal, machineSet);
+
+ // Process each entry in the slot
+ Vector<Entry *> *entries = slot->getEntries();
+ uint eSize = entries->size();
+ for(uint ei=0; ei < eSize; ei++) {
+ Entry * entry = entries->get(ei);
+ switch (entry->getType()) {
+ case TypeCommitPart:
+ processEntry((CommitPart *)entry);
+ break;
+ case TypeAbort:
+ processEntry((Abort *)entry);
+ break;
+ case TypeTransactionPart:
+ processEntry((TransactionPart *)entry);
+ break;
+ case TypeNewKey:
+ processEntry((NewKey *)entry);
+ break;
+ case TypeLastMessage:
+ processEntry((LastMessage *)entry, machineSet);
+ break;
+ case TypeRejectedMessage:
+ processEntry((RejectedMessage *)entry, indexer);
+ break;
+ case TypeTableStatus:
+ processEntry((TableStatus *)entry, slot->getSequenceNumber());
+ break;
+ default:
+ throw new Error("Unrecognized type: ");
+ }
+ }
+}
- // Go through each rejected message that this machine Id has not seen yet
- for (Iterator<RejectedMessage> rmit = watchset.iterator(); rmit.hasNext(); ) {
+/**
+ * Update the last message that was sent for a machine Id
+ */
+void Table::processEntry(LastMessage *entry, Hashset<int64_t> *machineSet) {
+ // Update what the last message received by a machine was
+ updateLastMessage(entry->getMachineID(), entry->getSequenceNumber(), entry, false, machineSet);
+}
- RejectedMessage rm = rmit.next();
+/**
+ * Add the new key to the arbitrators table and update the set of live
+ * new keys (in case of a rescued new key message)
+ */
+void Table::processEntry(NewKey *entry) {
+ // Update the arbitrator table with the new key information
+ arbitratorTable->put(entry->getKey(), entry->getMachineID());
+
+ // Update what the latest live new key is
+ NewKey *oldNewKey = liveNewKeyTable->put(entry->getKey(), entry);
+ if (oldNewKey != NULL) {
+ // Delete the old new key messages
+ oldNewKey->setDead();
+ }
+}
- // If this machine Id has seen this rejected message...
- if (rm.getSequenceNumber() <= seqNum) {
+/**
+ * Process new table status entries and set dead the old ones as new
+ * ones come in-> keeps track of the largest and smallest table status
+ * seen in this current round of updating the local copy of the block
+ * chain
+ */
+void Table::processEntry(TableStatus * entry, int64_t seq) {
+ int newNumSlots = entry->getMaxSlots();
+ updateCurrMaxSize(newNumSlots);
+ initExpectedSize(seq, newNumSlots);
+
+ if (liveTableStatus != NULL) {
+ // We have a larger table status so the old table status is no
+ // int64_ter alive
+ liveTableStatus->setDead();
+ }
- // Remove it from our watchlist
- rmit.remove();
+ // Make this new table status the latest alive table status
+ liveTableStatus = entry;
+}
- // Decrement machines that need to see this notification
- rm.removeWatcher(machineId);
- }
+/**
+ * Check old messages to see if there is a block chain violation->
+ * Also
+ */
+void Table::processEntry(RejectedMessage *entry, SlotIndexer *indexer) {
+ int64_t oldSeqNum = entry->getOldSeqNum();
+ int64_t newSeqNum = entry->getNewSeqNum();
+ bool isequal = entry->getEqual();
+ int64_t machineId = entry->getMachineID();
+ int64_t seq = entry->getSequenceNumber();
+
+ // Check if we have messages that were supposed to be rejected in
+ // our local block chain
+ for (int64_t seqNum = oldSeqNum; seqNum <= newSeqNum; seqNum++) {
+ // Get the slot
+ Slot *slot = indexer->getSlot(seqNum);
+
+ if (slot != NULL) {
+ // If we have this slot make sure that it was not supposed to be
+ // a rejected slot
+ int64_t slotMachineId = slot->getMachineID();
+ if (isequal != (slotMachineId == machineId)) {
+ throw new Error("Server Error: Trying to insert rejected message for slot ");
}
}
+ }
+
+ // Create a list of clients to watch until they see this rejected
+ // message entry->
+ Hashset<int64_t> *deviceWatchSet = new Hashset<int64_t>();
+ for (Map->Entry<int64_t, Pair<int64_t, Liveness *> > *lastMessageEntry : lastMessageTable->entrySet()) {
+ // Machine ID for the last message entry
+ int64_t lastMessageEntryMachineId = lastMessageEntry->getKey();
- // Set dead the abort
- for (Iterator<Map.Entry<Pair<int64_t, int64_t>, Abort>> i = liveAbortTable.entrySet().iterator(); i.hasNext();) {
- Abort abort = i.next().getValue();
+ // We've seen it, don't need to continue to watch-> Our next
+ // message will implicitly acknowledge it->
+ if (lastMessageEntryMachineId == localMachineId) {
+ continue;
+ }
- if ((abort.getTransactionMachineId() == machineId) && (abort.getSequenceNumber() <= seqNum)) {
- abort.setDead();
- i.remove();
+ Pair<int64_t, Liveness *> lastMessageValue = lastMessageEntry->getValue();
+ int64_t entrySequenceNumber = lastMessageValue.getFirst();
- if (abort.getTransactionArbitrator() == localMachineId) {
- liveAbortsGeneratedByLocal.remove(abort.getArbitratorLocalSequenceNumber());
- }
- }
+ if (entrySequenceNumber < seq) {
+ // Add this rejected message to the set of messages that this
+ // machine ID did not see yet
+ addWatchVector(lastMessageEntryMachineId, entry);
+ // This client did not see this rejected message yet so add it
+ // to the watch set to monitor
+ deviceWatchSet->add(lastMessageEntryMachineId);
+ }
+ }
+ if (deviceWatchSet->isEmpty()) {
+ // This rejected message has been seen by all the clients so
+ entry->setDead();
+ } else {
+ // We need to watch this rejected message
+ entry->setWatchSet(deviceWatchSet);
+ }
+}
+
+/**
+ * Check if this abort is live, if not then save it so we can kill it
+ * later-> update the last transaction number that was arbitrated on->
+ */
+void Table::processEntry(Abort *entry) {
+ if (entry->getTransactionSequenceNumber() != -1) {
+ // update the transaction status if it was sent to the server
+ TransactionStatus *status = outstandingTransactionStatus->remove(entry->getTransactionSequenceNumber());
+ if (status != NULL) {
+ status->setStatus(TransactionStatus_StatusAborted);
}
+ }
+
+ // Abort has not been seen by the client it is for yet so we need to
+ // keep track of it
+ Abort *previouslySeenAbort = liveAbortTable->put(entry->getAbortId(), entry);
+ if (previouslySeenAbort != NULL) {
+ previouslySeenAbort->setDead(); // Delete old version of the abort since we got a rescued newer version
+ }
+ if (entry->getTransactionArbitrator() == localMachineId) {
+ liveAbortsGeneratedByLocal->put(entry->getArbitratorLocalSequenceNumber(), entry);
+ }
+ if ((entry->getSequenceNumber() != -1) && (lastMessageTable->get(entry->getTransactionMachineId()).getFirst() >= entry->getSequenceNumber())) {
+ // The machine already saw this so it is dead
+ entry->setDead();
+ liveAbortTable->remove(&entry->getAbortId());
- if (machineId == localMachineId) {
- // Our own messages are immediately dead.
- if (liveness instanceof LastMessage) {
- ((LastMessage)liveness).setDead();
- } else if (liveness instanceof Slot) {
- ((Slot)liveness).setDead();
- } else {
- throw new Error("Unrecognized type");
- }
+ if (entry->getTransactionArbitrator() == localMachineId) {
+ liveAbortsGeneratedByLocal->remove(entry->getArbitratorLocalSequenceNumber());
}
+ return;
+ }
- // Get the old last message for this device
- Pair<Long, Liveness> lastMessageEntry = lastMessageTable.put(machineId, new Pair<Long, Liveness>(seqNum, liveness));
- if (lastMessageEntry == NULL) {
- // If no last message then there is nothing else to process
- return;
+ // Update the last arbitration data that we have seen so far
+ if (lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->contains(entry->getTransactionArbitrator())) {
+ int64_t lastArbitrationSequenceNumber = lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->get(entry->getTransactionArbitrator());
+ if (entry->getSequenceNumber() > lastArbitrationSequenceNumber) {
+ // Is larger
+ lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->put(entry->getTransactionArbitrator(), entry->getSequenceNumber());
}
+ } else {
+ // Never seen any data from this arbitrator so record the first one
+ lastArbitrationDataLocalSequenceNumberSeenFromArbitrator->put(entry->getTransactionArbitrator(), entry->getSequenceNumber());
+ }
- int64_t lastMessageSeqNum = lastMessageEntry.getFirst();
- Liveness lastEntry = lastMessageEntry.getSecond();
+ // Set dead a transaction if we can
+ Transaction *transactionToSetDead = liveTransactionByTransactionIdTable->remove(Pair<int64_t, int64_t>(entry->getTransactionMachineId(), entry->getTransactionClientLocalSequenceNumber()));
+ if (transactionToSetDead != NULL) {
+ liveTransactionBySequenceNumberTable->remove(transactionToSetDead->getSequenceNumber());
+ }
- // If it is not our machine Id since we already set ours to dead
- if (machineId != localMachineId) {
- if (lastEntry instanceof LastMessage) {
- ((LastMessage)lastEntry).setDead();
- } else if (lastEntry instanceof Slot) {
- ((Slot)lastEntry).setDead();
- } else {
- throw new Error("Unrecognized type");
- }
+ // Update the last transaction sequence number that the arbitrator
+ // arbitrated on
+ int64_t lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable->get(entry->getTransactionArbitrator());
+ if ((lastTransactionNumber == NULL) || (lastTransactionNumber < entry->getTransactionSequenceNumber())) {
+ // Is a valid one
+ if (entry->getTransactionSequenceNumber() != -1) {
+ lastArbitratedTransactionNumberByArbitratorTable->put(entry->getTransactionArbitrator(), entry->getTransactionSequenceNumber());
}
+ }
+}
- // Make sure the server is not playing any games
- if (machineId == localMachineId) {
+/**
+ * Set dead the transaction part if that transaction is dead and keep
+ * track of all new parts
+ */
+void Table::processEntry(TransactionPart *entry) {
+ // Check if we have already seen this transaction and set it dead OR
+ // if it is not alive
+ int64_t lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable->get(entry->getArbitratorId());
+ if ((lastTransactionNumber != NULL) && (lastTransactionNumber >= entry->getSequenceNumber())) {
+ // This transaction is dead, it was already committed or aborted
+ entry->setDead();
+ return;
+ }
- if (hadPartialSendToServer) {
- // We were not making any updates and we had a machine mismatch
- if (lastMessageSeqNum > seqNum && !acceptUpdatesToLocal) {
- throw new Error("Server Error: Mismatch on local machine sequence number, needed at least: " + lastMessageSeqNum + " got: " + seqNum);
- }
+ // This part is still alive
+ Hashtable<Pair<int64_t, int32_t> *, TransactionPart *, uintptr_t, 0, pairHashFunction, pairEquals> *transactionPart = newTransactionParts->get(entry->getMachineId());
- } else {
- // We were not making any updates and we had a machine mismatch
- if (lastMessageSeqNum != seqNum && !acceptUpdatesToLocal) {
- throw new Error("Server Error: Mismatch on local machine sequence number, needed: " + lastMessageSeqNum + " got: " + seqNum);
- }
+ if (transactionPart == NULL) {
+ // Dont have a table for this machine Id yet so make one
+ transactionPart = new Hashtable<Pair<int64_t, int32_t> *, TransactionPart *, uintptr_t, 0, pairHashFunction, pairEquals>();
+ newTransactionParts->put(entry->getMachineId(), transactionPart);
+ }
+
+ // Update the part and set dead ones we have already seen (got a
+ // rescued version)
+ TransactionPart *previouslySeenPart = transactionPart->put(entry->getPartId(), entry);
+ if (previouslySeenPart != NULL) {
+ previouslySeenPart->setDead();
+ }
+}
+
+/**
+ * Process new commit entries and save them for future use-> Delete duplicates
+ */
+void Table::processEntry(CommitPart *entry) {
+ // Update the last transaction that was updated if we can
+ if (entry->getTransactionSequenceNumber() != -1) {
+ int64_t lastTransactionNumber = lastArbitratedTransactionNumberByArbitratorTable->get(entry->getMachineId());
+ // Update the last transaction sequence number that the arbitrator
+ // arbitrated on
+ if ((lastTransactionNumber == NULL) || (lastTransactionNumber < entry->getTransactionSequenceNumber())) {
+ lastArbitratedTransactionNumberByArbitratorTable->put(entry->getMachineId(), entry->getTransactionSequenceNumber());
+ }
+ }
+
+ Hashtable<Pair<int64_t, int32_t> *, CommitPart *> *commitPart = newCommitParts->get(entry->getMachineId());
+ if (commitPart == NULL) {
+ // Don't have a table for this machine Id yet so make one
+ commitPart = new Hashtable<Pair<int64_t, int32_t> *, CommitPart *>();
+ newCommitParts->put(entry->getMachineId(), commitPart);
+ }
+ // Update the part and set dead ones we have already seen (got a
+ // rescued version)
+ CommitPart *previouslySeenPart = commitPart->put(entry->getPartId(), entry);
+ if (previouslySeenPart != NULL) {
+ previouslySeenPart->setDead();
+ }
+}
+
+/**
+ * Update the last message seen table-> Update and set dead the
+ * appropriate RejectedMessages as clients see them-> Updates the live
+ * aborts, removes those that are dead and sets them dead-> Check that
+ * the last message seen is correct and that there is no mismatch of
+ * our own last message or that other clients have not had a rollback
+ * on the last message->
+ */
+void Table::updateLastMessage(int64_t machineId, int64_t seqNum, Liveness *liveness, bool acceptUpdatesToLocal, Hashset<int64_t> *machineSet) {
+ // We have seen this machine ID
+ machineSet->remove(machineId);
+
+ // Get the set of rejected messages that this machine Id is has not seen yet
+ Hashset<RejectedMessage *> *watchset = rejectedMessageWatchVectorTable->get(machineId);
+ // If there is a rejected message that this machine Id has not seen yet
+ if (watchset != NULL) {
+ // Go through each rejected message that this machine Id has not
+ // seen yet
+
+ SetIterator<RejectedMessage *, RejectedMessage *> *rmit = watchset->iterator();
+ while(rmit->hasNext()) {
+ RejectedMessage *rm = rmit->next();
+ // If this machine Id has seen this rejected message->->->
+ if (rm->getSequenceNumber() <= seqNum) {
+ // Remove it from our watchlist
+ rmit->remove();
+ // Decrement machines that need to see this notification
+ rm->removeWatcher(machineId);
+ }
+ }
+ delete rmit;
+ }
+
+ // Set dead the abort
+ for (Iterator<Map->Entry<Pair<int64_t, int64_t>, Abort *> > i = liveAbortTable->entrySet()->iterator(); i->hasNext();) {
+ Abort *abort = i->next()->getValue();
+ if ((abort->getTransactionMachineId() == machineId) && (abort->getSequenceNumber() <= seqNum)) {
+ abort->setDead();
+ i->remove();
+ if (abort->getTransactionArbitrator() == localMachineId) {
+ liveAbortsGeneratedByLocal->remove(abort->getArbitratorLocalSequenceNumber());
}
+ }
+ }
+ if (machineId == localMachineId) {
+ // Our own messages are immediately dead->
+ char livenessType = liveness->getType();
+ if (livenessType==TypeLastMessage) {
+ ((LastMessage *)liveness)->setDead();
+ } else if (livenessType == TypeSlot) {
+ ((Slot *)liveness)->setDead();
} else {
- if (lastMessageSeqNum > seqNum) {
- throw new Error("Server Error: Rollback on remote machine sequence number");
- }
+ throw new Error("Unrecognized type");
}
}
+ // Get the old last message for this device
+ Pair<int64_t, Liveness *> * lastMessageEntry = lastMessageTable->put(machineId, new Pair<int64_t, Liveness *>(seqNum, liveness));
+ if (lastMessageEntry == NULL) {
+ // If no last message then there is nothing else to process
+ return;
+ }
- /**
- * Add a rejected message entry to the watch set to keep track of which clients have seen that
- * rejected message entry and which have not.
- */
- void addWatchList(int64_t machineId, RejectedMessage entry) {
- HashSet<RejectedMessage> entries = rejectedMessageWatchListTable.get(machineId);
- if (entries == NULL) {
- // There is no set for this machine ID yet so create one
- entries = new HashSet<RejectedMessage>();
- rejectedMessageWatchListTable.put(machineId, entries);
+ int64_t lastMessageSeqNum = lastMessageEntry->getFirst();
+ Liveness *lastEntry = lastMessageEntry->getSecond();
+ delete lastMessageEntry;
+
+ // If it is not our machine Id since we already set ours to dead
+ if (machineId != localMachineId) {
+ char lastEntryType = lastEntry->getType();
+
+ if (lastEntryType == TypeLastMessage) {
+ ((LastMessage *)lastEntry)->setDead();
+ } else if (lastEntryType == TypeSlot) {
+ ((Slot *)lastEntry)->setDead();
+ } else {
+ throw new Error("Unrecognized type");
}
- entries.add(entry);
}
-
- /**
- * Check if the HMAC chain is not violated
- */
- void checkHMACChain(SlotIndexer indexer, Slot[] newSlots) {
- for (int i = 0; i < newSlots.length; i++) {
- Slot currSlot = newSlots[i];
- Slot prevSlot = indexer.getSlot(currSlot.getSequenceNumber() - 1);
- if (prevSlot != NULL &&
- !Arrays.equals(prevSlot.getHMAC(), currSlot.getPrevHMAC()))
- throw new Error("Server Error: Invalid HMAC Chain" + currSlot + " " + prevSlot);
+ // Make sure the server is not playing any games
+ if (machineId == localMachineId) {
+ if (hadPartialSendToServer) {
+ // We were not making any updates and we had a machine mismatch
+ if (lastMessageSeqNum > seqNum && !acceptUpdatesToLocal) {
+ throw new Error("Server Error: Mismatch on local machine sequence number, needed at least: ");
+ }
+ } else {
+ // We were not making any updates and we had a machine mismatch
+ if (lastMessageSeqNum != seqNum && !acceptUpdatesToLocal) {
+ throw new Error("Server Error: Mismatch on local machine sequence number, needed: ");
+ }
+ }
+ } else {
+ if (lastMessageSeqNum > seqNum) {
+ throw new Error("Server Error: Rollback on remote machine sequence number");
}
}
}
+
+/**
+ * Add a rejected message entry to the watch set to keep track of
+ * which clients have seen that rejected message entry and which have
+ * not.
+ */
+void Table::addWatchVector(int64_t machineId, RejectedMessage *entry) {
+ Hashset<RejectedMessage *> *entries = rejectedMessageWatchVectorTable->get(machineId);
+ if (entries == NULL) {
+ // There is no set for this machine ID yet so create one
+ entries = new Hashset<RejectedMessage *>();
+ rejectedMessageWatchVectorTable->put(machineId, entries);
+ }
+ entries->add(entry);
+}
+
+/**
+ * Check if the HMAC chain is not violated
+ */
+void Table::checkHMACChain(SlotIndexer *indexer, Array<Slot *> *newSlots) {
+ for (int i = 0; i < newSlots->length(); i++) {
+ Slot *currSlot = newSlots->get(i);
+ Slot *prevSlot = indexer->getSlot(currSlot->getSequenceNumber() - 1);
+ if (prevSlot != NULL &&
+ !prevSlot->getHMAC()->equals(currSlot->getPrevHMAC()))
+ throw new Error("Server Error: Invalid HMAC Chain");
+ }
+}