1 files changed, 12 insertions, 2 deletions

diff --git a/src/mongo.go b/src/mongo.go
index 3a23131..764f877 100644
--- a/src/mongo.go
+++ b/src/mongo.go
@@ -78,6 +78,13 @@ func (mc *mongoClient) Finddoc(filter bson.M) ([]bson.M, error) {
         return results, err
 }
 
+func (mc *mongoClient) FindOneAndDelete(filter bson.M) (bson.M, error) {
+        res := mc.col.FindOneAndDelete(context.Background(), filter)
+        var result bson.M
+        err := res.Decode(&result)
+        return result, err
+}
+
 func (mc *mongoClient) Drop() error {
         return mc.col.Drop(context.Background())
 }
@@ -94,11 +101,14 @@ func (mc *mongoClient) Disconnect() error {
         return nil
 }
 
-func (mc *mongoClient) GetLock() {
-        for i := 0; i < 20000; {
+func (mc *mongoClient) GetLock() bool {
+        for i := 0; i < 200000; {
                 if !mc.Mutex.TryLock() {
                         i++
+                } else {
+                        return true
                 }
         }
         fmt.Printf("Die...\n")
+        return false
 }