[JobRouter] SDK Review updates (Azure#40011)

* SDK Review updates

* Update auto-generated models

* Add customization

* Fix customization

* Update package

* Update tests

* Linting

Author: williamzhao87

sdk/communication/azure-communication-jobrouter/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "java",
   "TagPrefix": "java/communication/azure-communication-jobrouter",
-  "Tag": "java/communication/azure-communication-jobrouter_b970df9fcf"
+  "Tag": "java/communication/azure-communication-jobrouter_db1da26178"
 }

sdk/communication/azure-communication-jobrouter/customization/src/main/java/JobRouterSdkCustomization.java

@@ -3,6 +3,7 @@
 import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration;
 import com.github.javaparser.ast.type.ClassOrInterfaceType;
 import org.slf4j.Logger;
+import java.lang.reflect.Modifier;
 import java.util.ArrayList;
 import java.util.List;
 
@@ -29,6 +30,13 @@ public void customize(LibraryCustomization customization, Logger logger) {
         addConnectionStringClientMethod(classCustomizationForJobRouterClientBuilder, "JobRouterClientBuilder");
         addHttpPipelineAuthPolicyMethod(classCustomizationForJobRouterClientBuilder);
         updateHttpPipelineMethod(classCustomizationForJobRouterClientBuilder);
+
+        logger.info("Customizing the ScoringRuleOptions class");
+        PackageCustomization modelsPackageCustomization = customization.getPackage("com.azure.communication.jobrouter.models");
+        ClassCustomization classCustomizationForScoringRuleOptions = modelsPackageCustomization.getClass("ScoringRuleOptions");
+        classCustomizationForScoringRuleOptions
+            .getMethod("setIsBatchScoringEnabled")
+            .setModifier(Modifier.PRIVATE);
     }
 
     private void addAuthTraits(ClassCustomization classCustomization) {

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/implementation/converters/RouterRuleAdapter.java

@@ -49,9 +49,9 @@ public static RouterRuleInternal getRouterRuleInternal(RouterRule routerRule) {
             } else if (routerRule.getClass() == WebhookRouterRule.class) {
                 WebhookRouterRule webhookRouterRule = (WebhookRouterRule) routerRule;
                 prioritizationRuleInternal = new WebhookRouterRuleInternal()
-                    .setAuthorizationServerUri(webhookRouterRule.getAuthorizationServerUri())
+                    .setAuthorizationServerUri(webhookRouterRule.getAuthorizationServerUrl())
                     .setClientCredential(webhookRouterRule.getClientCredential())
-                    .setWebhookUri(webhookRouterRule.getWebhookUri());
+                    .setWebhookUri(webhookRouterRule.getWebhookUrl());
             }
         }
 
@@ -75,11 +75,11 @@ public static RouterRuleInternal convertRouterRuleToInternal(RouterRule rule) {
             return new StaticRouterRuleInternal().setValue(RouterValueAdapter.getValue(((StaticRouterRule) rule).getValue()));
         } else if (rule instanceof WebhookRouterRule) {
             WebhookRouterRule webhookRouterRule = (WebhookRouterRule) rule;
-            return new WebhookRouterRuleInternal().setWebhookUri(webhookRouterRule.getWebhookUri())
+            return new WebhookRouterRuleInternal().setWebhookUri(webhookRouterRule.getWebhookUrl())
                 .setClientCredential(new OAuth2WebhookClientCredential()
                     .setClientId(webhookRouterRule.getClientCredential().getClientId())
                     .setClientSecret(webhookRouterRule.getClientCredential().getClientSecret()))
-                .setAuthorizationServerUri(webhookRouterRule.getAuthorizationServerUri());
+                .setAuthorizationServerUri(webhookRouterRule.getAuthorizationServerUrl());
         }
 
         return null;

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/models/BestWorkerMode.java

