Add null check for MDC values

wso2-extensions · Jun 27, 2024 · 08360a5 · 08360a5
1 parent 8271038
commit 08360a5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/...ain/java/org/wso2/carbon/identity/auth/service/handler/impl/OAuth2AccessTokenHandler.java b/...ain/java/org/wso2/carbon/identity/auth/service/handler/impl/OAuth2AccessTokenHandler.java
@@ -204,16 +204,15 @@ protected AuthenticationResult doAuthenticate(MessageContext messageContext) {
 
                 if (serviceProviderName != null){
                     authenticationContext.addParameter(SERVICE_PROVIDER_NAME, serviceProviderName);
+                    MDC.put(SERVICE_PROVIDER_NAME, serviceProviderName);
                 }
                 if (serviceProviderTenantDomain != null) {
                     authenticationContext.addParameter(SERVICE_PROVIDER_TENANT_DOMAIN, serviceProviderTenantDomain);
                 }
                 if (serviceProviderUUID != null) {
                     authenticationContext.addParameter(SERVICE_PROVIDER_UUID, serviceProviderUUID);
+                    MDC.put(SERVICE_PROVIDER_UUID, serviceProviderUUID);
                 }
-
-                MDC.put(SERVICE_PROVIDER_NAME, serviceProviderName);
-                MDC.put(SERVICE_PROVIDER_UUID, serviceProviderUUID);
                 // Set OAuth service provider details to be consumed by the provisioning framework.
                 setProvisioningServiceProviderThreadLocal(oAuth2IntrospectionResponseDTO.getClientId(),
                         serviceProviderTenantDomain);