Use a sliding window to track acknowledged values.

core/src/main/java/com/yahoo/ycsb/generator/AcknowledgedCounterGenerator.java

@@ -1,14 +1,17 @@
 package com.yahoo.ycsb.generator;
 
-import java.util.PriorityQueue;
+import java.util.concurrent.locks.ReentrantLock;
 
 /**
  * A CounterGenerator that reports generated integers via lastInt()
  * only after they have been acknowledged.
  */
 public class AcknowledgedCounterGenerator extends CounterGenerator
 {
-	private PriorityQueue<Integer> ack;
+	private static final int WINDOW_SIZE = 10000;
+
+	private ReentrantLock lock;
+	private boolean[] window;
 	private int limit;
 
 	/**
@@ -17,7 +20,8 @@ public class AcknowledgedCounterGenerator extends CounterGenerator
 	public AcknowledgedCounterGenerator(int countstart)
 	{
 		super(countstart);
-		ack = new PriorityQueue<Integer>();
+		lock = new ReentrantLock();
+		window = new boolean[WINDOW_SIZE];
 		limit = countstart - 1;
 	}
 
@@ -34,17 +38,35 @@ public int lastInt()
 	/**
 	 * Make a generated counter value available via lastInt().
 	 */
-	public synchronized void acknowledge(int value)
+	public void acknowledge(int value)
 	{
-		ack.add(value);
+		if (value > limit + WINDOW_SIZE) {
+			throw new RuntimeException("This should be a different exception.");
+		}
+
+		window[value % WINDOW_SIZE] = true;
+
+		if (lock.tryLock()) {
+			// move a contiguous sequence from the window
+			// over to the "limit" variable
+
+			try {
+				int index;
+
+				for (index = limit + 1; index <= value; ++index) {
+					int slot = index % WINDOW_SIZE;
 
-		// move a contiguous sequence from the priority queue
-		// over to the "limit" variable
+					if (!window[slot]) {
+						break;
+					}
 
-		Integer min;
+					window[slot] = false;
+				}
 
-		while ((min = ack.peek()) != null && min == limit + 1) {
-			limit = ack.poll();
+				limit = index - 1;
+			} finally {
+				lock.unlock();
+			}
 		}
 	}
 }