@@ -30,8 +30,14 @@ public final class BestWorkerMode extends DistributionMode {
     private ScoringRuleOptions scoringRuleOptions;
 
     /** Creates an instance of BestWorkerMode class. */
-    public BestWorkerMode() {
-        this.kind = DistributionModeKind.BEST_WORKER;
+    public BestWorkerMode() {}
+
+    /**
+     * Returns kind discriminator.
+     * @return kind.
+     */
+    public DistributionModeKind getKind() {
+        return DistributionModeKind.BEST_WORKER;
     }
 
     /**

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/models/CancelExceptionAction.java

@@ -29,8 +29,14 @@ public final class CancelExceptionAction extends ExceptionAction {
     private String dispositionCode;
 
     /** Creates an instance of CancelExceptionAction class. */
-    public CancelExceptionAction() {
-        this.kind = ExceptionActionKind.CANCEL;
+    public CancelExceptionAction() {}
+
+    /**
+     * Returns kind discriminator.
+     * @return kind.
+     */
+    public ExceptionActionKind getKind() {
+        return ExceptionActionKind.CANCEL;
     }
 
     /**

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/models/ConditionalQueueSelectorAttachment.java

@@ -53,7 +53,14 @@ public ConditionalQueueSelectorAttachment(
             @JsonProperty(value = "queueSelectors") List<RouterQueueSelector> queueSelectors) {
         this.condition = condition;
         this.queueSelectors = queueSelectors;
-        this.kind = QueueSelectorAttachmentKind.CONDITIONAL;
+    }
+
+    /**
+     * Returns kind discriminator.
+     * @return kind.
+     */
+    public QueueSelectorAttachmentKind getKind() {
+        return QueueSelectorAttachmentKind.CONDITIONAL;
     }
 
     /**

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/models/ConditionalWorkerSelectorAttachment.java

@@ -53,7 +53,14 @@ public ConditionalWorkerSelectorAttachment(
             @JsonProperty(value = "workerSelectors") List<RouterWorkerSelector> workerSelectors) {
         this.condition = condition;
         this.workerSelectors = workerSelectors;
-        this.kind = WorkerSelectorAttachmentKind.CONDITIONAL;
+    }
+
+    /**
+     * Returns kind discriminator.
+     * @return kind.
+     */
+    public WorkerSelectorAttachmentKind getKind() {
+        return WorkerSelectorAttachmentKind.CONDITIONAL;
     }
 
     /**

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/models/DirectMapRouterRule.java

@@ -14,7 +14,13 @@
 public final class DirectMapRouterRule extends RouterRule {
 
     /** Creates an instance of DirectMapRouterRule class. */
-    public DirectMapRouterRule() {
-        this.kind = RouterRuleKind.DIRECT_MAP;
+    public DirectMapRouterRule() {}
+
+    /**
+     * Returns kind discriminator.
+     * @return kind.
+     */
+    public RouterRuleKind getKind() {
+        return RouterRuleKind.DIRECT_MAP;
     }
 }

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/models/DistributionMode.java

@@ -28,7 +28,7 @@ public abstract class DistributionMode {
      * kind discriminator.
      */
     @JsonProperty(value = "kind")
-    protected DistributionModeKind kind;
+    private DistributionModeKind kind;
 
     /*
      * Governs the minimum desired number of active concurrent offers a job can have.
@@ -123,7 +123,5 @@ public DistributionMode setBypassSelectors(Boolean bypassSelectors) {
      * Returns kind discriminator.
      * @return kind.
      */
-    public DistributionModeKind getKind() {
-        return this.kind;
-    }
+    public abstract DistributionModeKind getKind();
 }

sdk/communication/azure-communication-jobrouter/src/main/java/com/azure/communication/jobrouter/models/ExceptionAction.java

@@ -3,6 +3,7 @@
 // Code generated by Microsoft (R) AutoRest Code Generator.
 package com.azure.communication.jobrouter.models;
 
+import com.azure.core.annotation.Immutable;
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.fasterxml.jackson.annotation.JsonSubTypes;
 import com.fasterxml.jackson.annotation.JsonTypeInfo;
@@ -20,12 +21,13 @@
     @JsonSubTypes.Type(name = "manualReclassify", value = ManualReclassifyExceptionAction.class),
     @JsonSubTypes.Type(name = "reclassify", value = ReclassifyExceptionAction.class)
 })
+@Immutable
 public abstract class ExceptionAction {
     /**
      * kind discriminator.
      */
     @JsonProperty(value = "kind")
-    protected ExceptionActionKind kind;
+    private ExceptionActionKind kind;
 
     /** Creates an instance of ExceptionAction class. */
     public ExceptionAction() {}
@@ -49,7 +51,5 @@ public String getId() {
      * Returns kind discriminator.
      * @return kind.
      */
-    public ExceptionActionKind getKind() {
-        return this.kind;
-    }
+    public abstract ExceptionActionKind getKind();
 }