Merge remote-tracking branch 'origin/main' into use-tsconfig-src-build

Author: deyaaeldeen

.github/CODEOWNERS

@@ -491,6 +491,9 @@
 # PRLabel: %Mgmt
 /sdk/databox/arm-databox/ @qiaozha @MaryGao
 
+# PRLabel: %Mgmt
+/sdk/dell/arm-dell-storage/ @qiaozha @MaryGao
+
 # PRLabel: %Mgmt
 /sdk/dashboard/arm-dashboard/ @qiaozha @MaryGao
 

common/config/rush/pnpm-lock.yaml

@@ -32,6 +32,7 @@ https://learn.microsoft.com/javascript/api/@azure/arm-portalservicescopilot?view
 https://learn.microsoft.com/javascript/api/@azure/arm-onlineexperimentation?view=azure-node-preview
 https://learn.microsoft.com/javascript/api/@azure-rest/analytics-onlineexperimentation?view=azure-node-preview
 https://learn.microsoft.com/javascript/api/@azure/arm-carbonoptimization?view=azure-node-preview
+https://learn.microsoft.com/javascript/api/@azure/arm-dell-storage?view=azure-node-preview
 https://learn.microsoft.com/javascript/api/@azure/arm-mongodbatlas?view=azure-node-preview
 https://learn.microsoft.com/javascript/api/@azure/arm-purestorageblock?view=azure-node-preview
 https://learn.microsoft.com/javascript/api/@azure/arm-kubernetesconfiguration-extensiontypes?view=azure-node-preview
@@ -44,3 +45,4 @@ https://learn.microsoft.com/javascript/api/@azure/arm-playwright?view=azure-node
 https://learn.microsoft.com/javascript/api/@azure/arm-cloudhealth?view=azure-node-preview
 https://learn.microsoft.com/javascript/api/@azure/arm-resourcesbicep?view=azure-node-preview
 https://learn.microsoft.com/javascript/api/@azure/arm-resourcesdeployments?view=azure-node-preview
+

eng/ignore-links.txt

@@ -1,7 +1,7 @@
 /**
  * This is the main configuration file for Rush.
  * For full documentation, please see https://rushjs.io
- */ {
+ */{
   "$schema": "https://developer.microsoft.com/json-schemas/rush/v5/rush.schema.json",
   /**
    * (Required) This specifies the version of the Rush engine to be used in this repo.
@@ -2380,6 +2380,11 @@
       "projectFolder": "sdk/carbonoptimization/arm-carbonoptimization",
       "versionPolicyName": "management"
     },
+    {
+      "packageName": "@azure/arm-dell-storage",
+      "projectFolder": "sdk/dell/arm-dell-storage",
+      "versionPolicyName": "management"
+    },
     {
       "packageName": "@azure/arm-mongodbatlas",
       "projectFolder": "sdk/mongodbatlas/arm-mongodbatlas",
@@ -2446,4 +2451,4 @@
       "versionPolicyName": "management"
     }
   ]
-}
+}

rush.json

@@ -161,6 +161,11 @@ export class CosmosClient {
       optionsOrConnectionString.connectionPolicy.enablePartitionLevelFailover ||
       optionsOrConnectionString.connectionPolicy.enablePartitionLevelCircuitBreaker
     ) {
+      if (!optionsOrConnectionString.connectionPolicy.enableEndpointDiscovery) {
+        throw new Error(
+          "enableEndpointDiscovery must be set to true to use partition level failover or circuit breaker.",
+        );
+      }
       this.globalPartitionEndpointManager = new GlobalPartitionEndpointManager(
         optionsOrConnectionString,
         globalEndpointManager,

sdk/cosmosdb/cosmos/src/CosmosClient.ts

@@ -13,6 +13,7 @@ import { DiagnosticNodeInternal } from "../../diagnostics/DiagnosticNodeInternal
 import { getEmptyCosmosDiagnostics, withDiagnostics } from "../../utils/diagnostics.js";
 import { ChangeFeedMode } from "./ChangeFeedMode.js";
 import { decryptChangeFeedResponse } from "./changeFeedUtils.js";
+import { computePartitionKeyRangeId } from "../ClientUtils.js";
 /**
  * @hidden
  * Provides iterator for change feed for one partition key.
@@ -187,6 +188,14 @@ export class ChangeFeedForPartitionKey<T> implements ChangeFeedPullModelIterator
       feedOptions.containerRid = this.container._rid;
     }
     try {
+      const isPartitionLevelFailOverEnabled = this.clientContext.isPartitionLevelFailOverEnabled();
+      const partitionKeyRangeId = await computePartitionKeyRangeId(
+        diagnosticNode,
+        convertToInternalPartitionKey(this.partitionKey),
+        this.clientContext.partitionKeyRangeCache,
+        isPartitionLevelFailOverEnabled,
+        this.container,
+      );
       const response: Response<Array<T & Resource>> = await (this.clientContext.queryFeed<T>({
         path: this.resourceLink,
         resourceType: ResourceType.item,
@@ -196,6 +205,7 @@ export class ChangeFeedForPartitionKey<T> implements ChangeFeedPullModelIterator
         query: undefined,
         options: feedOptions,
         partitionKey: this.partitionKey,
+        partitionKeyRangeId,
       }) as Promise<any>);
       return new ChangeFeedIteratorResponse(
         response.result,

sdk/cosmosdb/cosmos/src/client/ChangeFeed/ChangeFeedForPartitionKey.ts

@@ -233,6 +233,16 @@ export class Items {
   ): Promise<QueryIterator<ItemDefinition>> {
     const encryptionSqlQuerySpec = queryBuilder.toEncryptionSqlQuerySpec();
     const sqlQuerySpec = await this.buildSqlQuerySpec(encryptionSqlQuerySpec);
+    if (this.clientContext.enableEncryption && options.partitionKey) {
+      await this.container.checkAndInitializeEncryption();
+      const { partitionKeyList, encryptedCount } =
+        await this.container.encryptionProcessor.getEncryptedPartitionKeyValue([
+          options.partitionKey,
+        ] as PartitionKeyInternal);
+      if (encryptedCount > 0) {
+        options.partitionKey = partitionKeyList[0];
+      }
+    }
     const iterator = this.query<ItemDefinition>(sqlQuerySpec, options);
     return iterator;
   }

sdk/cosmosdb/cosmos/src/client/Item/Items.ts

@@ -5,6 +5,7 @@ import { randomBytes } from "node:crypto";
 import { ProtectedDataEncryptionKey } from "../EncryptionKey/ProtectedDataEncryptionKey.js";
 import type { KeyEncryptionKey } from "../KeyEncryptionKey.js";
 import { Constants } from "../../common/index.js";
+import { startBackgroundTask } from "../../utils/time.js";
 
 /**
  * The cache used to store the protected data encryption key.
@@ -37,7 +38,7 @@ export class ProtectedDataEncryptionKeyCache {
   }
 
   private async clearCacheOnTtlExpiry(): Promise<void> {
-    this.cacheRefresher = setInterval(() => {
+    this.cacheRefresher = startBackgroundTask(async () => {
       const now = new Date();
       for (const key of this.cache.keys()) {
         if (now.getTime() - this.cache.get(key)[0].getTime() > this.cacheTimeToLive) {