reprogl

1
package middlewares
2

3
import (
4
	"net/http"
5

6
	"xelbot.com/reprogl/container"
7
	pkghttp "xelbot.com/reprogl/http"
8
	"xelbot.com/reprogl/utils/tracking"
9
)
10

11
func Track(next http.Handler, app *container.Application) http.Handler {
12
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
13
		activity := tracking.CreateActivity(r)
14
		next.ServeHTTP(w, r)
15

16
		if activity != nil {
17
			lrw, ok := w.(pkghttp.LogResponseWriter)
18
			if ok {
19
				activity.Status = lrw.Status()
20
				activity.Duration = lrw.Duration()
21
			}
22

23
			go tracking.SaveActivity(activity, app)
24
		}
25
	})
26
}
27