fix(otelgorm): ignore gorm.ErrRecordNotFound and other errors from ot…

…elsql/otel.go (#48) Co-authored-by: jixiufeng <jixiufeng@luojilab.com>
uptrace · Apr 6, 2022 · 1c5d1f7 · 1c5d1f7
1 parent 01faa43
commit 1c5d1f7
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/otelgorm/otelgorm.go b/otelgorm/otelgorm.go
@@ -2,7 +2,9 @@ package otelgorm
 
 import (
 	"database/sql"
+	"database/sql/driver"
 	"fmt"
+	"io"
 
 	"go.opentelemetry.io/otel"
 	"go.opentelemetry.io/otel/attribute"
@@ -134,7 +136,14 @@ func (p *otelPlugin) after() gormHookFunc {
 		}
 
 		span.SetAttributes(attrs...)
-		if tx.Error != nil {
+		switch tx.Error {
+		case nil,
+			gorm.ErrRecordNotFound,
+			driver.ErrSkip,
+			io.EOF, // end of rows iterator
+			sql.ErrNoRows:
+			// ignore
+		default:
 			span.RecordError(tx.Error)
 			span.SetStatus(codes.Error, tx.Error.Error())
 		}