adding enableSmbDirectoryLease property

sdk/storage/azure-storage-file-share/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "java",
   "TagPrefix": "java/storage/azure-storage-file-share",
-  "Tag": "java/storage/azure-storage-file-share_bf4bf2f437"
+  "Tag": "java/storage/azure-storage-file-share_e7a870d29a"
 }

sdk/storage/azure-storage-file-share/src/main/java/com/azure/storage/file/share/ShareAsyncClient.java

@@ -361,7 +361,6 @@ public Mono<Response<ShareInfo>> createWithResponse(ShareCreateOptions options)
     }
 
     Mono<Response<ShareInfo>> createWithResponse(ShareCreateOptions options, Context context) {
-        context = context == null ? Context.NONE : context;
         options = options == null ? new ShareCreateOptions() : options;
         String enabledProtocol = options.getProtocols() == null ? null : options.getProtocols().toString();
         enabledProtocol = "".equals(enabledProtocol) ? null : enabledProtocol;
@@ -370,7 +369,8 @@ Mono<Response<ShareInfo>> createWithResponse(ShareCreateOptions options, Context
                 options.getAccessTier(), enabledProtocol, options.getRootSquash(),
                 options.isSnapshotVirtualDirectoryAccessEnabled(), options.isPaidBurstingEnabled(),
                 options.getPaidBurstingMaxBandwidthMibps(), options.getPaidBurstingMaxIops(),
-                options.getProvisionedMaxIops(), options.getProvisionedMaxBandwidthMibps(), context)
+                options.getProvisionedMaxIops(), options.getProvisionedMaxBandwidthMibps(),
+                options.isEnableSmbDirectoryLease())
             .map(ModelHelper::mapToShareInfoResponse);
     }
 
@@ -933,13 +933,13 @@ Mono<Response<ShareInfo>> setPropertiesWithResponse(ShareSetPropertiesOptions op
         StorageImplUtils.assertNotNull("options", options);
         ShareRequestConditions requestConditions
             = options.getRequestConditions() == null ? new ShareRequestConditions() : options.getRequestConditions();
-        context = context == null ? Context.NONE : context;
         return azureFileStorageClient.getShares()
             .setPropertiesNoCustomHeadersWithResponseAsync(shareName, null, options.getQuotaInGb(),
                 options.getAccessTier(), requestConditions.getLeaseId(), options.getRootSquash(),
                 options.isSnapshotVirtualDirectoryAccessEnabled(), options.isPaidBurstingEnabled(),
                 options.getPaidBurstingMaxBandwidthMibps(), options.getPaidBurstingMaxIops(),
-                options.getProvisionedMaxIops(), options.getProvisionedMaxBandwidthMibps(), context)
+                options.getProvisionedMaxIops(), options.getProvisionedMaxBandwidthMibps(),
+                options.isEnableSmbDirectoryLease())
             .map(ModelHelper::mapToShareInfoResponse);
     }
 

sdk/storage/azure-storage-file-share/src/main/java/com/azure/storage/file/share/ShareClient.java

@@ -367,7 +367,8 @@ public Response<ShareInfo> createWithResponse(ShareCreateOptions options, Durati
                 finalOptions.getAccessTier(), finalEnabledProtocol, finalOptions.getRootSquash(),
                 finalOptions.isSnapshotVirtualDirectoryAccessEnabled(), finalOptions.isPaidBurstingEnabled(),
                 finalOptions.getPaidBurstingMaxBandwidthMibps(), finalOptions.getPaidBurstingMaxIops(),
-                finalOptions.getProvisionedMaxIops(), finalOptions.getProvisionedMaxBandwidthMibps(), finalContext);
+                finalOptions.getProvisionedMaxIops(), finalOptions.getProvisionedMaxBandwidthMibps(),
+                finalOptions.isEnableSmbDirectoryLease(), finalContext);
 
         return ModelHelper.mapToShareInfoResponse(sendRequest(operation, timeout, ShareStorageException.class));
     }
@@ -897,7 +898,8 @@ public Response<ShareInfo> setPropertiesWithResponse(ShareSetPropertiesOptions o
                 requestConditions.getLeaseId(), options.getRootSquash(),
                 options.isSnapshotVirtualDirectoryAccessEnabled(), options.isPaidBurstingEnabled(),
                 options.getPaidBurstingMaxBandwidthMibps(), options.getPaidBurstingMaxIops(),
-                options.getProvisionedMaxIops(), options.getProvisionedMaxBandwidthMibps(), finalContext);
+                options.getProvisionedMaxIops(), options.getProvisionedMaxBandwidthMibps(),
+                options.isEnableSmbDirectoryLease(), finalContext);
 
         return ModelHelper.mapToShareInfoResponse(sendRequest(operation, timeout, ShareStorageException.class));
     }

