initial commit

Author: hmottestad

core/common/iterator/src/main/java/org/eclipse/rdf4j/common/iteration/DistinctIteration.java

@@ -8,9 +8,11 @@
  *
  * SPDX-License-Identifier: BSD-3-Clause
  *******************************************************************************/
-
+// Some portions generated by Codex
 package org.eclipse.rdf4j.common.iteration;
 
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
 import java.util.HashSet;
 import java.util.Set;
 import java.util.function.Supplier;
@@ -20,6 +22,8 @@
  */
 public class DistinctIteration<E> extends FilterIteration<E> {
 
+	private static final String OPERATOR_NAME = "DISTINCT";
+
 	/*-----------*
 	 * Variables *
 	 *-----------*/
@@ -76,6 +80,8 @@ public DistinctIteration(CloseableIteration<? extends E> iter, Supplier<Set<E>>
 	 */
 	@Override
 	protected boolean accept(E object) {
+		QueryExecutionContextBridge.markHeavy(OPERATOR_NAME);
+		QueryExecutionContextBridge.checkpoint(OPERATOR_NAME);
 		if (inExcludeSet(object)) {
 			// object has already been returned
 			return false;
@@ -104,4 +110,73 @@ private boolean inExcludeSet(E object) {
 	protected boolean add(E object) {
 		return excludeSet.add(object);
 	}
+
+	private static final class QueryExecutionContextBridge {
+
+		private static final String QUERY_EXECUTION_CONTEXT_CLASS = "org.eclipse.rdf4j.http.client.QueryExecutionContext";
+
+		private static volatile boolean initialized;
+		private static volatile Method markHeavyMethod;
+		private static volatile Method checkpointMethod;
+
+		private QueryExecutionContextBridge() {
+		}
+
+		private static void markHeavy(String operator) {
+			Method method = getMethod(true);
+			if (method != null) {
+				invoke(method, operator);
+			}
+		}
+
+		private static void checkpoint(String operator) {
+			Method method = getMethod(false);
+			if (method != null) {
+				invoke(method, operator);
+			}
+		}
+
+		private static Method getMethod(boolean markHeavy) {
+			if (!initialized) {
+				initialize();
+			}
+			return markHeavy ? markHeavyMethod : checkpointMethod;
+		}
+
+		private static synchronized void initialize() {
+			if (initialized) {
+				return;
+			}
+			try {
+				Class<?> contextType = Class.forName(QUERY_EXECUTION_CONTEXT_CLASS);
+				markHeavyMethod = contextType.getMethod("markHeavy", String.class);
+				checkpointMethod = contextType.getMethod("checkpoint", String.class);
+			} catch (ClassNotFoundException | NoSuchMethodException e) {
+				markHeavyMethod = null;
+				checkpointMethod = null;
+			}
+			initialized = true;
+		}
+
+		private static void invoke(Method method, String operator) {
+			try {
+				method.invoke(null, operator);
+			} catch (IllegalAccessException e) {
+				throw new IllegalStateException("Unable to access query execution context bridge", e);
+			} catch (InvocationTargetException e) {
+				throw propagate(e.getCause());
+			}
+		}
+
+		private static RuntimeException propagate(Throwable throwable) {
+			if (throwable instanceof RuntimeException) {
+				throw (RuntimeException) throwable;
+			}
+			if (throwable instanceof Error) {
+				throw (Error) throwable;
+			}
+			throw new IllegalStateException("Unexpected checked exception from query execution context bridge",
+					throwable);
+		}
+	}
 }

core/common/iterator/src/test/java/org/eclipse/rdf4j/http/client/QueryExecutionContext.java

@@ -0,0 +1,54 @@
+/*******************************************************************************
+ * Copyright (c) 2026 Eclipse RDF4J contributors.
+ *
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Distribution License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/org/documents/edl-v10.php.
+ *
+ * SPDX-License-Identifier: BSD-3-Clause
+ *******************************************************************************/
+// Some portions generated by Codex
+package org.eclipse.rdf4j.http.client;
+
+import java.util.concurrent.atomic.AtomicInteger;
+
+public final class QueryExecutionContext {
+
+	private static final AtomicInteger MARK_HEAVY_CALLS = new AtomicInteger();
+	private static final AtomicInteger CHECKPOINT_CALLS = new AtomicInteger();
+
+	private static volatile RuntimeException checkpointFailure;
+
+	private QueryExecutionContext() {
+	}
+
+	public static void markHeavy(String operator) {
+		MARK_HEAVY_CALLS.incrementAndGet();
+	}
+
+	public static void checkpoint(String operator) {
+		CHECKPOINT_CALLS.incrementAndGet();
+		if (checkpointFailure != null) {
+			throw checkpointFailure;
+		}
+	}
+
+	public static void failOnCheckpoint(RuntimeException runtimeException) {
+		checkpointFailure = runtimeException;
+	}
+
+	public static int getMarkHeavyCalls() {
+		return MARK_HEAVY_CALLS.get();
+	}
+
+	public static int getCheckpointCalls() {
+		return CHECKPOINT_CALLS.get();
+	}
+
+	public static void reset() {
+		MARK_HEAVY_CALLS.set(0);
+		CHECKPOINT_CALLS.set(0);
+		checkpointFailure = null;
+	}
+}