add 404 page logic if the URL is not found

faisaltheparttimecoder · faisaltheparttimecoder · commit 6ba1db9e3d69 · 2020-08-20T14:59:43.000+01:00
diff --git a/server.go b/server.go
@@ -5,6 +5,7 @@ import (
 	"github.com/labstack/echo/v4/middleware"
 	"html/template"
 	"io"
+	"net/http"
 )
 
 type Template struct {
@@ -47,6 +48,11 @@ func startWebServer() {
 		}
 	}
 
+	// Render 404 page
+	echo.NotFoundHandler = func(c echo.Context) error {
+		return c.Render(http.StatusNotFound, "404.html", "")
+	}
+
 	// Serve the routes
 	webRouter(e)
 

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@ import (`
`5`	`5`	`"github.com/labstack/echo/v4/middleware"`
`6`	`6`	`"html/template"`
`7`	`7`	`"io"`
	`8`	`+ "net/http"`
`8`	`9`	`)`
`9`	`10`
`10`	`11`	`type Template struct {`
`@@ -47,6 +48,11 @@ func startWebServer() {`
`47`	`48`	`}`
`48`	`49`	`}`
`49`	`50`
	`51`	`+ // Render 404 page`
	`52`	`+ echo.NotFoundHandler = func(c echo.Context) error {`
	`53`	`+ return c.Render(http.StatusNotFound, "404.html", "")`
	`54`	`+ }`
	`55`	`+`
`50`	`56`	`// Serve the routes`
`51`	`57`	`webRouter(e)`
`52`	`58`