Replace Function<X, X> with UnaryOperator<X> in unstable modules

Co-authored-by: trask <[email protected]>

Author: Copilot

instrumentation/cassandra/cassandra-4.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/cassandra/v4_0/CompletionStageFunction.java

@@ -6,9 +6,9 @@
 package io.opentelemetry.javaagent.instrumentation.cassandra.v4_0;
 
 import com.datastax.oss.driver.api.core.CqlSession;
-import java.util.function.Function;
+import java.util.function.UnaryOperator;
 
-public class CompletionStageFunction implements Function<Object, Object> {
+public class CompletionStageFunction implements UnaryOperator<Object> {
 
   @Override
   public Object apply(Object session) {

instrumentation/cassandra/cassandra-4.4/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/cassandra/v4_4/CompletionStageFunction.java

@@ -6,9 +6,9 @@
 package io.opentelemetry.javaagent.instrumentation.cassandra.v4_4;
 
 import com.datastax.oss.driver.api.core.CqlSession;
-import java.util.function.Function;
+import java.util.function.UnaryOperator;
 
-public class CompletionStageFunction implements Function<Object, Object> {
+public class CompletionStageFunction implements UnaryOperator<Object> {
 
   @Override
   public Object apply(Object session) {

instrumentation/elasticsearch/elasticsearch-rest-7.0/library/src/main/java/io/opentelemetry/instrumentation/elasticsearch/rest/v7_0/RestClientWrapper.java

@@ -14,8 +14,8 @@
 import java.lang.reflect.Constructor;
 import java.lang.reflect.Field;
 import java.util.List;
-import java.util.function.Function;
 import java.util.function.Supplier;
+import java.util.function.UnaryOperator;
 import net.bytebuddy.ByteBuddy;
 import net.bytebuddy.description.modifier.Visibility;
 import net.bytebuddy.dynamic.loading.ClassLoadingStrategy;
@@ -32,7 +32,7 @@ class RestClientWrapper {
   private static final Class<?> proxyClass = createProxyClass();
   private static final Field targetField = getTargetField(proxyClass);
   private static final Field instrumenterSupplierField = getInstrumenterSupplierField(proxyClass);
-  private static final Function<RestClient, RestClient> proxyFactory = getProxyFactory(proxyClass);
+  private static final UnaryOperator<RestClient> proxyFactory = getProxyFactory(proxyClass);
 
   private static Class<?> createProxyClass() {
     return new ByteBuddy()
@@ -139,7 +139,7 @@ private static Instrumenter<ElasticsearchRestRequest, Response> getInstrumenter(
     return supplier != null ? supplier.get() : null;
   }
 
-  private static Function<RestClient, RestClient> getProxyFactory(Class<?> clazz) {
+  private static UnaryOperator<RestClient> getProxyFactory(Class<?> clazz) {
     for (Constructor<?> constructor : clazz.getDeclaredConstructors()) {
       Class<?>[] parameterTypes = constructor.getParameterTypes();
       if (parameterTypes.length >= 3

instrumentation/jmx-metrics/library/src/main/java/io/opentelemetry/instrumentation/jmx/engine/UnitConverter.java

@@ -8,7 +8,7 @@
 import java.util.HashMap;
 import java.util.Map;
 import java.util.concurrent.TimeUnit;
-import java.util.function.Function;
+import java.util.function.UnaryOperator;
 import javax.annotation.Nullable;
 
 /**
@@ -24,7 +24,7 @@ class UnitConverter {
     registerConversion("ns", "s", value -> value.doubleValue() / TimeUnit.SECONDS.toNanos(1));
   }
 
-  private final Function<Number, Number> convertingFunction;
+  private final UnaryOperator<Number> convertingFunction;
 
   /**
    * Get an instance of converter that is able to convert a value from a given source to a target
@@ -66,7 +66,7 @@ public static UnitConverter getInstance(@Nullable String sourceUnit, String targ
    */
   // visible for testing
   static void registerConversion(
-      String sourceUnit, String targetUnit, Function<Number, Number> convertingFunction) {
+      String sourceUnit, String targetUnit, UnaryOperator<Number> convertingFunction) {
     if (sourceUnit.isEmpty()) {
       throw new IllegalArgumentException("Non empty sourceUnit must be provided");
     }
@@ -92,7 +92,7 @@ private static String buildConverterKey(String sourceUnit, String targetUnit) {
    *
    * @param convertingFunction an algorithm applied when converting value
    */
-  UnitConverter(Function<Number, Number> convertingFunction) {
+  UnitConverter(UnaryOperator<Number> convertingFunction) {
     this.convertingFunction = convertingFunction;
   }
 

instrumentation/opentelemetry-api/opentelemetry-api-1.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/opentelemetryapi/context/InstrumentationApiContextBridging.java

@@ -12,7 +12,7 @@
 import java.lang.invoke.MethodType;
 import java.util.ArrayList;
 import java.util.List;
-import java.util.function.Function;
+import java.util.function.UnaryOperator;
 import javax.annotation.Nullable;
 
 final class InstrumentationApiContextBridging {
@@ -199,13 +199,13 @@ final class InstrumentationApiContextBridging {
     }
   }
 
-  private static Function<Object, Object> httpRouteStateConvert(
+  private static UnaryOperator<Object> httpRouteStateConvert(
       MethodHandle create,
       MethodHandle getMethod,
       MethodHandle getRoute,
       MethodHandle getUpdatedBySourceOrder,
       MethodHandle getSpan,
-      Function<Object, Object> convertSpan) {
+      UnaryOperator<Object> convertSpan) {
     return httpRouteHolder -> {
       try {
         String method = (String) getMethod.invoke(httpRouteHolder);

instrumentation/reactor/reactor-3.1/library/src/main/java/io/opentelemetry/instrumentation/reactor/v3_1/ContextPropagationOperator.java

@@ -31,7 +31,7 @@
 import java.lang.invoke.MethodHandle;
 import java.lang.invoke.MethodHandles;
 import java.util.function.BiFunction;
-import java.util.function.Function;
+import java.util.function.UnaryOperator;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 import javax.annotation.Nullable;
@@ -175,7 +175,7 @@ public void registerOnEachOperator() {
     }
   }
 
-  private static void registerScheduleHook(String key, Function<Runnable, Runnable> function) {
+  private static void registerScheduleHook(String key, UnaryOperator<Runnable> function) {
     if (SCHEDULERS_HOOK_METHOD == null) {
       return;
     }
@@ -252,7 +252,7 @@ private static <T extends Throwable> T sneakyThrow(Throwable t) throws T {
   }
 
   private static class StoreOpenTelemetryContext
-      implements Function<reactor.util.context.Context, reactor.util.context.Context> {
+      implements UnaryOperator<reactor.util.context.Context> {
 
     private final Context tracingContext;
 

instrumentation/resources/library/src/main/java/io/opentelemetry/instrumentation/resources/internal/MainJarPathFinder.java

@@ -9,14 +9,14 @@
 import java.nio.file.InvalidPathException;
 import java.nio.file.Path;
 import java.nio.file.Paths;
-import java.util.function.Function;
 import java.util.function.Predicate;
 import java.util.function.Supplier;
+import java.util.function.UnaryOperator;
 import javax.annotation.Nullable;
 
 class MainJarPathFinder {
   private final Supplier<String[]> getProcessHandleArguments;
-  private final Function<String, String> getSystemProperty;
+  private final UnaryOperator<String> getSystemProperty;
   private final Predicate<Path> fileExists;
 
   public MainJarPathFinder() {
@@ -26,7 +26,7 @@ public MainJarPathFinder() {
   // visible for tests
   MainJarPathFinder(
       Supplier<String[]> getProcessHandleArguments,
-      Function<String, String> getSystemProperty,
+      UnaryOperator<String> getSystemProperty,
       Predicate<Path> fileExists) {
     this.getProcessHandleArguments = getProcessHandleArguments;
     this.getSystemProperty = getSystemProperty;

javaagent-extension-api/src/main/java/io/opentelemetry/javaagent/extension/instrumentation/TypeTransformer.java

@@ -5,7 +5,7 @@
 
 package io.opentelemetry.javaagent.extension.instrumentation;
 
-import java.util.function.Function;
+import java.util.function.UnaryOperator;
 import net.bytebuddy.agent.builder.AgentBuilder;
 import net.bytebuddy.asm.Advice;
 import net.bytebuddy.description.method.MethodDescription;
@@ -25,7 +25,7 @@ public interface TypeTransformer {
    */
   default void applyAdviceToMethod(
       ElementMatcher<? super MethodDescription> methodMatcher, String adviceClassName) {
-    applyAdviceToMethod(methodMatcher, Function.identity(), adviceClassName);
+    applyAdviceToMethod(methodMatcher, UnaryOperator.identity(), adviceClassName);
   }
 
   /**
@@ -34,7 +34,7 @@ default void applyAdviceToMethod(
    */
   void applyAdviceToMethod(
       ElementMatcher<? super MethodDescription> methodMatcher,
-      Function<Advice.WithCustomMapping, Advice.WithCustomMapping> mappingCustomizer,
+      UnaryOperator<Advice.WithCustomMapping> mappingCustomizer,
       String adviceClassName);
 
   /**

javaagent-tooling/src/main/java/io/opentelemetry/javaagent/tooling/instrumentation/TypeTransformerImpl.java

@@ -9,7 +9,7 @@
 import io.opentelemetry.javaagent.tooling.Utils;
 import io.opentelemetry.javaagent.tooling.bytebuddy.ExceptionHandlers;
 import io.opentelemetry.javaagent.tooling.instrumentation.indy.ForceDynamicallyTypedAssignReturnedFactory;
-import java.util.function.Function;
+import java.util.function.UnaryOperator;
 import net.bytebuddy.agent.builder.AgentBuilder;
 import net.bytebuddy.asm.Advice;
 import net.bytebuddy.description.method.MethodDescription;
@@ -31,7 +31,7 @@ final class TypeTransformerImpl implements TypeTransformer {
   @Override
   public void applyAdviceToMethod(
       ElementMatcher<? super MethodDescription> methodMatcher,
-      Function<Advice.WithCustomMapping, Advice.WithCustomMapping> mappingCustomizer,
+      UnaryOperator<Advice.WithCustomMapping> mappingCustomizer,
       String adviceClassName) {
     agentBuilder =
         agentBuilder.transform(

javaagent-tooling/src/main/java/io/opentelemetry/javaagent/tooling/instrumentation/indy/IndyTypeTransformerImpl.java

@@ -12,7 +12,7 @@
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.util.function.BiFunction;
-import java.util.function.Function;
+import java.util.function.UnaryOperator;
 import net.bytebuddy.agent.builder.AgentBuilder;
 import net.bytebuddy.asm.Advice;
 import net.bytebuddy.description.method.MethodDescription;
@@ -51,7 +51,7 @@ public IndyTypeTransformerImpl(
   @Override
   public void applyAdviceToMethod(
       ElementMatcher<? super MethodDescription> methodMatcher,
-      Function<Advice.WithCustomMapping, Advice.WithCustomMapping> mappingCustomizer,
+      UnaryOperator<Advice.WithCustomMapping> mappingCustomizer,
       String adviceClassName) {
     // default strategy used by AgentBuilder.Transformer.ForAdvice
     AgentBuilder.PoolStrategy poolStrategy = AgentBuilder.PoolStrategy.Default.FAST;