Move responsibility for storing method context to dedicated component.

Author: christophstrobl

src/main/java/org/springframework/data/repository/core/RepositoryMethodContext.java

@@ -20,9 +20,9 @@
 /**
  * Interface containing methods and value objects to obtain information about the current repository method invocation.
  * <p>
- * The {@link #getMetadata()} method is usable if the repository factory is configured to expose the current
- * repository method metadata (not the default). It returns the invoked repository method. Target objects or advice can
- * use this to make advised calls.
+ * The {@link #getMetadata()} method is usable if the repository factory is configured to expose the current repository
+ * method metadata (not the default). It returns the invoked repository method. Target objects or advice can use this to
+ * make advised calls.
  * <p>
  * Spring Data's framework does not expose method metadata by default, as there is a performance cost in doing so.
  * <p>
@@ -50,7 +50,7 @@ public interface RepositoryMethodContext {
 	 * The method object represents the method as being invoked on the repository interface. It doesn't match the backing
 	 * repository implementation in case the method invocation is delegated to an implementation method.
 	 *
-	 * @return the current method, will never be {@literal null}..
+	 * @return the current method, will never be {@literal null}.
 	 */
 	Method getMethod();
 }

src/main/java/org/springframework/data/repository/core/RepositoryMethodContextHolder.java

@@ -0,0 +1,92 @@
+/*
+ * Copyright 2024 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.repository.core;
+
+import org.springframework.core.NamedThreadLocal;
+import org.springframework.lang.Nullable;
+
+/**
+ * @author Christoph Strobl
+ * @since 3.4.0
+ */
+public class RepositoryMethodContextHolder {
+
+	private static ContextProvider contextSupplier;
+
+	static {
+		contextSupplier = new ThreadLocalContextProvider();
+	}
+
+	@Nullable
+	public static RepositoryMethodContext setContext(@Nullable RepositoryMethodContext context) {
+		return contextSupplier.set(context);
+	}
+
+	@Nullable
+	public static RepositoryMethodContext current() {
+		return contextSupplier.get();
+	}
+
+	public static void clearContext() {
+		contextSupplier.clear();
+	}
+
+	interface ContextProvider {
+
+		@Nullable
+		RepositoryMethodContext get();
+
+		@Nullable
+		RepositoryMethodContext set(@Nullable RepositoryMethodContext context);
+
+		void clear();
+	}
+
+	static class ThreadLocalContextProvider implements ContextProvider {
+
+		/**
+		 * ThreadLocal holder for repository method associated with this thread. Will contain {@code null} unless the
+		 * "exposeMetadata" property on the controlling repository factory configuration has been set to "true".
+		 */
+		private static final ThreadLocal<RepositoryMethodContext> currentMethod = new NamedThreadLocal<>(
+				"Current Repository Method");
+
+		@Override
+		@Nullable
+		public RepositoryMethodContext get() {
+			return currentMethod.get();
+		}
+
+		public void clear() {
+			currentMethod.remove();
+		}
+
+		@Override
+		@Nullable
+		public RepositoryMethodContext set(@Nullable RepositoryMethodContext context) {
+
+			RepositoryMethodContext old = currentMethod.get();
+
+			if (context != null) {
+				currentMethod.set(context);
+			} else {
+				currentMethod.remove();
+			}
+
+			return old;
+		}
+	}
+}

src/main/java/org/springframework/data/repository/core/support/DefaultRepositoryMethodContext.java

@@ -17,11 +17,8 @@
 
 import java.lang.reflect.Method;
 
-import org.springframework.aop.framework.ProxyFactory;
-import org.springframework.core.NamedThreadLocal;
 import org.springframework.data.repository.core.RepositoryMetadata;
 import org.springframework.data.repository.core.RepositoryMethodContext;
-import org.springframework.lang.Nullable;
 import org.springframework.util.Assert;
 
 /**
@@ -34,13 +31,6 @@
  */
 public class DefaultRepositoryMethodContext implements RepositoryMethodContext {
 
-	/**
-	 * ThreadLocal holder for repository method associated with this thread. Will contain {@code null} unless the
-	 * "exposeMetadata" property on the controlling repository factory configuration has been set to "true".
-	 */
-	private static final ThreadLocal<RepositoryMethodContext> currentMethod = new NamedThreadLocal<>(
-			"Current Repository Method");
-
 	private final RepositoryMetadata repositoryMetadata;
 	private final Method method;
 
@@ -64,36 +54,6 @@ public static RepositoryMethodContext forMethod(Method method) {
 				method);
 	}
 
