add jcstress tests

Author: andimarek

.github/workflows/master.yml

@@ -28,6 +28,6 @@ jobs:
  - name: Setup Gradle
  uses: gradle/actions/setup-gradle@v5
  - name: build test and publish
- run: ./gradlew assemble && ./gradlew check --info && ./gradlew publishToSonatype closeAndReleaseSonatypeStagingRepository -x check --info --stacktrace
+ run: ./gradlew assemble && ./gradlew check --info && && ./gradlew jcstress && ./gradlew publishToSonatype closeAndReleaseSonatypeStagingRepository -x check --info --stacktrace
  env:
  CI: true

.github/workflows/pull_request.yml

@@ -26,6 +26,6 @@ jobs:
  - name: Setup Gradle
  uses: gradle/actions/setup-gradle@v5
  - name: build and test
- run: ./gradlew assemble && ./gradlew check --info --stacktrace
+ run: ./gradlew assemble && ./gradlew check --info --stacktrace && ./gradlew jcstress
  env:
  CI: true

build.gradle

@@ -1,6 +1,7 @@
+import net.ltgt.gradle.errorprone.CheckSeverity
 import org.jetbrains.kotlin.gradle.dsl.JvmTarget
 import org.jetbrains.kotlin.gradle.dsl.KotlinVersion
-import net.ltgt.gradle.errorprone.CheckSeverity
+
 import java.text.SimpleDateFormat
 
 plugins {
@@ -15,6 +16,7 @@ plugins {
  id 'com.github.ben-manes.versions' version '0.53.0'
  id "me.champeau.jmh" version "0.7.3"
  id "net.ltgt.errorprone" version '4.3.0'
+ id "io.github.reyerizo.gradle.jcstress" version "0.8.15"
 
  // Kotlin just for tests - not production code
  id 'org.jetbrains.kotlin.jvm' version '2.2.20'
@@ -229,7 +231,8 @@ nexusPublishing {
  // https://central.sonatype.org/publish/publish-portal-ossrh-staging-api/#configuration
  nexusUrl.set(uri("https://ossrh-staging-api.central.sonatype.com/service/local/"))
  // GraphQL Java does not publish snapshots, but adding this URL for completeness
- snapshotRepositoryUrl.set(uri("https://central.sonatype.com/repository/maven-snapshots/")) }
+ snapshotRepositoryUrl.set(uri("https://central.sonatype.com/repository/maven-snapshots/"))
+ }
  }
 }
 

src/jcstress/java/org/dataloader/DataLoaderDispatchJCStress.java

@@ -0,0 +1,100 @@
+package org.dataloader;
+
+import org.openjdk.jcstress.annotations.Actor;
+import org.openjdk.jcstress.annotations.Arbiter;
+import org.openjdk.jcstress.annotations.JCStressTest;
+import org.openjdk.jcstress.annotations.Outcome;
+import org.openjdk.jcstress.annotations.State;
+import org.openjdk.jcstress.infra.results.II_Result;
+
+import java.util.List;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.atomic.AtomicInteger;
+
+import static org.openjdk.jcstress.annotations.Expect.ACCEPTABLE;
+
+@JCStressTest
+@State
+@Outcome(id = "2000, 2000", expect = ACCEPTABLE, desc = "accepted")
+public class DataLoaderDispatchJCStress {
+
+
+ AtomicInteger counter = new AtomicInteger();
+ AtomicInteger batchLoaderCount = new AtomicInteger();
+ volatile boolean finished1;
+ volatile boolean finished2;
+
+
+ BatchLoader<String, String> batchLoader = keys -> {
+ return CompletableFuture.supplyAsync(() -> {
+ batchLoaderCount.getAndAdd(keys.size());
+ return keys;
+ });
+ };
+ DataLoader<String, String> dataLoader = DataLoaderFactory.newDataLoader(batchLoader);
+
+ public DataLoaderDispatchJCStress() {
+
+ }
+
+ @Actor
+ public void load1() {
+ for (int i = 0; i < 1000; i++) {
+ dataLoader.load("load-1-" + i);
+ }
+ finished1 = true;
+ }
+
+ @Actor
+ public void load2() {
+ for (int i = 0; i < 1000; i++) {
+ dataLoader.load("load-2-" + i);
+ }
+ finished2 = true;
+ }
+
+
+ @Actor
+ public void dispatch1() {
+ while (!finished1 || !finished2) {
+ try {
+ List<String> dispatchedResult = dataLoader.dispatch().get();
+ counter.getAndAdd(dispatchedResult.size());
+ } catch (Exception e) {
+ throw new RuntimeException(e);
+ }
+ }
+ try {
+ List<String> dispatchedResult = dataLoader.dispatch().get();
+ counter.getAndAdd(dispatchedResult.size());
+ } catch (Exception e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ @Actor
+ public void dispatch2() {
+ while (!finished1 || !finished2) {
+ try {
+ List<String> dispatchedResult = dataLoader.dispatch().get();
+ counter.getAndAdd(dispatchedResult.size());
+ } catch (Exception e) {
+ throw new RuntimeException(e);
+ }
+ }
+ try {
+ List<String> dispatchedResult = dataLoader.dispatch().get();
+ counter.getAndAdd(dispatchedResult.size());
+ } catch (Exception e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ @Arbiter
+ public void arbiter(II_Result r) {
+ r.r1 = counter.get();
+ r.r2 = batchLoaderCount.get();
+ }
+
+
+}