Adds Jicofo endpoint

main
Yasen Pramatarov 2024-09-02 22:34:38 +03:00
parent 1a8ef2ec3a
commit 22d287f5f3
1 changed files with 29 additions and 1 deletions

View File

@ -55,6 +55,20 @@ func getNginxState() string {
return "not running" return "not running"
} }
// getJicofoState checks the status of the Jicofo service
func getJicofoState() string {
output, err := exec.Command("systemctl", "is-active", "jicofo").Output()
if err != nil {
log.Printf("Error checking the jicofo state: %v", err)
return "error"
}
state := strings.TrimSpace(string(output))
if state == "active" {
return "running"
}
return "not running"
}
// getNginxConnections gets the number of active connections to the specified web port // getNginxConnections gets the number of active connections to the specified web port
func getNginxConnections(nginxPort int) int { func getNginxConnections(nginxPort int) int {
cmd := fmt.Sprintf("netstat -an | grep ':%d' | wc -l", nginxPort) cmd := fmt.Sprintf("netstat -an | grep ':%d' | wc -l", nginxPort)
@ -119,6 +133,17 @@ func nginxHandler(config Config, w http.ResponseWriter, r *http.Request) {
json.NewEncoder(w).Encode(data) json.NewEncoder(w).Encode(data)
} }
// jicofoHandler handles the /jicofo endpoint
func jicofoHandler(config Config, w http.ResponseWriter, r *http.Request) {
data := JicofoData {
JicofoState: getJicofoState(),
JicofoAPIData: getJicofoAPIData(),
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(data)
}
// main sets up the http server and the routes // main sets up the http server and the routes
func main() { func main() {
@ -132,10 +157,13 @@ func main() {
http.HandleFunc("/nginx", func(w http.ResponseWriter, r *http.Request) { http.HandleFunc("/nginx", func(w http.ResponseWriter, r *http.Request) {
nginxHandler(config, w, r) nginxHandler(config, w, r)
}) })
http.HandleFunc("/jicofo", func(w http.ResponseWriter, r *http.Request) {
jicofoHandler(config, w, r)
})
// start the http server // start the http server
agentPortStr := fmt.Sprintf(":%d", config.AgentPort) agentPortStr := fmt.Sprintf(":%d", config.AgentPort)
fmt.Printf("Starting agent server on port %d.\n", config.AgentPort) fmt.Printf("Starting Jilo agent server on port %d.\n", config.AgentPort)
if err := http.ListenAndServe(agentPortStr, nil); err != nil { if err := http.ListenAndServe(agentPortStr, nil); err != nil {
log.Fatalf("Could not start the server: %v\n", err) log.Fatalf("Could not start the server: %v\n", err)
} }