-	/**
-	 * Creates a proxy {@link RepositoryMethodContext} instance suitable for dependency injection.
-	 *
-	 * @return will never be {@literal null}.
-	 */
-	public static RepositoryMethodContext getInjectionProxy() {
-
-		return ProxyFactory.getProxy(RepositoryMethodContext.class,
-				new DynamicLookupTargetSource<>(RepositoryMethodContext.class, () -> getInstance()));
-	}
-
-	@Nullable
-	static RepositoryMethodContext getInstance() {
-		return currentMethod.get();
-	}
-
-	@Nullable
-	static RepositoryMethodContext setMetadata(@Nullable RepositoryMethodContext metadata) {
-
-		RepositoryMethodContext old = currentMethod.get();
-
-		if (metadata != null) {
-			currentMethod.set(metadata);
-		} else {
-			currentMethod.remove();
-		}
-
-		return old;
-	}
-
 	@Override
 	public RepositoryMetadata getMetadata() {
 		return repositoryMetadata;

src/main/java/org/springframework/data/repository/core/support/RepositoryFactorySupport.java

@@ -57,6 +57,7 @@
 import org.springframework.data.repository.core.RepositoryInformation;
 import org.springframework.data.repository.core.RepositoryMetadata;
 import org.springframework.data.repository.core.RepositoryMethodContext;
+import org.springframework.data.repository.core.RepositoryMethodContextHolder;
 import org.springframework.data.repository.core.support.RepositoryComposition.RepositoryFragments;
 import org.springframework.data.repository.core.support.RepositoryInvocationMulticaster.DefaultRepositoryInvocationMulticaster;
 import org.springframework.data.repository.core.support.RepositoryInvocationMulticaster.NoOpRepositoryInvocationMulticaster;
@@ -776,13 +777,13 @@ public Object invoke(MethodInvocation invocation) throws Throwable {
 
 			try {
 
-				oldMetadata = DefaultRepositoryMethodContext
-						.setMetadata(new DefaultRepositoryMethodContext(repositoryMetadata, invocation.getMethod()));
+				oldMetadata = RepositoryMethodContextHolder
+						.setContext(new DefaultRepositoryMethodContext(repositoryMetadata, invocation.getMethod()));
 
 				return invocation.proceed();
 
 			} finally {
-				DefaultRepositoryMethodContext.setMetadata(oldMetadata);
+				RepositoryMethodContextHolder.setContext(oldMetadata);
 			}
 		}
 	}

src/test/java/org/springframework/data/repository/core/support/RepositoryFactorySupportUnitTests.java

@@ -58,6 +58,7 @@
 import org.springframework.data.repository.core.RepositoryInformation;
 import org.springframework.data.repository.core.RepositoryMetadata;
 import org.springframework.data.repository.core.RepositoryMethodContext;
+import org.springframework.data.repository.core.RepositoryMethodContextHolder;
 import org.springframework.data.repository.core.support.DummyRepositoryFactory.MyRepositoryQuery;
 import org.springframework.data.repository.core.support.RepositoryComposition.RepositoryFragments;
 import org.springframework.data.repository.core.support.RepositoryMethodInvocationListener.RepositoryMethodInvocation;
@@ -255,7 +256,7 @@ void capturesRepositoryMetadata() {
 		record Metadata(RepositoryMethodContext context, MethodInvocation methodInvocation) {}
 
 		when(factory.queryOne.execute(any(Object[].class)))
-				.then(invocation -> new Metadata(DefaultRepositoryMethodContext.getInstance(),
+				.then(invocation -> new Metadata(RepositoryMethodContextHolder.current(),
 						ExposeInvocationInterceptor.currentInvocation()));
 
 		factory.setExposeMetadata(true);
@@ -278,7 +279,7 @@ record Metadata(RepositoryMethodContext context, MethodInvocation methodInvocati
 		}
 
 		when(factory.queryOne.execute(any(Object[].class)))
-				.then(invocation -> new Metadata(DefaultRepositoryMethodContext.getInstance(),
+				.then(invocation -> new Metadata(RepositoryMethodContextHolder.current(),
 						ExposeInvocationInterceptor.currentInvocation()));
 
 		var repository = factory.getRepository(ObjectRepository.class, new RepositoryMetadataAccess() {});