go/kubeprobes
go/kubeprobes
3
0
Fork 0
Code Issues 1 Pull Requests 1 Releases 24 Activity

feat(handler): change path resolver logic #5

Merged
Piotr Icikowski merged 1 commits from resolver into devel 2024-02-29 01:23:54 +01:00
Conversation 0 Commits 1 Files Changed 1 +5 -3
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 40bad6b345 - Show all commits

8
probes.go
View File

@ -2,6 +2,7 @@ package kubeprobes
import ( import (
"net/http" "net/http"
"strings"
) )
type kubeprobes struct { type kubeprobes struct {
@ -11,15 +12,16 @@ type kubeprobes struct {
// ServeHTTP implements http.Handler interface // ServeHTTP implements http.Handler interface
func (kp *kubeprobes) ServeHTTP(w http.ResponseWriter, r *http.Request) { func (kp *kubeprobes) ServeHTTP(w http.ResponseWriter, r *http.Request) {
switch r.URL.Path { subs := strings.Split(strings.TrimSuffix(r.URL.Path, "/"), "/")
case "/live": switch subs[len(subs)-1] {
case "live":
sq := newStatusQuery(kp.livenessProbes) sq := newStatusQuery(kp.livenessProbes)
if sq.isAllGreen() { if sq.isAllGreen() {
w.WriteHeader(http.StatusOK) w.WriteHeader(http.StatusOK)
} else { } else {
w.WriteHeader(http.StatusServiceUnavailable) w.WriteHeader(http.StatusServiceUnavailable)
} }
case "/ready": case "ready":
sq := newStatusQuery(append(kp.livenessProbes, kp.readinessProbes...)) sq := newStatusQuery(append(kp.livenessProbes, kp.readinessProbes...))
if sq.isAllGreen() { if sq.isAllGreen() {
w.WriteHeader(http.StatusOK) w.WriteHeader(http.StatusOK)