[AutoPR] batchai/resource-manager

azure-mgmt-batchai/src/main/java/com/microsoft/azure/management/batchai/AppInsightsReference.java

@@ -0,0 +1,100 @@
+/**
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for
+ * license information.
+ *
+ * Code generated by Microsoft (R) AutoRest Code Generator.
+ */
+
+package com.microsoft.azure.management.batchai;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * Specifies Azure Application Insights information for performance counters
+ * reporting.
+ */
+public class AppInsightsReference {
+    /**
+     * Specifies the Azure Application Insights component resource id.
+     */
+    @JsonProperty(value = "component", required = true)
+    private ResourceId component;
+
+    /**
+     * Value of the Azure Application Insights instrumentation key.
+     */
+    @JsonProperty(value = "instrumentationKey")
+    private String instrumentationKey;
+
+    /**
+     * Specifies a KeyVault Secret containing Azure Application Insights
+     * instrumentation key.
+     * Specifies KeyVault Store and Secret which contains Azure Application
+     * Insights instrumentation key. One of instumentationKey or
+     * instrumentationKeySecretReference must be specified.
+     */
+    @JsonProperty(value = "instrumentationKeySecretReference")
+    private KeyVaultSecretReference instrumentationKeySecretReference;
+
+    /**
+     * Get the component value.
+     *
+     * @return the component value
+     */
+    public ResourceId component() {
+        return this.component;
+    }
+
+    /**
+     * Set the component value.
+     *
+     * @param component the component value to set
+     * @return the AppInsightsReference object itself.
+     */
+    public AppInsightsReference withComponent(ResourceId component) {
+        this.component = component;
+        return this;
+    }
+
+    /**
+     * Get the instrumentationKey value.
+     *
+     * @return the instrumentationKey value
+     */
+    public String instrumentationKey() {
+        return this.instrumentationKey;
+    }
+
+    /**
+     * Set the instrumentationKey value.
+     *
+     * @param instrumentationKey the instrumentationKey value to set
+     * @return the AppInsightsReference object itself.
+     */
+    public AppInsightsReference withInstrumentationKey(String instrumentationKey) {
+        this.instrumentationKey = instrumentationKey;
+        return this;
+    }
+
+    /**
+     * Get the instrumentationKeySecretReference value.
+     *
+     * @return the instrumentationKeySecretReference value
+     */
+    public KeyVaultSecretReference instrumentationKeySecretReference() {
+        return this.instrumentationKeySecretReference;
+    }
+
+    /**
+     * Set the instrumentationKeySecretReference value.
+     *
+     * @param instrumentationKeySecretReference the instrumentationKeySecretReference value to set
+     * @return the AppInsightsReference object itself.
+     */
+    public AppInsightsReference withInstrumentationKeySecretReference(KeyVaultSecretReference instrumentationKeySecretReference) {
+        this.instrumentationKeySecretReference = instrumentationKeySecretReference;
+        return this;
+    }
+
+}

azure-mgmt-batchai/src/main/java/com/microsoft/azure/management/batchai/AzureBlobFileSystemReference.java

