index_view: add DropOneWithKey function (#161)

sv-tools · Sep 24, 2024 · f74dc72 · f74dc72
1 parent 580c703
commit f74dc72
Show file tree

Hide file tree

Showing 3 changed files with 107 additions and 0 deletions.
diff --git a/index_view.go b/index_view.go
@@ -15,6 +15,11 @@ type IndexView interface {
 	CreateOne(ctx context.Context, model mongo.IndexModel, opts ...*options.CreateIndexesOptions) (string, error)
 	DropAll(ctx context.Context, opts ...*options.DropIndexesOptions) (bson.Raw, error)
 	DropOne(ctx context.Context, name string, opts ...*options.DropIndexesOptions) (bson.Raw, error)
+	DropOneWithKey(
+		ctx context.Context,
+		keySpecDocument interface{},
+		opts ...*options.DropIndexesOptions,
+	) (bson.Raw, error)
 	List(ctx context.Context, opts ...*options.ListIndexesOptions) (Cursor, error)
 	ListSpecifications(ctx context.Context, opts ...*options.ListIndexesOptions) ([]*mongo.IndexSpecification, error)
 }
@@ -47,6 +52,14 @@ func (i *indexView) DropOne(ctx context.Context, name string, opts ...*options.D
 	return i.iv.DropOne(ctx, name, opts...)
 }
 
+func (i *indexView) DropOneWithKey(
+	ctx context.Context,
+	keySpecDocument interface{},
+	opts ...*options.DropIndexesOptions,
+) (bson.Raw, error) {
+	return i.iv.DropOneWithKey(ctx, keySpecDocument, opts...)
+}
+
 func (i *indexView) List(ctx context.Context, opts ...*options.ListIndexesOptions) (Cursor, error) {
 	cr, err := i.iv.List(ctx, opts...)
 	if err != nil {

diff --git a/mocks/gomock/mocks.go b/mocks/gomock/mocks.go
diff --git a/mocks/mockery/index_view.go b/mocks/mockery/index_view.go