db/mongodb: add filter methodsdb/mongodb/v0.3.0

1 files changed, 18 insertions, 6 deletions

diff --git a/db/mongodb/client.go b/db/mongodb/client.go
index e6e9f0b..705b151 100644
--- a/db/mongodb/client.go
+++ b/db/mongodb/client.go
@@ -75,14 +75,18 @@ 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 {
+func (m *Mongo) FindOneByFilter(ctx context.Context, cn string, filter interface{}, out interface{}) error {
+        if err := m.db.Collection(cn).FindOne(ctx, filter).Decode(out); err != nil {
                 return err
         }
 
         return nil
 }
 
+func (m *Mongo) FindOneById(ctx context.Context, cn string, id string, out interface{}) error {
+        return m.FindOneByFilter(ctx, cn, bson.M{"_id": id}, out)
+}
+
 func (m *Mongo) InsertOne(ctx context.Context, cn string, in interface{}) error {
         _, err := m.db.Collection(cn).InsertOne(ctx, in)
         if err != nil {
@@ -92,18 +96,22 @@ func (m *Mongo) InsertOne(ctx context.Context, cn string, in interface{}) error
         return nil
 }
 
-func (m *Mongo) ReplaceOneById(ctx context.Context, cn string, id string, in interface{}) error {
+func (m *Mongo) ReplaceOneByFilter(ctx context.Context, cn string, filter interface{}, in interface{}) error {
         tp := true
         opts := &options.ReplaceOptions{Upsert: &tp}
-        if _, err := m.db.Collection(cn).ReplaceOne(ctx, bson.M{"_id": id}, in, opts); err != nil {
+        if _, err := m.db.Collection(cn).ReplaceOne(ctx, filter, in, opts); err != nil {
                 return err
         }
 
         return nil
 }
 
-func (m *Mongo) DeleteOneById(ctx context.Context, cn string, id string) error {
-        dr, err := m.db.Collection(cn).DeleteOne(ctx, bson.M{"_id": id})
+func (m *Mongo) ReplaceOneById(ctx context.Context, cn string, id string, in interface{}) error {
+        return m.ReplaceOneByFilter(ctx, cn, bson.M{"_id": id}, in)
+}
+
+func (m *Mongo) DeleteOneByFilter(ctx context.Context, cn string, filter interface{}) error {
+        dr, err := m.db.Collection(cn).DeleteOne(ctx, filter)
         if err != nil {
                 return err
         }
@@ -114,3 +122,7 @@ func (m *Mongo) DeleteOneById(ctx context.Context, cn string, id string) error {
 
         return nil
 }
+
+func (m *Mongo) DeleteOneById(ctx context.Context, cn string, id string) error {
+        return m.DeleteOneByFilter(ctx, cn, bson.M{"_id": id})
+}