fix: read id from data object

webiny · Oct 3, 2023 · acd3e6b · acd3e6b
1 parent f1bd2df
commit acd3e6b
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/packages/api-prerendering-service-so-ddb/src/operations/tenant.ts b/packages/api-prerendering-service-so-ddb/src/operations/tenant.ts
@@ -1,19 +1,22 @@
 import { PrerenderingServiceTenantStorageOperations } from "@webiny/api-prerendering-service/types";
 import { Entity } from "dynamodb-toolbox";
 import { queryAll } from "@webiny/db-dynamodb/utils/query";
-import { cleanupItems } from "@webiny/db-dynamodb/utils/cleanup";
 
 export interface CreateTenantStorageOperationsParams {
     entity: Entity<any>;
 }
 
+interface Tenant {
+    data: { id: string };
+}
+
 export const createTenantStorageOperations = (
     params: CreateTenantStorageOperationsParams
 ): PrerenderingServiceTenantStorageOperations => {
     const { entity } = params;
 
     const getTenantIds = async (): Promise<string[]> => {
-        const tenants = await queryAll<{ id: string }>({
+        const tenants = await queryAll<Tenant>({
             entity,
             partitionKey: "TENANTS",
             options: {
@@ -22,10 +25,8 @@ export const createTenantStorageOperations = (
             }
         });
 
-        return cleanupItems(entity, tenants).map(tenant => tenant.id);
+        return tenants.map(tenant => tenant.data.id);
     };
 
-    return {
-        getTenantIds
-    };
+    return { getTenantIds };
 };