From 36f695c146515ce1de90d640f9249f8c06e497b1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ask=20Bj=C3=B8rn=20Hansen?= <ask@develooper.com>
Date: Thu, 14 Dec 2023 23:15:09 -0800
Subject: [PATCH] trace tweaks

---
 server/graph_image.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/server/graph_image.go b/server/graph_image.go
index d8022dc..4c5d855 100644
--- a/server/graph_image.go
+++ b/server/graph_image.go
@@ -34,6 +34,8 @@ func (srv *Server) graphImage(c echo.Context) error {
 	log = log.With("serverID", serverID).With("type", imageType)
 	log.Info("graph parameters")
 
+	span.SetAttributes(attribute.String("url.server_parameter", serverID))
+
 	if imageType != "offset.png" {
 		return c.String(http.StatusNotFound, "invalid image name")
 	}
@@ -47,6 +49,10 @@ func (srv *Server) graphImage(c echo.Context) error {
 	}
 
 	serverData, err := srv.FindServer(ctx, serverID)
+	if err != nil {
+		span.RecordError(err)
+		return c.String(http.StatusInternalServerError, "server error")
+	}
 	if serverData.ID == 0 {
 		return c.String(http.StatusNotFound, "not found")
 	}