sdk/storage/azure-storage-file-share/src/main/java/com/azure/storage/file/share/implementation/models/SharePropertiesInternal.java

@@ -183,6 +183,12 @@ public final class SharePropertiesInternal implements XmlSerializable<SharePrope
     @Generated
     private DateTimeRfc1123 nextAllowedProvisionedBandwidthDowngradeTime;
 
+    /*
+     * The EnableSmbDirectoryLease property.
+     */
+    @Generated
+    private Boolean enableSmbDirectoryLease;
+
     /*
      * Dictionary of <string>
      */
@@ -819,6 +825,28 @@ public OffsetDateTime getNextAllowedProvisionedBandwidthDowngradeTime() {
         return this;
     }
 
+    /**
+     * Get the enableSmbDirectoryLease property: The EnableSmbDirectoryLease property.
+     * 
+     * @return the enableSmbDirectoryLease value.
+     */
+    @Generated
+    public Boolean isEnableSmbDirectoryLease() {
+        return this.enableSmbDirectoryLease;
+    }
+
+    /**
+     * Set the enableSmbDirectoryLease property: The EnableSmbDirectoryLease property.
+     * 
+     * @param enableSmbDirectoryLease the enableSmbDirectoryLease value to set.
+     * @return the SharePropertiesInternal object itself.
+     */
+    @Generated
+    public SharePropertiesInternal setEnableSmbDirectoryLease(Boolean enableSmbDirectoryLease) {
+        this.enableSmbDirectoryLease = enableSmbDirectoryLease;
+        return this;
+    }
+
     /**
      * Get the metadata property: Dictionary of &lt;string&gt;.
      * 
@@ -884,6 +912,7 @@ public XmlWriter toXml(XmlWriter xmlWriter, String rootElementName) throws XMLSt
             Objects.toString(this.nextAllowedProvisionedIopsDowngradeTime, null));
         xmlWriter.writeStringElement("NextAllowedProvisionedBandwidthDowngradeTime",
             Objects.toString(this.nextAllowedProvisionedBandwidthDowngradeTime, null));
+        xmlWriter.writeBooleanElement("EnableSmbDirectoryLease", this.enableSmbDirectoryLease);
         if (this.metadata != null) {
             xmlWriter.writeStartElement("Metadata");
             for (Map.Entry<String, String> entry : this.metadata.entrySet()) {
@@ -996,6 +1025,9 @@ public static SharePropertiesInternal fromXml(XmlReader xmlReader, String rootEl
                 } else if ("NextAllowedProvisionedBandwidthDowngradeTime".equals(elementName.getLocalPart())) {
                     deserializedSharePropertiesInternal.nextAllowedProvisionedBandwidthDowngradeTime
                         = reader.getNullableElement(DateTimeRfc1123::new);
+                } else if ("EnableSmbDirectoryLease".equals(elementName.getLocalPart())) {
+                    deserializedSharePropertiesInternal.enableSmbDirectoryLease
+                        = reader.getNullableElement(Boolean::parseBoolean);
                 } else if ("Metadata".equals(elementName.getLocalPart())) {
                     while (reader.nextElement() != XmlToken.END_ELEMENT) {
                         if (deserializedSharePropertiesInternal.metadata == null) {

sdk/storage/azure-storage-file-share/src/main/java/com/azure/storage/file/share/implementation/models/SharesGetPropertiesHeaders.java

@@ -106,6 +106,12 @@ public final class SharesGetPropertiesHeaders {
     @Generated
     private LeaseDurationType xMsLeaseDuration;
 
+    /*
+     * The x-ms-enable-smb-directory-lease property.
+     */
+    @Generated
+    private Boolean xMsEnableSmbDirectoryLease;
+
     /*
      * The x-ms-request-id property.
      */
