db: add FindAllByFilter

1 files changed, 6 insertions, 2 deletions

diff --git a/db/mongodb/client.go b/db/mongodb/client.go
index 8180991..963860e 100644
--- a/db/mongodb/client.go
+++ b/db/mongodb/client.go
@@ -58,8 +58,8 @@ func (m *Mongo) Collection(name string) *mongo.Collection {
         return m.db.Collection(name)
 }
 
-func (m *Mongo) FindAll(ctx context.Context, cn string, out interface{}) error {
-        res, err := m.db.Collection(cn).Find(ctx, bson.D{})
+func (m *Mongo) FindAllByFilter(ctx context.Context, cn string, filter interface{}, out interface{}) error {
+        res, err := m.db.Collection(cn).Find(ctx, filter)
         if err != nil {
                 return err
         }
@@ -71,6 +71,10 @@ func (m *Mongo) FindAll(ctx context.Context, cn string, out interface{}) error {
         return nil
 }
 
+func (m *Mongo) FindAll(ctx context.Context, cn string, out interface{}) error {
+        return m.FindAllByFilter(ctx, cn, bson.D{}, out)
+}
+
 func (m *Mongo) FindOneById(ctx context.Context, cn string, id string, out interface{}) error {
         if err := m.db.Collection(cn).FindOne(ctx, bson.M{"_id": id}).Decode(out); err != nil {
                 return err