updates

osmosis-labs · May 30, 2024 · c14cd47 · c14cd47
1 parent 436de0d
commit c14cd47
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/baseapp/grpcserver.go b/baseapp/grpcserver.go
@@ -7,6 +7,7 @@ import (
 
 	"go.opentelemetry.io/otel"
 	"go.opentelemetry.io/otel/attribute"
+	otelcodes "go.opentelemetry.io/otel/codes"
 	"go.opentelemetry.io/otel/propagation"
 	"go.opentelemetry.io/otel/trace"
 
@@ -90,7 +91,15 @@ func (app *BaseApp) RegisterGRPCServer(server gogogrpc.Server, logQueries bool)
 
 		span.SetAttributes(attribute.String("http.method", grpcInfo.FullMethod))
 
-		return handler(grpcCtx, req)
+		resp, err = handler(grpcCtx, req)
+
+		if err != nil {
+			span.SetStatus(otelcodes.Error, err.Error())
+		} else {
+			span.SetStatus(otelcodes.Ok, "OK")
+		}
+
+		return resp, err
 	}
 
 	// Loop through all services and methods, add the interceptor, and register