GitHub-Proxy/signoz
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz synced 2025-08-14 01:36:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

live tail api excluded from timeout middleware

Browse Source
This commit is contained in:
nityanandagohain 2022-07-21 17:55:08 +05:30
parent 5912d3a4a0
commit 2132d1059c
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/query-service/app/server.go
View File

@ -242,8 +242,14 @@ func (s *Server) analyticsMiddleware(next http.Handler) http.Handler {
func setTimeoutMiddleware(next http.Handler) http.Handler { func setTimeoutMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
ctx, cancel := context.WithTimeout(r.Context(), constants.ContextTimeout*time.Second) ctx := r.Context()
defer cancel() var cancel context.CancelFunc
// check if route is not excluded
url := r.URL.Path
if _, ok := constants.TimeoutExcludedRoutes[url]; !ok {
ctx, cancel = context.WithTimeout(r.Context(), constants.ContextTimeout*time.Second)
defer cancel()
}
r = r.WithContext(ctx) r = r.WithContext(ctx)
next.ServeHTTP(w, r) next.ServeHTTP(w, r)

4
pkg/query-service/constants/constants.go
View File

@ -81,6 +81,10 @@ const (
SIGNOZ_TIMESERIES_TABLENAME = "time_series_v2" SIGNOZ_TIMESERIES_TABLENAME = "time_series_v2"
) )
var TimeoutExcludedRoutes = map[string]bool{
"/api/v1/logs/tail": true,
}
// alert related constants // alert related constants
const ( const (
// AlertHelpPage is used in case default alert repo url is not set // AlertHelpPage is used in case default alert repo url is not set