@@ -223,6 +229,9 @@ public final class SharesGetPropertiesHeaders {
 
     private static final HttpHeaderName X_MS_LEASE_DURATION = HttpHeaderName.fromString("x-ms-lease-duration");
 
+    private static final HttpHeaderName X_MS_ENABLE_SMB_DIRECTORY_LEASE
+        = HttpHeaderName.fromString("x-ms-enable-smb-directory-lease");
+
     private static final HttpHeaderName X_MS_ACCESS_TIER_TRANSITION_STATE
         = HttpHeaderName.fromString("x-ms-access-tier-transition-state");
 
@@ -333,6 +342,12 @@ public SharesGetPropertiesHeaders(HttpHeaders rawHeaders) {
         } else {
             this.xMsLeaseDuration = null;
         }
+        String xMsEnableSmbDirectoryLease = rawHeaders.getValue(X_MS_ENABLE_SMB_DIRECTORY_LEASE);
+        if (xMsEnableSmbDirectoryLease != null) {
+            this.xMsEnableSmbDirectoryLease = Boolean.parseBoolean(xMsEnableSmbDirectoryLease);
+        } else {
+            this.xMsEnableSmbDirectoryLease = null;
+        }
         this.xMsRequestId = rawHeaders.getValue(HttpHeaderName.X_MS_REQUEST_ID);
         this.xMsAccessTierTransitionState = rawHeaders.getValue(X_MS_ACCESS_TIER_TRANSITION_STATE);
         String xMsShareNextAllowedQuotaDowngradeTime
@@ -734,6 +749,28 @@ public SharesGetPropertiesHeaders setXMsLeaseDuration(LeaseDurationType xMsLease
         return this;
     }
 
+    /**
+     * Get the xMsEnableSmbDirectoryLease property: The x-ms-enable-smb-directory-lease property.
+     * 
+     * @return the xMsEnableSmbDirectoryLease value.
+     */
+    @Generated
+    public Boolean isXMsEnableSmbDirectoryLease() {
+        return this.xMsEnableSmbDirectoryLease;
+    }
+
+    /**
+     * Set the xMsEnableSmbDirectoryLease property: The x-ms-enable-smb-directory-lease property.
+     * 
+     * @param xMsEnableSmbDirectoryLease the xMsEnableSmbDirectoryLease value to set.
+     * @return the SharesGetPropertiesHeaders object itself.
+     */
+    @Generated
+    public SharesGetPropertiesHeaders setXMsEnableSmbDirectoryLease(Boolean xMsEnableSmbDirectoryLease) {
+        this.xMsEnableSmbDirectoryLease = xMsEnableSmbDirectoryLease;
+        return this;
+    }
+
     /**
      * Get the xMsRequestId property: The x-ms-request-id property.
      * 

sdk/storage/azure-storage-file-share/src/main/java/com/azure/storage/file/share/implementation/util/ModelHelper.java

@@ -238,6 +238,7 @@ public static ShareProperties populateShareProperties(SharePropertiesInternal sh
             sharePropertiesInternal.getNextAllowedProvisionedIopsDowngradeTime());
         properties.setNextAllowedProvisionedBandwidthDowngradeTime(
             sharePropertiesInternal.getNextAllowedProvisionedBandwidthDowngradeTime());
+        properties.setEnableSmbDirectoryLease(sharePropertiesInternal.isEnableSmbDirectoryLease());
 
         return properties;
     }
@@ -517,7 +518,8 @@ public static Response<ShareInfo> mapToShareInfoResponse(Response<?> response) {
             .setMaxBurstCreditsForIops(headers.getXMsShareMaxBurstCreditsForIops())
             .setNextAllowedProvisionedIopsDowngradeTime(headers.getXMsShareNextAllowedProvisionedIopsDowngradeTime())
             .setNextAllowedProvisionedBandwidthDowngradeTime(
-                headers.getXMsShareNextAllowedProvisionedBandwidthDowngradeTime());
+                headers.getXMsShareNextAllowedProvisionedBandwidthDowngradeTime())
+            .setEnableSmbDirectoryLease(headers.isXMsEnableSmbDirectoryLease());
 
         return new SimpleResponse<>(response, shareProperties);
     }

sdk/storage/azure-storage-file-share/src/main/java/com/azure/storage/file/share/models/ShareProperties.java

@@ -162,6 +162,12 @@ public final class ShareProperties implements XmlSerializable<ShareProperties> {
      */
     private OffsetDateTime nextAllowedProvisionedBandwidthDowngradeTime;
 
+    /*
+     * Optional, default value is true. Only applicable to SMB shares.
+     * Specifies whether granting of new directory leases for directories present in a share are to be enabled or disabled.
+     */
+    private Boolean enableSmbDirectoryLease;
+
     /**
      * Creates a new instance of {@link ShareProperties}.
      */
@@ -796,6 +802,30 @@ public OffsetDateTime getNextAllowedProvisionedBandwidthDowngradeTime() {
         return this;
     }
 
+    /**
+     * Optional, default value is true. Only applicable to SMB shares.
+     * Specifies whether granting of new directory leases for directories present in a share are to be enabled or disabled.
+     * An input of true specifies that granting of new directory leases is to be allowed.
+     * An input of false specifies that granting of new directory leases is to be blocked.
+     * @return the enableSmbDirectoryLease value.
+     */
+    public Boolean isEnableSmbDirectoryLease() {
+        return enableSmbDirectoryLease;
+    }
+
+    /**
+     * Optional, default value is true. Only applicable to SMB shares.
+     * Specifies whether granting of new directory leases for directories present in a share are to be enabled or disabled.
+     * An input of true specifies that granting of new directory leases is to be allowed.
+     * An input of false specifies that granting of new directory leases is to be blocked.
+     * @param enableSmbDirectoryLease the enableSmbDirectoryLease value to set.
+     * @return the ShareProperties object itself.
+     */
+    public ShareProperties setEnableSmbDirectoryLease(Boolean enableSmbDirectoryLease) {
+        this.enableSmbDirectoryLease = enableSmbDirectoryLease;
+        return this;
+    }
+
     @Override
     public XmlWriter toXml(XmlWriter xmlWriter) throws XMLStreamException {
         return toXml(xmlWriter, null);

sdk/storage/azure-storage-file-share/src/main/java/com/azure/storage/file/share/options/ShareCreateOptions.java

@@ -26,6 +26,7 @@ public class ShareCreateOptions {
     private Long paidBurstingMaxBandwidthMibps;
     private Long provisionedMaxIops;
     private Long provisionedMaxBandwidthMibps;
+    private Boolean enableSmbDirectoryLease;
 
     /**
      * Creates a new instance of {@link ShareCreateOptions}.
@@ -277,4 +278,28 @@ public ShareCreateOptions setProvisionedMaxBandwidthMibps(Long provisionedMaxBan
         this.provisionedMaxBandwidthMibps = provisionedMaxBandwidthMibps;
         return this;
     }
+
+    /**
+     * Optional, default value is true. Only applicable to SMB shares.
+     * Specifies whether granting of new directory leases for directories present in a share are to be enabled or disabled.
+     * An input of true specifies that granting of new directory leases is to be allowed.
+     * An input of false specifies that granting of new directory leases is to be blocked.
+     * @return the enableSmbDirectoryLease value.
+     */
+    public Boolean isEnableSmbDirectoryLease() {
+        return enableSmbDirectoryLease;
+    }
+
+    /**
+     * Optional, default value is true. Only applicable to SMB shares.
+     * Specifies whether granting of new directory leases for directories present in a share are to be enabled or disabled.
+     * An input of true specifies that granting of new directory leases is to be allowed.
+     * An input of false specifies that granting of new directory leases is to be blocked.
+     * @param enableSmbDirectoryLease the enableSmbDirectoryLease value to set.
+     * @return the ShareCreateOptions object itself.
+     */
+    public ShareCreateOptions setEnableSmbDirectoryLease(Boolean enableSmbDirectoryLease) {
+        this.enableSmbDirectoryLease = enableSmbDirectoryLease;
+        return this;
+    }
 }

sdk/storage/azure-storage-file-share/src/main/java/com/azure/storage/file/share/options/ShareSetPropertiesOptions.java

@@ -23,6 +23,7 @@ public class ShareSetPropertiesOptions {
     private Long paidBurstingMaxBandwidthMibps;
     private Long provisionedMaxIops;
     private Long provisionedMaxBandwidthMibps;
+    private Boolean enableSmbDirectoryLease;
 
     /**
      * Creates a new instance of {@link ShareSetPropertiesOptions}.
@@ -256,4 +257,28 @@ public ShareSetPropertiesOptions setProvisionedMaxBandwidthMibps(Long provisione
         this.provisionedMaxBandwidthMibps = provisionedMaxBandwidthMibps;
         return this;
     }
+
+    /**
+     * Optional, default value is true. Only applicable to SMB shares.
+     * Specifies whether granting of new directory leases for directories present in a share are to be enabled or disabled.
+     * An input of true specifies that granting of new directory leases is to be allowed.
+     * An input of false specifies that granting of new directory leases is to be blocked.
+     * @return the enableSmbDirectoryLease value.
+     */
+    public Boolean isEnableSmbDirectoryLease() {
+        return enableSmbDirectoryLease;
+    }
+
+    /**
+     * Optional, default value is true. Only applicable to SMB shares.
+     * Specifies whether granting of new directory leases for directories present in a share are to be enabled or disabled.
+     * An input of true specifies that granting of new directory leases is to be allowed.
+     * An input of false specifies that granting of new directory leases is to be blocked.
+     * @param enableSmbDirectoryLease the enableSmbDirectoryLease value to set.
+     * @return the ShareSetPropertiesOptions object itself.
+     */
+    public ShareSetPropertiesOptions setEnableSmbDirectoryLease(Boolean enableSmbDirectoryLease) {
+        this.enableSmbDirectoryLease = enableSmbDirectoryLease;
+        return this;
+    }
 }