Added Grpc Delegated Token Jwt.

Author: zaxxio

docker-compose.yml

@@ -130,6 +130,7 @@ services:
       - POSTGRESQL_DATABASE=${DB_NAME}
       - ALLOW_EMPTY_PASSWORD=yes
 
+  # Postgres Slave
   postgresql-slave:
     image: bitnami/postgresql
     restart: always

pom.xml

@@ -21,7 +21,7 @@
 
         <protobuf.version>3.23.4</protobuf.version>
         <protobuf-plugin.version>0.6.1</protobuf-plugin.version>
-        <grpc.version>1.58.0</grpc.version>
+        <grpc.version>1.62.2</grpc.version>
     </properties>
     <dependencies>
         <dependency>

src/main/java/org/wsd/app/config/FeignConfig.java

@@ -1,9 +1,27 @@
 package org.wsd.app.config;
 
+import feign.RequestInterceptor;
+import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.http.HttpHeaders;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+import org.springframework.security.oauth2.server.resource.authentication.JwtAuthenticationToken;
 
 
 @Configuration
 public class FeignConfig {
 
+    @Bean
+    public RequestInterceptor requestInterceptor() {
+        return requestTemplate -> {
+            Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+            if (authentication != null && authentication instanceof JwtAuthenticationToken) {
+                JwtAuthenticationToken jwtAuthenticationToken = (JwtAuthenticationToken) authentication;
+                String tokenValue = jwtAuthenticationToken.getToken().getTokenValue();
+                requestTemplate.header(HttpHeaders.AUTHORIZATION, "Bearer " + tokenValue);
+            }
+        };
+    }
+
 }

src/main/java/org/wsd/app/config/GrpcConfig.java

@@ -23,12 +23,14 @@
 package org.wsd.app.config;
 
 import io.grpc.ManagedChannelBuilder;
+import lombok.RequiredArgsConstructor;
 import net.devh.boot.grpc.client.channelfactory.GrpcChannelConfigurer;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.autoconfigure.ImportAutoConfiguration;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.wsd.app.grpc.interceptor.ErrorHandlingClientInterceptor;
+import org.wsd.app.grpc.interceptor.GrpcClientInterceptor;
 
 @Configuration
 @ImportAutoConfiguration({
@@ -40,17 +42,19 @@
         net.devh.boot.grpc.client.autoconfigure.GrpcDiscoveryClientAutoConfiguration.class,
         net.devh.boot.grpc.common.autoconfigure.GrpcCommonCodecAutoConfiguration.class,
 })
+@RequiredArgsConstructor
 public class GrpcConfig {
 
 
-    @Autowired
-    private ErrorHandlingClientInterceptor errorHandlingClientInterceptor;
+    private final ErrorHandlingClientInterceptor errorHandlingClientInterceptor;
+    private final GrpcClientInterceptor grpcClientInterceptor;
 
     @Bean
     public GrpcChannelConfigurer clientInterceptorConfigurer() {
         return (channelBuilder, name) -> {
-            if (channelBuilder instanceof ManagedChannelBuilder) {
+            if (channelBuilder != null) {
                 channelBuilder.intercept(errorHandlingClientInterceptor);
+                channelBuilder.intercept(grpcClientInterceptor);
             }
         };
     }

src/main/java/org/wsd/app/config/RetryConfiguration.java

@@ -32,5 +32,4 @@
 @Configuration
 public class RetryConfiguration {
 
-
 }

src/main/java/org/wsd/app/config/TransactionManagerConfig.java

@@ -45,10 +45,7 @@
 @RequiredArgsConstructor
 @EnableTransactionManagement
 public class TransactionManagerConfig {
-
-
     private final ProducerFactory<UUID, Object> producerFactory;
-
     @Primary
     @Bean(name = "transactionManager")
     public PlatformTransactionManager transactionManager(EntityManagerFactory entityManagerFactory) {

src/main/java/org/wsd/app/controller/AuthenticationController.java

@@ -28,7 +28,6 @@
 @RequiredArgsConstructor
 public class AuthenticationController {
     private final AuthenticationService authenticationService;
-
     @ResponseStatus(value = HttpStatus.OK)
     @Operation(description = "Sign In", summary = "Endpoint for user sign in.")
     @PostMapping(path = "/signIn",
@@ -42,7 +41,6 @@ public ResponseEntity<?> signIn(@Valid @RequestBody
         return ResponseEntity.status(HttpStatus.OK)
                 .body(authenticationService.signIn(signInRequest));
     }
-
     @Operation(description = "Sign Up", summary = "Endpoint for user sign up.")
     @PostMapping(path = "/signUp",
             consumes = {MediaType.APPLICATION_JSON_VALUE, MediaType.APPLICATION_XML_VALUE},

src/main/java/org/wsd/app/controller/OrderController.java

@@ -14,9 +14,7 @@
 @SecurityRequirement(name = "BEARER_TOKEN")
 @RequestMapping("/api/orders")
 public class OrderController {
-
     private final OrderService orderService;
-
     @GetMapping("/{orderId}")
     @ResponseStatus(HttpStatus.FOUND)
     public ResponseEntity<?> getOrderById(@PathVariable("orderId") int orderId) {
@@ -25,5 +23,4 @@ public ResponseEntity<?> getOrderById(@PathVariable("orderId") int orderId) {
         return ResponseEntity.
                 ok("Success");
     }
-
 }

src/main/java/org/wsd/app/controller/TwoFactorAuthController.java

@@ -18,13 +18,10 @@
 @Tag(name = "Two Factor Authentication")
 @SecurityRequirement(name = "BEARER_TOKEN")
 public class TwoFactorAuthController {
-
     private final TwoFactorService twoFactorService;
-
     @Operation(description = "Two Factor Authentication", summary = "Endpoint for two factor authentication service enable and disable.")
     @GetMapping("/2Fa/configure")
     public Payload<TwoFactorResponse> twoFactorSetup(TwoFactorRequest request) {
         return twoFactorService.setup(request);
     }
-
 }

src/main/java/org/wsd/app/grpc/OrderService.java

@@ -38,7 +38,7 @@
 public class OrderService {
 
     @GrpcClient("grpc-client")
-    OrderServiceGrpc.OrderServiceBlockingStub orderServiceBlockingStub;
+    public OrderServiceGrpc.OrderServiceBlockingStub orderServiceBlockingStub;
 
     public void getOrderById(int orderId) {
         log.info("Order Request Id : " + orderId);