1 files changed, 30 insertions, 6 deletions

diff --git a/src/mongo.go b/listener/mongo.go
index 1d9f74f..a51350e 100644
--- a/src/mongo.go
+++ b/listener/mongo.go
@@ -2,6 +2,8 @@ package main
 
 import (
         "context"
+        "fmt"
+        "reflect"
         "time"
 
         "go.mongodb.org/mongo-driver/bson"
@@ -62,16 +64,38 @@ func (mc *mongoClient) UpdateMany(filter, update interface{}) error {
         return err
 }
 
-func (mc *mongoClient) Finddoc(filter bson.M) ([]bson.M, error) {
-        cur, err := mc.col.Find(context.Background(), filter)
+func (mc *mongoClient) ReplaceOne(filter, new interface{}) error {
+        _, err := mc.col.ReplaceOne(context.Background(), filter, new)
+        return err
+}
+
+func (mc *mongoClient) Finddoc(filter bson.M, results interface{}) error {
+        sliceValue := reflect.ValueOf(results)
+
+        if sliceValue.Kind() != reflect.Ptr || sliceValue.Elem().Kind() != reflect.Slice {
+                return fmt.Errorf("Error: result argument must be pointer to slice")
+        }
+        cur, err := mc.col.Find(context.TODO(), filter)
         if err != nil {
-                return nil, err
+                return err
         }
+        defer cur.Close(context.TODO())
+
+        elemType := sliceValue.Elem().Type().Elem()
 
-        var results []bson.M
-        err = cur.All(context.Background(), &results)
+        sliceValue = sliceValue.Elem()
 
-        return results, err
+        for cur.Next(context.TODO()) {
+                elem := reflect.New(elemType).Interface()
+                err := cur.Decode(elem)
+                if err != nil {
+                        return err
+                }
+                sliceValue = reflect.Append(sliceValue, reflect.ValueOf(elem).Elem())
+        }
+
+        reflect.ValueOf(results).Elem().Set(sliceValue)
+        return nil
 }
 
 func (mc *mongoClient) FindOneAndDelete(filter bson.M) (bson.M, error) {