recovering from panics while parsing bson documents (#16)

coroot · May 18, 2023 · 01c0260 · 01c0260
1 parent e10593c
commit 01c0260
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/tracing/mongo.go b/tracing/mongo.go
@@ -56,10 +56,13 @@ func parseMongo(payload []byte) string {
 	return bsonToString(reader)
 }
 
-func bsonToString(r io.Reader) string {
-	raw, err := bson.NewFromIOReader(r)
-	if err != nil {
-		return "<truncated>"
+func bsonToString(r io.Reader) (res string) {
+	res = "<truncated>"
+	defer func() {
+		recover()
+	}()
+	if raw, err := bson.NewFromIOReader(r); err == nil {
+		res = raw.String()
 	}
-	return raw.String()
+	return
 }