ynerant/ghostream
1
0
Fork 0
mirror of https://gitlab.crans.org/nounous/ghostream.git synced 2025-06-27 11:28:51 +02:00
Code Issues Releases Wiki Activity

Add prometheus monitoring endpoint

Browse Source
This commit is contained in:
Alexandre Iooss
2020-09-21 21:33:32 +02:00
parent ad75d2c774
commit 7cdd78a70e
6 changed files with 116 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
web/web.go
View File

@ -7,6 +7,7 @@ import (
"os"
"gitlab.crans.org/nounous/ghostream/internal/config"
"gitlab.crans.org/nounous/ghostream/monitoring"
)
// Preload templates
@ -28,6 +29,9 @@ func viewerHandler(w http.ResponseWriter, r *http.Request, cfg *config.Config) {
log.Println(err.Error())
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
}
// Increment monitoring
monitoring.ViewerServed.Inc()
}
// Auth incoming stream
@ -59,9 +63,10 @@ func makeHandler(fn func(http.ResponseWriter, *http.Request, *config.Config), cf
// ServeHTTP server
func ServeHTTP(cfg *config.Config) {
// Set up HTTP router and server
http.HandleFunc("/", makeHandler(viewerHandler, cfg))
http.HandleFunc("/rtmp/auth", makeHandler(streamAuthHandler, cfg))
http.HandleFunc("/static/", makeHandler(staticHandler, cfg))
log.Printf("Listening on http://%s/", cfg.Site.ListenAdress)
log.Fatal(http.ListenAndServe(cfg.Site.ListenAdress, nil))
mux := http.NewServeMux()
mux.HandleFunc("/", makeHandler(viewerHandler, cfg))
mux.HandleFunc("/rtmp/auth", makeHandler(streamAuthHandler, cfg))
mux.HandleFunc("/static/", makeHandler(staticHandler, cfg))
log.Printf("Listening on http://%s/", cfg.Site.ListenAddress)
log.Fatal(http.ListenAndServe(cfg.Site.ListenAddress, mux))
}