Commitfeee09f

committed

fix(response): add custom response

1 parent6b5b686 commitfeee09fCopy full SHA for feee09f

File tree

-3

lines changed

-3

lines changed

Lines changed: 7 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -30,9 +30,13 @@ func (m Middleware) blockRequest(recorder http.ResponseWriter, r http.Request,`
`30`	`30`	`recorder.Header().Set("Content-Type","text/plain")`
`31`	`31`	`recorder.WriteHeader(statusCode)`
`32`	`32`
`33`		`-message:=fmt.Sprintf("Request blocked by WAF. Reason: %s",reason)`
`34`		`-if_,err:=recorder.Write([]byte(message));err!=nil {`
`35`		`-m.logger.Error("Failed to write blocked response",zap.Error(err))`
	`33`	`+ifm.CustomResponses!=nil {`
	`34`	`+m.writeCustomResponse(recorder,state.StatusCode)`
	`35`	`+}else {`
	`36`	`+message:=fmt.Sprintf("Request blocked by WAF. Reason: %s",reason)`
	`37`	`+if_,err:=recorder.Write([]byte(message));err!=nil {`
	`38`	`+m.logger.Error("Failed to write blocked response",zap.Error(err))`
	`39`	`+}`
`36`	`40`	`}`
`37`	`41`	`}`
`38`	`42`

Comments

(0)