db: Add DeleteOneById

1 files changed, 14 insertions, 0 deletions

diff --git a/db/mongodb/client.go b/db/mongodb/client.go
index 98369b9..82082fc 100644
--- a/db/mongodb/client.go
+++ b/db/mongodb/client.go
@@ -2,6 +2,7 @@ package mongodb
 
 import (
         "context"
+        "fmt"
         "net/url"
 
         "go.mongodb.org/mongo-driver/bson"
@@ -88,3 +89,16 @@ func (m *Mongo) ReplaceOneById(ctx context.Context, cn string, id string, in int
 
         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})
+        if err != nil {
+                return err
+        }
+
+        if dr.DeletedCount != 1 {
+                return fmt.Errorf("Invalid deletion record count %d not 1", dr.DeletedCount)
+        }
+
+        return nil
+}