Feat(deadline) replaced bean deadlineStubTransformer with deadlineGrpcChannelConfigurer

Author: Бацура Сергей Александрович

grpc-client-spring-boot-starter/src/main/java/net/devh/boot/grpc/client/autoconfigure/GrpcClientDeadlineAutoConfiguration.java

@@ -18,6 +18,8 @@
 
 import static java.util.Objects.requireNonNull;
 
+import java.time.Duration;
+
 import org.springframework.boot.autoconfigure.AutoConfigureBefore;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -26,9 +28,8 @@
 import io.grpc.Channel;
 import io.grpc.MethodDescriptor;
 import lombok.extern.slf4j.Slf4j;
-import net.devh.boot.grpc.client.config.GrpcChannelProperties;
+import net.devh.boot.grpc.client.channelfactory.GrpcChannelConfigurer;
 import net.devh.boot.grpc.client.config.GrpcChannelsProperties;
-import net.devh.boot.grpc.client.inject.StubTransformer;
 import net.devh.boot.grpc.client.interceptor.DeadlineSetupClientInterceptor;
 
 /**
@@ -50,7 +51,7 @@
 public class GrpcClientDeadlineAutoConfiguration {
 
     /**
-     * Creates a {@link StubTransformer} bean with interceptor that will call withDeadlineAfter with deadline from
+     * Creates a {@link GrpcChannelConfigurer} bean with interceptor that will call withDeadlineAfter with deadline from
      * props.
      *
      *
@@ -59,16 +60,13 @@ public class GrpcClientDeadlineAutoConfiguration {
      * @see DeadlineSetupClientInterceptor#interceptCall(MethodDescriptor, CallOptions, Channel)
      */
     @Bean
-    StubTransformer deadlineStubTransformer(final GrpcChannelsProperties props) {
+    GrpcChannelConfigurer deadlineGrpcChannelConfigurer(final GrpcChannelsProperties props) {
         requireNonNull(props, "properties");
 
-        return (name, stub) -> {
-            GrpcChannelProperties channelProps = props.getChannel(name);
-            if (channelProps != null && channelProps.getDeadline() != null
-                    && channelProps.getDeadline().toMillis() > 0L) {
-                return stub.withInterceptors(new DeadlineSetupClientInterceptor(channelProps.getDeadline()));
-            } else {
-                return stub;
+        return (channel, name) -> {
+            Duration deadline = props.getChannel(name).getDeadline();
+            if (deadline != null && deadline.toMillis() > 0L) {
+                channel.intercept(new DeadlineSetupClientInterceptor(deadline));
             }
         };
     }