From bc6490187dd628d163b7b76b77a452b0f0c1ca36 Mon Sep 17 00:00:00 2001
From: DarkCat09 <gh@dc09.ru>
Date: Tue, 2 Jul 2024 19:13:22 +0400
Subject: [PATCH] feat: 404 on ErrNoRows, 500 on err != nil

---
 server.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/server.go b/server.go
index 3a94d40..fd53a9d 100644
--- a/server.go
+++ b/server.go
@@ -72,9 +72,13 @@ func handler(ctx *fasthttp.RequestCtx) {
 	var code int
 
 	err := row.Scan(&id, &code)
-	if err != nil {
+	if err == sql.ErrNoRows {
 		ctx.Response.SetStatusCode(404)
 		return
+	} else if err != nil {
+		ctx.Response.SetStatusCode(500)
+		ctx.Response.SetBodyString("Unable to fetch row: " + err.Error())
+		return
 	}
 
 	// -- set status code