@@ -36,8 +36,9 @@ public class AzureBlobFileSystemReference {
     /**
      * Specifies the relative path on the compute node where the Azure Blob
      * file system will be mounted.
-     * Note that all blob file systems will be mounted under
-     * $AZ_BATCHAI_MOUNT_ROOT location.
+     * Note that all cluster level blob file systems will be mounted under
+     * $AZ_BATCHAI_MOUNT_ROOT location and all job level blob file systems will
+     * be mounted under $AZ_BATCHAI_JOB_MOUNT_ROOT.
      */
     @JsonProperty(value = "relativeMountPath", required = true)
     private String relativeMountPath;

azure-mgmt-batchai/src/main/java/com/microsoft/azure/management/batchai/AzureFileShareReference.java

@@ -35,8 +35,9 @@ public class AzureFileShareReference {
     /**
      * Specifies the relative path on the compute node where the Azure file
      * share will be mounted.
-     * Note that all file shares will be mounted under $AZ_BATCHAI_MOUNT_ROOT
-     * location.
+     * Note that all cluster level file shares will be mounted under
+     * $AZ_BATCHAI_MOUNT_ROOT location and all job level file shares will be
+     * mounted under $AZ_BATCHAI_JOB_MOUNT_ROOT.
      */
     @JsonProperty(value = "relativeMountPath", required = true)
     private String relativeMountPath;

azure-mgmt-batchai/src/main/java/com/microsoft/azure/management/batchai/DataDisks.java

@@ -26,7 +26,7 @@ public class DataDisks {
      * patchable.
      * Possible values include: 'none', 'readonly', 'readwrite'.
      */
-    @JsonProperty(value = "cachingType", access = JsonProperty.Access.WRITE_ONLY)
+    @JsonProperty(value = "cachingType")
     private CachingType cachingType;
 
     /**
@@ -73,6 +73,17 @@ public CachingType cachingType() {
         return this.cachingType;
     }
 
+    /**
+     * Set the cachingType value.
+     *
+     * @param cachingType the cachingType value to set
+     * @return the DataDisks object itself.
+     */
+    public DataDisks withCachingType(CachingType cachingType) {
+        this.cachingType = cachingType;
+        return this;
+    }
+
     /**
      * Get the diskCount value.
      *

azure-mgmt-batchai/src/main/java/com/microsoft/azure/management/batchai/EnvironmentVariable.java

@@ -0,0 +1,69 @@
+/**
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for
+ * license information.
+ *
+ * Code generated by Microsoft (R) AutoRest Code Generator.
+ */
+
+package com.microsoft.azure.management.batchai;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * A collection of environment variables to set.
+ */
+public class EnvironmentVariable {
+    /**
+     * The name of the environment variable.
+     */
+    @JsonProperty(value = "name", required = true)
+    private String name;
+
+    /**
+     * The value of the environment variable.
+     */
+    @JsonProperty(value = "value", required = true)
+    private String value;
+
+    /**
+     * Get the name value.
+     *
+     * @return the name value
+     */
+    public String name() {
+        return this.name;
+    }
+
+    /**
+     * Set the name value.
+     *
+     * @param name the name value to set
+     * @return the EnvironmentVariable object itself.
+     */
+    public EnvironmentVariable withName(String name) {
+        this.name = name;
+        return this;
+    }
+
+    /**
+     * Get the value value.
+     *
+     * @return the value value
+     */
+    public String value() {
+        return this.value;
+    }
+
+    /**
+     * Set the value value.
+     *
+     * @param value the value value to set
+     * @return the EnvironmentVariable object itself.
+     */
+    public EnvironmentVariable withValue(String value) {
+        this.value = value;
+        return this;
+    }
+
+}

azure-mgmt-batchai/src/main/java/com/microsoft/azure/management/batchai/EnvironmentVariableWithSecretValue.java

@@ -0,0 +1,100 @@
+/**
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for
+ * license information.
+ *
+ * Code generated by Microsoft (R) AutoRest Code Generator.
+ */
+
+package com.microsoft.azure.management.batchai;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * A collection of environment variables with secret values to set.
+ */
+public class EnvironmentVariableWithSecretValue {
+    /**
+     * The name of the environment variable to store the secret value.
+     */
+    @JsonProperty(value = "name", required = true)
+    private String name;
+
+    /**
+     * The value of the environment variable. This value will never be reported
+     * back by Batch AI.
+     */
+    @JsonProperty(value = "value")
+    private String value;
+
+    /**
+     * Specifies the location of the Azure KeyVault secret which will be used
+     * as the environment variable value.
+     * Specifies KeyVault Store and Secret which contains the value for the
+     * environment variable. One of value or valueSecretReference must be
+     * provided.
+     */
+    @JsonProperty(value = "valueSecretReference")
+    private KeyVaultSecretReference valueSecretReference;
+
+    /**
+     * Get the name value.
+     *
+     * @return the name value
+     */
+    public String name() {
+        return this.name;
+    }
+
+    /**
+     * Set the name value.
+     *
+     * @param name the name value to set
+     * @return the EnvironmentVariableWithSecretValue object itself.
+     */
+    public EnvironmentVariableWithSecretValue withName(String name) {
+        this.name = name;
+        return this;
+    }
+
+    /**
+     * Get the value value.
+     *
+     * @return the value value
+     */
+    public String value() {
+        return this.value;
+    }
+
+    /**
+     * Set the value value.
+     *
+     * @param value the value value to set
+     * @return the EnvironmentVariableWithSecretValue object itself.
+     */
+    public EnvironmentVariableWithSecretValue withValue(String value) {
+        this.value = value;
+        return this;
+    }
+
+    /**
+     * Get the valueSecretReference value.
+     *
+     * @return the valueSecretReference value
+     */
+    public KeyVaultSecretReference valueSecretReference() {
+        return this.valueSecretReference;
+    }
+
+    /**
+     * Set the valueSecretReference value.
+     *
+     * @param valueSecretReference the valueSecretReference value to set
+     * @return the EnvironmentVariableWithSecretValue object itself.
+     */
+    public EnvironmentVariableWithSecretValue withValueSecretReference(KeyVaultSecretReference valueSecretReference) {
+        this.valueSecretReference = valueSecretReference;
+        return this;
+    }
+
+}

azure-mgmt-batchai/src/main/java/com/microsoft/azure/management/batchai/FileServerReference.java

@@ -32,8 +32,9 @@ public class FileServerReference {
     /**
      * Specifies the relative path on the compute node where the File Server
      * will be mounted.
-     * Note that all file shares will be mounted under $AZ_BATCHAI_MOUNT_ROOT
-     * location.
+     * Note that all cluster level file servers will be mounted under
+     * $AZ_BATCHAI_MOUNT_ROOT location and job level file servers will be
+     * mouted under $AZ_BATCHAI_JOB_MOUNT_ROOT.
      */
     @JsonProperty(value = "relativeMountPath", required = true)
     private String relativeMountPath;

azure-mgmt-batchai/src/main/java/com/microsoft/azure/management/batchai/ImageReference.java

@@ -38,6 +38,21 @@ public class ImageReference {
     @JsonProperty(value = "version")
     private String version;
 
+    /**
+     * The ARM resource identifier of the virtual machine image. Computes nodes
+     * of the cluster will be created using this custom image. This is of the
+     * form
+     * /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/images/{imageName}.
+     * The virtual machine image must be in the same region and subscription as
+     * the cluster. For information about the firewall settings for the Batch
+     * node agent to communicate with the Batch service see
+     * https://docs.microsoft.com/en-us/azure/batch/batch-api-basics#virtual-network-vnet-and-firewall-configuration.
+     * Note, you need to provide publisher, offer and sku of the base OS image
+     * of which the custom image has been derived from.
+     */
+    @JsonProperty(value = "virtualMachineImageId")
+    private String virtualMachineImageId;
+
     /**
      * Get the publisher value.
      *
@@ -118,4 +133,24 @@ public ImageReference withVersion(String version) {
         return this;
     }
 
+    /**
+     * Get the virtualMachineImageId value.
+     *
+     * @return the virtualMachineImageId value
+     */
+    public String virtualMachineImageId() {
+        return this.virtualMachineImageId;
+    }
+
+    /**
+     * Set the virtualMachineImageId value.
+     *
+     * @param virtualMachineImageId the virtualMachineImageId value to set
+     * @return the ImageReference object itself.
+     */
+    public ImageReference withVirtualMachineImageId(String virtualMachineImageId) {
+        this.virtualMachineImageId = virtualMachineImageId;
+        return this;
+    }
+
 }

azure-mgmt-batchai/src/main/java/com/microsoft/azure/management/batchai/InputDirectory.java

@@ -16,11 +16,9 @@
 public class InputDirectory {
     /**
      * The id for the input directory.
-     * It will be available for the job as an environment variable under
-     * AZ_BATCHAI_INPUT_id. The service will also provide the following
-     * environment variable: AZ_BATCHAI_PREV_OUTPUT_Name. The value of the
-     * variable will be populated if the job is being retried after a previous
-     * failure, otherwise it will be set to nothing.
+     * The path of the input directory will be available as a value of an
+     * environment variable with AZ_BATCHAI_INPUT_<id> name, where
+     * <id> is the value of id attribute.
      */
     @JsonProperty(value = "id", required = true)
     private String id;