mitm-archive/server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Compare commits

base: mitm-archive:459d10664a33bc5942a1bc56e92a7fa79db69bff
Branches Tags
mitm-archive:main
..
compare: mitm-archive:27a8c3d1d1d534f8744e0397ff37e2cc25ec38c5
Branches Tags
mitm-archive:main

2 commits
459d10664a ... 27a8c3d1d1

Author SHA1 Message Date
DarkCat09
27a8c3d1d1
fix: convert ctx.Method() []byte to string 
see similar commit 10d174f626
 2024-07-02 19:14:30 +04:00
DarkCat09
bc6490187d
feat: 404 on ErrNoRows, 500 on err != nil 2024-07-02 19:13:22 +04:00
1 changed files with 6 additions and 2 deletions
Show stats Expand all files Collapse all files

8
server.go
View file

@ -58,7 +58,7 @@ func handler(ctx *fasthttp.RequestCtx) {
) )
row := stmt.QueryRow( row := stmt.QueryRow(
ctx.Method(), string(ctx.Method()),
fmt.Sprintf( fmt.Sprintf(
"%s://%s:%s%s", "%s://%s:%s%s",
scheme, scheme,
@ -72,9 +72,13 @@ func handler(ctx *fasthttp.RequestCtx) {
var code int var code int
err := row.Scan(&id, &code) err := row.Scan(&id, &code)
if err != nil { if err == sql.ErrNoRows {
ctx.Response.SetStatusCode(404) ctx.Response.SetStatusCode(404)
return return
} else if err != nil {
ctx.Response.SetStatusCode(500)
ctx.Response.SetBodyString("Unable to fetch row: " + err.Error())
return
} }
// -- set status code // -- set status code