Fix reference to new CleanedTransactionReport msg

dgomezferro · Nov 23, 2012 · 12d2337 · 12d2337
1 parent 062702a
commit 12d2337
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/com/yahoo/omid/client/TSOClient.java b/src/main/java/com/yahoo/omid/client/TSOClient.java
@@ -56,6 +56,7 @@
 import com.yahoo.omid.tso.TSOMessage;
 import com.yahoo.omid.tso.messages.AbortRequest;
 import com.yahoo.omid.tso.messages.AbortedTransactionReport;
+import com.yahoo.omid.tso.messages.CleanedTransactionReport;
 import com.yahoo.omid.tso.messages.CommitQueryRequest;
 import com.yahoo.omid.tso.messages.CommitQueryResponse;
 import com.yahoo.omid.tso.messages.CommitRequest;
@@ -547,8 +548,8 @@ public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) {
       } else if (msg instanceof CommittedTransactionReport) {
          CommittedTransactionReport ctr = (CommittedTransactionReport) msg;
          committed.commit(ctr.startTimestamp, ctr.commitTimestamp);
-      } else if (msg instanceof FullAbortRequest) {
-         FullAbortRequest r = (FullAbortRequest) msg;
+      } else if (msg instanceof CleanedTransactionReport) {
+          CleanedTransactionReport r = (CleanedTransactionReport) msg;
          aborted.remove(r.startTimestamp);
       } else if (msg instanceof AbortedTransactionReport) {
          AbortedTransactionReport r = (AbortedTransactionReport) msg;