AzureKeyCredential to KeyCredential (Azure#37682)

* AzureKeyCredential to KeyCredential

* Make exceptionPolicy methods public

Author: cparisineti

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/JobRouterAdministrationAsyncClient.java

@@ -18,6 +18,7 @@
 import com.azure.communication.jobrouter.models.CreateExceptionPolicyOptions;
 import com.azure.communication.jobrouter.models.CreateQueueOptions;
 import com.azure.communication.jobrouter.models.DistributionPolicy;
+import com.azure.communication.jobrouter.models.ExceptionPolicy;
 import com.azure.communication.jobrouter.models.RouterQueue;
 import com.azure.core.annotation.Generated;
 import com.azure.core.annotation.ReturnType;
@@ -1393,13 +1394,12 @@ public Mono<Void> deleteClassificationPolicy(String classificationPolicyId) {
      * @return a policy that defines actions to execute when exception are triggered on successful completion of
      * {@link Mono}.
      */
-    @Generated
     @ServiceMethod(returns = ReturnType.SINGLE)
-    Mono<ExceptionPolicyInternal> getExceptionPolicy(String exceptionPolicyId) {
+    public Mono<ExceptionPolicy> getExceptionPolicy(String exceptionPolicyId) {
         // Generated convenience method for getExceptionPolicyWithResponse
         RequestOptions requestOptions = new RequestOptions();
         return getExceptionPolicyWithResponse(exceptionPolicyId, requestOptions).flatMap(FluxUtil::toMono)
-            .map(protocolMethodData -> protocolMethodData.toObject(ExceptionPolicyInternal.class));
+            .map(protocolMethodData -> protocolMethodData.toObject(ExceptionPolicy.class));
     }
 
     /**
@@ -1412,22 +1412,20 @@ Mono<ExceptionPolicyInternal> getExceptionPolicy(String exceptionPolicyId) {
      * @throws RuntimeException all other wrapped checked exceptions if the request fails to be sent.
      * @return paged collection of ExceptionPolicy items as paginated response with {@link PagedFlux}.
      */
-    @Generated
     @ServiceMethod(returns = ReturnType.COLLECTION)
-    PagedFlux<ExceptionPolicyInternal> listExceptionPolicies() {
+    public PagedFlux<ExceptionPolicy> listExceptionPolicies() {
         // Generated convenience method for listExceptionPolicies
         RequestOptions requestOptions = new RequestOptions();
         PagedFlux<BinaryData> pagedFluxResponse = listExceptionPolicies(requestOptions);
         return PagedFlux.create(() -> (continuationToken, pageSize) -> {
             Flux<PagedResponse<BinaryData>> flux = (continuationToken == null) ? pagedFluxResponse.byPage().take(1)
                 : pagedFluxResponse.byPage(continuationToken).take(1);
-            return flux
-                .map(pagedResponse -> new PagedResponseBase<Void, ExceptionPolicyInternal>(pagedResponse.getRequest(),
-                    pagedResponse.getStatusCode(), pagedResponse.getHeaders(),
-                    pagedResponse.getValue().stream()
-                        .map(protocolMethodData -> protocolMethodData.toObject(ExceptionPolicyInternal.class))
-                        .collect(Collectors.toList()),
-                    pagedResponse.getContinuationToken(), null));
+            return flux.map(pagedResponse -> new PagedResponseBase<Void, ExceptionPolicy>(pagedResponse.getRequest(),
+                pagedResponse.getStatusCode(), pagedResponse.getHeaders(),
+                pagedResponse.getValue().stream()
+                    .map(protocolMethodData -> protocolMethodData.toObject(ExceptionPolicy.class))
+                    .collect(Collectors.toList()),
+                pagedResponse.getContinuationToken(), null));
         });
     }
 

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/JobRouterAdministrationClientBuilder.java

@@ -10,7 +10,7 @@
 import com.azure.core.client.traits.ConfigurationTrait;
 import com.azure.core.client.traits.EndpointTrait;
 import com.azure.core.client.traits.HttpTrait;
-import com.azure.core.credential.AzureKeyCredential;
+import com.azure.core.credential.KeyCredential;
 import com.azure.core.credential.TokenCredential;
 import com.azure.core.http.HttpClient;
 import com.azure.core.http.HttpHeaderName;
@@ -65,7 +65,7 @@ public final class JobRouterAdministrationClientBuilder implements HttpTrait<Job
 
     private CommunicationConnectionString connectionString;
 
-    private AzureKeyCredential credential;
+    private KeyCredential credential;
 
     /**
      * Create an instance of the JobRouterAdministrationClientBuilder.
@@ -315,7 +315,7 @@ public JobRouterAdministrationClient buildClient() {
      */
     public JobRouterAdministrationClientBuilder connectionString(String connectionString) {
         this.connectionString = new CommunicationConnectionString(connectionString);
-        this.credential(new AzureKeyCredential(this.connectionString.getAccessKey()));
+        this.credential(new KeyCredential(this.connectionString.getAccessKey()));
         this.endpoint(this.connectionString.getEndpoint());
         return this;
     }
@@ -340,7 +340,7 @@ public JobRouterAdministrationClientBuilder credential(TokenCredential tokenCred
      * @param credential valid credential as a string
      * @return the updated RouterAdministrationClientBuilder object
      */
-    public JobRouterAdministrationClientBuilder credential(AzureKeyCredential credential) {
+    public JobRouterAdministrationClientBuilder credential(KeyCredential credential) {
         this.credential = Objects.requireNonNull(credential, "'credential' cannot be null.");
         return this;
     }

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/JobRouterClientBuilder.java

@@ -11,6 +11,7 @@
 import com.azure.core.client.traits.EndpointTrait;
 import com.azure.core.client.traits.HttpTrait;
 import com.azure.core.credential.AzureKeyCredential;
+import com.azure.core.credential.KeyCredential;
 import com.azure.core.credential.TokenCredential;
 import com.azure.core.http.HttpClient;
 import com.azure.core.http.HttpHeaderName;
@@ -62,7 +63,7 @@ public final class JobRouterClientBuilder implements HttpTrait<JobRouterClientBu
 
     private TokenCredential tokenCredential;
 
-    private AzureKeyCredential credential;
+    private KeyCredential credential;
 
     private CommunicationConnectionString connectionString;
 
@@ -326,7 +327,7 @@ public JobRouterClientBuilder credential(TokenCredential tokenCredential) {
      * @param credential valid credential as a string
      * @return the updated RouterClientBuilder object
      */
-    public JobRouterClientBuilder credential(AzureKeyCredential credential) {
+    public JobRouterClientBuilder credential(KeyCredential credential) {
         this.credential = Objects.requireNonNull(credential, "'credential' cannot be null.");
         return this;
     }