DocumentCollection+Collection: add createIndex

Author: devmatteini

.changeset/dull-icons-nail.md

@@ -0,0 +1,5 @@
+---
+"effect-mongodb": patch
+---
+
+Add createIndex function in Collection and DocumentCollection

packages/effect-mongodb/src/Collection.ts

@@ -18,6 +18,7 @@ import type {
   DropCollectionOptions,
   FindOptions as MongoFindOptions,
   IndexDescription,
+  IndexSpecification,
   InsertManyResult,
   InsertOneOptions,
   InsertOneResult,
@@ -327,6 +328,31 @@ export const createIndexes: {
     )
 )
 
+export const createIndex: {
+  (
+    indexSpec: IndexSpecification,
+    options?: CreateIndexesOptions
+  ): <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>
+  ) => Effect.Effect<string, MongoError.MongoError, R>
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    indexSpec: IndexSpecification,
+    options?: CreateIndexesOptions
+  ): Effect.Effect<string, MongoError.MongoError, R>
+} = F.dual(
+  (args) => isCollection(args[0]),
+  <A extends Document, I extends Document, R>(
+    collection: Collection<A, I, R>,
+    indexSpec: IndexSpecification,
+    options?: CreateIndexesOptions
+  ): Effect.Effect<string, MongoError.MongoError, R> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.createIndex(indexSpec, options)),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<string>("createIndex error"))
+    )
+)
+
 export const aggregate: {
   <B extends Document, BI extends Document, BR>(
     pipeline: Array<Document>,

packages/effect-mongodb/src/DocumentCollection.ts

@@ -18,6 +18,7 @@ import type {
   Filter,
   FindOptions,
   IndexDescription,
+  IndexSpecification,
   InsertManyResult,
   InsertOneOptions,
   InsertOneResult,
@@ -300,6 +301,29 @@ export const createIndexes: {
     )
 )
 
+export const createIndex: {
+  (
+    indexSpec: IndexSpecification,
+    options?: CreateIndexesOptions
+  ): (collection: DocumentCollection) => Effect.Effect<string, MongoError.MongoError>
+  (
+    collection: DocumentCollection,
+    indexSpec: IndexSpecification,
+    options?: CreateIndexesOptions
+  ): Effect.Effect<string, MongoError.MongoError>
+} = F.dual(
+  (args) => isDocumentCollection(args[0]),
+  (
+    collection: DocumentCollection,
+    indexSpec: IndexSpecification,
+    options?: CreateIndexesOptions
+  ): Effect.Effect<string, MongoError.MongoError> =>
+    F.pipe(
+      Effect.promise(() => collection.collection.createIndex(indexSpec, options)),
+      Effect.catchAllDefect(MongoError.mongoErrorDie<string>("createIndex error"))
+    )
+)
+
 export const aggregate: {
   (
     pipeline?: Array<Document>,