From ffabe67e36537fc4758e00c69ce564837012421d Mon Sep 17 00:00:00 2001 From: Mike Crute Date: Mon, 15 Nov 2021 23:06:13 -0800 Subject: db: Add DeleteOneById --- db/mongodb/client.go | 14 ++++++++++++++ 1 file changed, 14 insertions(+) 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 +} -- cgit v1.2.3