thomasdarimont · January 4, 2016 02:49 · thomasdarimont · Jan 22, 2014 · thomasdarimont · Jan 22, 2014
diff --git a/QueryMapperUnitTests.java b/QueryMapperUnitTests.java
 /**
 	 * @see DATAMONGO-686
 	 */
 	@SuppressWarnings("serial")
 	@Test
 	public void thereShouldBeNoRaceConditionWhenReusingQueryFromMultipleThreads() throws Exception {

 		Query query = new Query().addCriteria(where("_id").in("42"));

 		final CountDownLatch latch = new CountDownLatch(1);
 		final List<Object> observedValues = new ArrayList<Object>();

 		final DBObject queryObject = query.getQueryObject();
 		queryObject.put("_id", new BasicDBObject(((DBObject) queryObject.get("_id")).toMap()) {
 			@Override
 			public Object get(String key) {

 				if (key.endsWith("$in") && Thread.currentThread().getName().equals("t2")) {
 					// provokes the race-condition in
 					// org.springframework.data.mongodb.core.convert.QueryMapper.getMappedValue(Field, Object)
 					try {
 						latch.await();
 					} catch (InterruptedException e) {
 						e.printStackTrace();
 					}
 				}

 				Object result = super.get(key);
 				observedValues.add(result);
 				return result;
 			}
 		});

 		final List<Exception> exceptions = new ArrayList<Exception>();

 		Runnable code = new Runnable() {

 			@Override
 			public void run() {
 				try {
 					mapper.getMappedObject(queryObject, context.getPersistentEntity(Sample.class));
 				} catch (Exception ex) {
 					exceptions.add(ex);
 				}
 				latch.countDown();
 			}
 		};

 		Thread t1 = new Thread(code, "t1");
 		Thread t2 = new Thread(code, "t2");

 		t1.start();
 		t2.start();

 		t1.join();
 		t2.join();

 		assertThat(observedValues, is(not(empty())));
 		assertThat(observedValues.size(), is(2));
 		assertThat("Both observed values should be of the same type! But was: "
 				+ observedValues.get(0).getClass().getName() + " and " + observedValues.get(1).getClass().getName(),
 				observedValues.get(0).getClass().equals(observedValues.get(1).getClass()), is(true));
 		assertThat(exceptions.toString(), exceptions, is(empty()));
 	}
	/**
	* @see DATAMONGO-686
	*/
	@SuppressWarnings("serial")
	@Test
	public void thereShouldBeNoRaceConditionWhenReusingQueryFromMultipleThreads() throws Exception {

	Query query = new Query().addCriteria(where("_id").in("42"));

	final CountDownLatch latch = new CountDownLatch(1);
	final List<Object> observedValues = new ArrayList<Object>();

	final DBObject queryObject = query.getQueryObject();
	queryObject.put("_id", new BasicDBObject(((DBObject) queryObject.get("_id")).toMap()) {
	@Override
	public Object get(String key) {

	if (key.endsWith("$in") && Thread.currentThread().getName().equals("t2")) {
	// provokes the race-condition in
	// org.springframework.data.mongodb.core.convert.QueryMapper.getMappedValue(Field, Object)
	try {
	latch.await();
	} catch (InterruptedException e) {
	e.printStackTrace();
	}
	}

	Object result = super.get(key);
	observedValues.add(result);
	return result;
	}
	});

	final List<Exception> exceptions = new ArrayList<Exception>();

	Runnable code = new Runnable() {

	@Override
	public void run() {
	try {
	mapper.getMappedObject(queryObject, context.getPersistentEntity(Sample.class));
	} catch (Exception ex) {
	exceptions.add(ex);
	}
	latch.countDown();
	}
	};

	Thread t1 = new Thread(code, "t1");
	Thread t2 = new Thread(code, "t2");

	t1.start();
	t2.start();

	t1.join();
	t2.join();

	assertThat(observedValues, is(not(empty())));
	assertThat(observedValues.size(), is(2));
	assertThat("Both observed values should be of the same type! But was: "
	+ observedValues.get(0).getClass().getName() + " and " + observedValues.get(1).getClass().getName(),
	observedValues.get(0).getClass().equals(observedValues.get(1).getClass()), is(true));
	assertThat(exceptions.toString(), exceptions, is(empty()));
	}
No results found