Add ”SetNotFoundHandler“ and "SetForbiddenHandler" method for frontend util

websubsvc/frontend_util.go

@@ -53,6 +53,14 @@ func (s *FrontendUtil) StaticFilesHandler(ctx *gin.Context) {
 	s.fileServer.ServeHTTP(ctx.Writer, ctx.Request)
 }
 
+func (s *FrontendUtil) SetNotFoundHandler(fn gin.HandlerFunc) {
+	s.notFoundHdl = fn
+}
+
+func (s *FrontendUtil) SetForbiddenHandler(fn gin.HandlerFunc) {
+	s.forbiddenHdl = fn
+}
+
 func (s *FrontendUtil) DefaultNotFoundHandler(ctx *gin.Context) {
 	ctx.Writer.WriteString("404 not found")
 	ctx.Status(404)