buyfakett/server_torii
1
0
Fork 0
mirror of https://github.com/Rayzggz/server_torii.git synced 2025-06-17 04:31:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: health check api

Browse Source
This commit is contained in:
Roi Feng
2025-04-29 17:38:52 -04:00
parent e73a786311
commit 591ddb99f7
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/server/torii.go
View File

@ -18,8 +18,19 @@ func CheckTorii(w http.ResponseWriter, r *http.Request, reqData dataType.UserReq
decision.SetCode(action.Continue, []byte("403")) decision.SetCode(action.Continue, []byte("403"))
if reqData.Uri == cfg.WebPath+"/captcha" { if reqData.Uri == cfg.WebPath+"/captcha" {
check.CheckCaptcha(r, reqData, ruleSet, decision) check.CheckCaptcha(r, reqData, ruleSet, decision)
} else if reqData.Uri == cfg.WebPath+"/health_check" {
decision.SetResponse(action.Done, []byte("200"), []byte("ok"))
} }
if bytes.Compare(decision.HTTPCode, []byte("200")) == 0 { if bytes.Compare(decision.HTTPCode, []byte("200")) == 0 {
if bytes.Compare(decision.ResponseData, []byte("ok")) == 0 {
w.WriteHeader(http.StatusOK)
_, err := w.Write([]byte("ok"))
if err != nil {
utils.LogError(reqData, "Error writing response: "+err.Error(), "CheckTorii")
return
}
return
}
if bytes.Compare(decision.ResponseData, []byte("bad")) == 0 { if bytes.Compare(decision.ResponseData, []byte("bad")) == 0 {
w.WriteHeader(http.StatusOK) w.WriteHeader(http.StatusOK)
_, err := w.Write([]byte("bad")) _, err := w.Write([]byte("bad"))