org.openqa.selenium.server.BrowserResponseSequencer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of selenium-server

The newest version!

package org.openqa.selenium.server;

import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

import org.apache.commons.logging.Log;
import org.mortbay.log.LogFactory;

public class BrowserResponseSequencer {
	static Log log = LogFactory.getLog(BrowserResponseSequencer.class);
	int num = 0;
	final Lock lock;
	final Condition numIncreased;
	final String uniqueId;
	public BrowserResponseSequencer(String uniqueId) {
		this.uniqueId = uniqueId;
		this.lock = new ReentrantLock();
		numIncreased = lock.newCondition();
	}
	
	public void increaseNum() {
		lock.lock();
		try {
			num++;
			numIncreased.signalAll();
		} finally {
			lock.unlock();
		}
		
	}
	
	public void waitUntilNumIsAtLeast(int expected) {
		lock.lock();
		try {
			while(true) {
				if (num >= expected) return;
				log.debug("Waiting "+uniqueId+", expected sequence number " + expected + ", was " + num + ".");
				boolean timedOut = false;
				try {
					timedOut = !numIncreased.await(5, TimeUnit.SECONDS);
				} catch (InterruptedException e) {
					log.debug("interrupted", e);
				}
				if (timedOut) {
					log.warn(uniqueId + " expected sequence number " + expected + ", was " + num + ".  Continuing anyway");
					num++;
					numIncreased.signalAll();
				}
			}
		} finally {
			lock.unlock();
		}
	}
	
	@Override
	public String toString() {
		return uniqueId + ": " + num;
	}
}