feat(service-spec-types): index for scrutinizable resources

cdklabs · Nov 3, 2023 · 49fe58c · 49fe58c
1 parent 79dac13
commit 49fe58c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/packages/@aws-cdk/service-spec-types/src/types/database.ts b/packages/@aws-cdk/service-spec-types/src/types/database.ts
@@ -1,6 +1,6 @@
 import { promises as fs } from 'fs';
 import { gunzipSync } from 'zlib';
-import { Database, entityCollection, fieldIndex, stringCmp } from '@cdklabs/tskb';
+import { Database, entityCollection, fieldIndex, fieldIndexWithDefault, stringCmp } from '@cdklabs/tskb';
 import { IsAugmentedResource, ResourceAugmentation } from './augmentations';
 import {
   DimensionSet,
@@ -21,13 +21,15 @@ import {
   RegionHasResource,
   RegionHasService,
   UsesType,
+  ResourceScrutinyType,
 } from './resource';
 
 export function emptyDatabase() {
   return new Database(
     {
       resource: entityCollection<Resource>().index({
         cloudFormationType: fieldIndex('cloudFormationType', stringCmp),
+        scrutinizable: fieldIndexWithDefault('scrutinizable', stringCmp, ResourceScrutinyType.None),
       }),
       region: entityCollection<Region>().index({
         name: fieldIndex('name', stringCmp),

diff --git a/packages/@cdklabs/tskb/src/entity.ts b/packages/@cdklabs/tskb/src/entity.ts
@@ -127,6 +127,14 @@ export function fieldIndex<A extends Entity, P extends keyof A>(
   return calculatedIndex((x) => x[propName], comparator);
 }
 
+export function fieldIndexWithDefault<A extends Entity, P extends keyof A>(
+  propName: P,
+  comparator: sortedMap.Comparator<NonNullable<A[P]>>,
+  defaultValue: NonNullable<A[P]>,
+): EntityIndex<A, A[P]> {
+  return calculatedIndex((x) => x[propName] ?? defaultValue!, comparator);
+}
+
 /**
  * An index that is calculated based on a function applied to an entity
  */