suppoert speedtests

netwatcherio · Jul 13, 2024 · 57bb304 · 57bb304
1 parent 0d0d8ac
commit 57bb304
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/internal/agent/probe.go b/internal/agent/probe.go
@@ -336,11 +336,17 @@ func (probe *Probe) GetAll(db *mongo.Database) ([]*Probe, error) {
 func (probe *Probe) UpdateFirstProbeTarget(db *mongo.Database, targetStatus string) error {
 	var filter = bson.D{{"_id", probe.ID}}
 
+	get, err := probe.Get(db)
+	if err != nil {
+		return err
+	}
+	get[0].Config.Target[0].Target = targetStatus
+
 	update := bson.D{
-		{"$set", probe},
+		{"$set", get},
 	}
 
-	_, err := db.Collection("probes").UpdateOne(context.TODO(), filter, update)
+	_, err = db.Collection("probes").UpdateOne(context.TODO(), filter, update)
 	if err != nil {
 		return err
 	}