Collection+DocumentCollection: add createIndexes

doubleloop-io · Nov 11, 2024 · efa6c05 · efa6c05
1 parent a1c9463
commit efa6c05
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 0 deletions.
diff --git a/packages/effect-mongodb/src/Collection.ts b/packages/effect-mongodb/src/Collection.ts
@@ -10,10 +10,12 @@ import * as O from "effect/Option"
 import type {
   BulkWriteOptions,
   Collection as MongoCollection,
+  CreateIndexesOptions,
   DeleteOptions,
   DeleteResult,
   Document,
   FindOptions as MongoFindOptions,
+  IndexDescription,
   InsertManyResult,
   InsertOneOptions,
   InsertOneResult,
@@ -272,4 +274,29 @@ export const rename: {
     )
 )
 
+export const createIndexes: {
+  (
+    indexSpecs: Array<IndexDescription>,
+    options?: CreateIndexesOptions
+  ): <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>
+  ) => Effect.Effect<Array<string>, MongoError.MongoError, R>
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    indexSpecs: Array<IndexDescription>,
+    options?: CreateIndexesOptions
+  ): Effect.Effect<Array<string>, MongoError.MongoError, R>
+} = F.dual(
+  (args) => isCollection(args[0]),
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    indexSpecs: Array<IndexDescription>,
+    options?: CreateIndexesOptions
+  ): Effect.Effect<Array<string>, MongoError.MongoError, R> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.createIndexes(indexSpecs, options)),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<Array<string>>("createIndexes error"))
+    )
+)
+
 const isCollection = (x: unknown) => x instanceof Collection
diff --git a/packages/effect-mongodb/src/DocumentCollection.ts b/packages/effect-mongodb/src/DocumentCollection.ts
@@ -9,11 +9,13 @@ import * as O from "effect/Option"
 import type {
   BulkWriteOptions,
   Collection as MongoCollection,
+  CreateIndexesOptions,
   DeleteOptions,
   DeleteResult,
   Document,
   Filter,
   FindOptions,
+  IndexDescription,
   InsertManyResult,
   InsertOneOptions,
   InsertOneResult,
@@ -251,6 +253,31 @@ export const rename: {
     )
 )
 
+export const createIndexes: {
+  (
+    indexSpecs: Array<IndexDescription>,
+    options?: CreateIndexesOptions
+  ): (
+    collection: DocumentCollection
+  ) => Effect.Effect<Array<string>, MongoError.MongoError>
+  (
+    collection: DocumentCollection,
+    indexSpecs: Array<IndexDescription>,
+    options?: CreateIndexesOptions
+  ): Effect.Effect<Array<string>, MongoError.MongoError>
+} = F.dual(
+  (args) => isDocumentCollection(args[0]),
+  (
+    collection: DocumentCollection,
+    indexSpecs: Array<IndexDescription>,
+    options?: CreateIndexesOptions
+  ): Effect.Effect<Array<string>, MongoError.MongoError> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.createIndexes(indexSpecs, options)),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<Array<string>>("createIndexes error"))
+    )
+)
+
 export const typed: {
   <A extends Document, I extends Document = A, R = never>(
     schema: Schema.Schema<A, I, R>