Spring/spring-data-couchbase
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

DATACOUCH-484 - test to demonstrate that parameters are thread-safe

Browse Source
This commit is contained in:
mikereiche
2020-06-15 13:00:25 -07:00
parent f4eee34faf
commit c3fbf38ba2
2 changed files with 90 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/test/java/org/springframework/data/couchbase/domain/AirportRepository.java
View File

@@ -18,6 +18,7 @@ package org.springframework.data.couchbase.domain;
import java.util.List;
import org.springframework.data.couchbase.repository.Query;
import org.springframework.data.couchbase.repository.ScanConsistency;
import org.springframework.data.repository.PagingAndSortingRepository;
import org.springframework.stereotype.Repository;
@@ -33,4 +34,7 @@ public interface AirportRepository extends PagingAndSortingRepository<Airport, S
List<Airport> findAllByIata(String iata);
@Query("#{#n1ql.selectEntity} where iata = $1")
List<Airport> getAllByIata(String iata);
}

86
src/test/java/org/springframework/data/couchbase/repository/CouchbaseRepositoryQueryIntegrationTests.java
View File

@@ -19,6 +19,10 @@ package org.springframework.data.couchbase.repository;
import static org.junit.jupiter.api.Assertions.*;
import java.util.List;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.stream.Collectors;
import java.util.stream.StreamSupport;
@@ -75,6 +79,88 @@ public class CouchbaseRepositoryQueryIntegrationTests extends ClusterAwareIntegr
System.err.println(airports);
}
@Test
void threadSafeParametersTest() {
String[] iatas = { "JFK", "IAD", "SFO", "SJC", "SEA", "LAX", "PHX" };
Future[] future = new Future[iatas.length];
ExecutorService executorService = Executors.newFixedThreadPool(iatas.length);
try {
Callable<Boolean>[] suppliers = new Callable[iatas.length];
for (int i = 0; i < iatas.length; i++) {
Airport airport = new Airport("airports::" + iatas[i], iatas[i] /*iata*/, iatas[i] /* lcao */);
airportRepository.save(airport);
final int idx = i;
suppliers[i] = () -> {
System.out.println(Thread.currentThread() + " " + iatas[idx] + " ->");
try {
Thread.sleep(iatas.length - idx); // so they are executed out-of-order
} catch (InterruptedException ie) {
;
}
String foundName = airportRepository.findAllByIata(iatas[idx]).get(0).getIata();
System.out.println(Thread.currentThread() + " " + iatas[idx] + " <- ");
assertEquals(iatas[idx], foundName);
return iatas[idx].equals(foundName);
};
}
for (int i = 0; i < iatas.length; i++) {
future[i] = executorService.submit(suppliers[i]);
}
for (int i = 0; i < iatas.length; i++) {
future[i].get();
}
} catch (Exception e) {
throw new RuntimeException(e);
} finally {
for (int i = 0; i < iatas.length; i++) {
Airport airport = new Airport("airports::" + iatas[i], iatas[i] /*iata*/, iatas[i] /* lcao */);
airportRepository.delete(airport);
}
}
}
@Test
void threadSafeStringParametersTest() {
String[] iatas = { "JFK", "IAD", "SFO", "SJC", "SEA", "LAX", "PHX" };
Future[] future = new Future[iatas.length];
ExecutorService executorService = Executors.newFixedThreadPool(iatas.length);
try {
Callable<Boolean>[] suppliers = new Callable[iatas.length];
for (int i = 0; i < iatas.length; i++) {
Airport airport = new Airport("airports::" + iatas[i], iatas[i] /*iata*/, iatas[i] /* lcao */);
airportRepository.save(airport);
final int idx = i;
suppliers[i] = () -> {
System.out.println(Thread.currentThread() + " " + iatas[idx] + " ->");
try {
Thread.sleep(iatas.length - idx); // so they are executed out-of-order
} catch (InterruptedException ie) {
;
}
String foundName = airportRepository.getAllByIata(iatas[idx]).get(0).getIata();
System.out.println(Thread.currentThread() + " " + iatas[idx] + " <- ");
assertEquals(iatas[idx], foundName);
return iatas[idx].equals(foundName);
};
}
for (int i = 0; i < iatas.length; i++) {
future[i] = executorService.submit(suppliers[i]);
}
for (int i = 0; i < iatas.length; i++) {
future[i].get();
}
} catch (Exception e) {
throw new RuntimeException(e);
} finally {
for (int i = 0; i < iatas.length; i++) {
Airport airport = new Airport("airports::" + iatas[i], iatas[i] /*iata*/, iatas[i] /* lcao */);
airportRepository.delete(airport);
}
}
}
@Configuration
@EnableCouchbaseRepositories("org.springframework.data.couchbase")
static class Config extends AbstractCouchbaseConfiguration {