Explorar el Código

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

ZRY hace 10 meses
padre
commit
b6b0c96225
Se han modificado 1 ficheros con 8 adiciones y 0 borrados
  1. 8 0
      websubsvc/frontend_util.go

+ 8 - 0
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)