Add deleteAllItemsForPartitionKey samples and add script for generating sample (#36315)

### Packages impacted by this PR
@azure/cosmos

### Issues associated with this PR


### Describe the problem that is addressed by this PR
This PR adds sample for deleteAllItemsForPartitionKey API and add a
script in `package.json` to generate samples.

### What are the possible designs available to address the problem? If
there are more than one possible design, why was the one in this PR
chosen?


### Are there test cases added in this PR? _(If not, why?)_


### Provide a list of related PRs _(if any)_


### Command used to generate this PR:**_(Applicable only to SDK release
request PRs)_

### Checklists
- [ ] Added impacted package name to the issue description
- [ ] Does this PR needs any fixes in the SDK Generator?** _(If so,
create an Issue in the
[Autorest/typescript](https://github.com/Azure/autorest.typescript)
repository and link it here)_
- [ ] Added a changelog (if necessary)

---------

Co-authored-by: Aditishree . <[email protected]>
Co-authored-by: Copilot <[email protected]>

Author: 3 people

sdk/cosmosdb/cosmos/package.json

@@ -54,7 +54,8 @@
     "test:node": "npm run test:node:unit",
     "test:node:integration": "cross-env NODE_OPTIONS='--dns-result-order=ipv4first' dev-tool run test:vitest --no-test-proxy -- -c vitest.int.config.ts",
     "test:node:unit": "cross-env NODE_OPTIONS='--dns-result-order=ipv4first' dev-tool run test:vitest --no-test-proxy -- -c vitest.unit.config.ts",
-    "update-snippets": "dev-tool run update-snippets"
+    "update-snippets": "dev-tool run update-snippets",
+    "generate-samples": "npm run build && npx dev-tool samples publish"
   },
   "repository": "github:Azure/azure-sdk-for-js",
   "license": "MIT",

sdk/cosmosdb/cosmos/samples-dev/ContainerManagement.ts

@@ -209,6 +209,40 @@ async function run(): Promise<void> {
   });
   console.log("Container with full text search policy created");
 
+  logStep("Delete all items for a specific partition key");
+
+  // Create a container with partition key on 'state' for the deleteAllItemsForPartitionKey demo
+  const { container: containerForDeletion } = await database.containers.createIfNotExists({
+    id: "ContainerForDeletion",
+    partitionKey: { paths: ["/state"] },
+  });
+
+  // Create some sample items with different partition key values
+  const cities = [
+    { id: "1", name: "Olympia", state: "WA", isCapital: true },
+    { id: "2", name: "Redmond", state: "WA", isCapital: false },
+    { id: "3", name: "Seattle", state: "WA", isCapital: false },
+    { id: "4", name: "Springfield", state: "IL", isCapital: true },
+    { id: "5", name: "Chicago", state: "IL", isCapital: false },
+  ];
+
+  console.log("Creating sample cities...");
+  for (const city of cities) {
+    await containerForDeletion.items.create(city);
+  }
+
+  // Delete all items for partition key 'WA'
+  await containerForDeletion.deleteAllItemsForPartitionKey("WA");
+  console.log("Deleted all items for partition key 'WA'");
+
+  // Query to verify items after deletion
+  const queryToVerify = "SELECT c.id, c.name, c.state FROM c WHERE c.state = 'WA'";
+  const { resources: waItems } = await containerForDeletion.items.query(queryToVerify).fetchAll();
+  console.log(`Items in WA after deletion: ${waItems.length}`);
+
+  // Clean up the container
+  await containerForDeletion.delete();
+
   await finish();
 }
 

sdk/cosmosdb/cosmos/samples/v4/javascript/ContainerManagement.js

@@ -208,6 +208,40 @@ async function run() {
   });
   console.log("Container with full text search policy created");
 
+  logStep("Delete all items for a specific partition key");
+
+  // Create a container with partition key on 'state' for the deleteAllItemsForPartitionKey demo
+  const { container: containerForDeletion } = await database.containers.createIfNotExists({
+    id: "ContainerForDeletion",
+    partitionKey: { paths: ["/state"] },
+  });
+
+  // Create some sample items with different partition key values
+  const cities = [
+    { id: "1", name: "Olympia", state: "WA", isCapital: true },
+    { id: "2", name: "Redmond", state: "WA", isCapital: false },
+    { id: "3", name: "Seattle", state: "WA", isCapital: false },
+    { id: "4", name: "Springfield", state: "IL", isCapital: true },
+    { id: "5", name: "Chicago", state: "IL", isCapital: false },
+  ];
+
+  console.log("Creating sample cities...");
+  for (const city of cities) {
+    await containerForDeletion.items.create(city);
+  }
+
+  // Delete all items for partition key 'WA'
+  await containerForDeletion.deleteAllItemsForPartitionKey("WA");
+  console.log("Deleted all items for partition key 'WA'");
+
+  // Query to verify items after deletion
+  const queryToVerify = "SELECT c.id, c.name, c.state FROM c WHERE c.state = 'WA'";
+  const { resources: waItems } = await containerForDeletion.items.query(queryToVerify).fetchAll();
+  console.log(`Items in WA after deletion: ${waItems.length}`);
+
+  // Clean up the container
+  await containerForDeletion.delete();
+
   await finish();
 }
 

sdk/cosmosdb/cosmos/samples/v4/typescript/src/ContainerManagement.ts

@@ -209,6 +209,42 @@ async function run(): Promise<void> {
   });
   console.log("Container with full text search policy created");
 
+  logStep("Delete all items for a specific partition key");
+
+  // Create a container with partition key on 'state' for the deleteAllItemsForPartitionKey demo
+  const { container: containerForDeletion } = await database.containers.createIfNotExists({
+    id: "ContainerForDeletion",
+    partitionKey: { paths: ["/state"] },
+  });
+
+  // Create some sample items with different partition key values
+  const cities = [
+    { id: "1", name: "Olympia", state: "WA", isCapital: true },
+    { id: "2", name: "Redmond", state: "WA", isCapital: false },
+    { id: "3", name: "Seattle", state: "WA", isCapital: false },
+    { id: "4", name: "Springfield", state: "IL", isCapital: true },
+    { id: "5", name: "Chicago", state: "IL", isCapital: false },
+  ];
+
+  console.log("Creating sample cities...");
+  for (const city of cities) {
+    await containerForDeletion.items.create(city);
+  }
+
+  // Delete all items for partition key 'WA'
+  await containerForDeletion.deleteAllItemsForPartitionKey("WA");
+  console.log("Deleted all items for partition key 'WA'");
+
+  // Query to verify items after deletion
+  const queryToVerify = "SELECT c.id, c.name, c.state FROM c WHERE c.state = 'WA'";
+  const { resources: waItems } = await containerForDeletion.items
+    .query(queryToVerify)
+    .fetchAll();
+  console.log(`Items in WA after deletion: ${waItems.length}`);
+
+  // Clean up the container
+  await containerForDeletion.delete();
+
   await finish();
 }