2016-11-21 12:24:50 +00:00
|
|
|
package api
|
2016-11-19 21:35:23 +00:00
|
|
|
|
|
|
|
import (
|
2016-12-11 13:50:01 +00:00
|
|
|
"encoding/base64"
|
|
|
|
"net/http"
|
2016-12-24 13:57:04 +00:00
|
|
|
"time"
|
2016-12-25 15:37:45 +00:00
|
|
|
|
2016-12-11 13:50:01 +00:00
|
|
|
"github.com/mssola/user_agent"
|
2018-05-07 14:05:53 +00:00
|
|
|
"github.com/usefathom/fathom/pkg/counter"
|
2018-04-24 08:28:23 +00:00
|
|
|
"github.com/usefathom/fathom/pkg/datastore"
|
|
|
|
"github.com/usefathom/fathom/pkg/models"
|
2016-11-19 21:35:23 +00:00
|
|
|
)
|
|
|
|
|
2017-01-25 21:48:24 +00:00
|
|
|
/* middleware */
|
|
|
|
func NewCollectHandler() http.Handler {
|
2018-05-07 14:05:53 +00:00
|
|
|
go aggregate()
|
2016-12-11 13:50:01 +00:00
|
|
|
|
2018-04-25 09:59:30 +00:00
|
|
|
return HandlerFunc(func(w http.ResponseWriter, r *http.Request) error {
|
2017-01-25 21:48:24 +00:00
|
|
|
// abort if this is a bot.
|
2018-05-02 13:33:01 +00:00
|
|
|
userAgent := r.UserAgent()
|
|
|
|
ua := user_agent.New(userAgent)
|
2017-01-25 21:48:24 +00:00
|
|
|
if ua.Bot() {
|
2018-04-25 09:59:30 +00:00
|
|
|
return nil
|
2017-01-13 15:45:17 +00:00
|
|
|
}
|
2017-01-25 21:48:24 +00:00
|
|
|
|
|
|
|
q := r.URL.Query()
|
|
|
|
now := time.Now()
|
2016-12-11 13:50:01 +00:00
|
|
|
|
2018-05-02 13:33:01 +00:00
|
|
|
// get pageview details
|
2018-05-07 14:05:53 +00:00
|
|
|
pageview := &models.Pageview{
|
2018-05-06 09:53:19 +00:00
|
|
|
SessionID: q.Get("sid"),
|
|
|
|
Pathname: q.Get("p"),
|
2018-05-08 09:52:01 +00:00
|
|
|
IsNewVisitor: q.Get("nv") == "1",
|
|
|
|
IsNewSession: q.Get("ns") == "1",
|
2018-05-06 09:53:19 +00:00
|
|
|
IsUnique: q.Get("u") == "1",
|
|
|
|
IsBounce: q.Get("b") != "0",
|
|
|
|
Referrer: q.Get("r"),
|
|
|
|
Duration: 0,
|
|
|
|
Timestamp: now,
|
2017-01-25 21:48:24 +00:00
|
|
|
}
|
|
|
|
|
2018-05-07 14:05:53 +00:00
|
|
|
// find previous pageview by same visitor
|
2018-05-08 09:52:01 +00:00
|
|
|
if !pageview.IsNewSession {
|
|
|
|
previousPageview, err := datastore.GetMostRecentPageviewBySessionID(pageview.SessionID)
|
|
|
|
if err != nil && err != datastore.ErrNoResults {
|
2018-05-07 14:05:53 +00:00
|
|
|
return err
|
|
|
|
}
|
2018-05-08 09:52:01 +00:00
|
|
|
|
|
|
|
// if we have a recent pageview that is less than 30 minutes old
|
|
|
|
if previousPageview != nil && previousPageview.Timestamp.After(now.Add(-30*time.Minute)) {
|
|
|
|
previousPageview.Duration = (now.Unix() - previousPageview.Timestamp.Unix())
|
|
|
|
previousPageview.IsBounce = false
|
|
|
|
err := datastore.UpdatePageview(previousPageview)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
2018-05-07 14:05:53 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// save new pageview
|
2018-05-08 09:52:01 +00:00
|
|
|
err := datastore.SavePageview(pageview)
|
2018-05-06 09:53:19 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
2017-01-25 21:48:24 +00:00
|
|
|
}
|
2016-12-11 13:50:01 +00:00
|
|
|
|
2017-01-25 21:48:24 +00:00
|
|
|
// don't you cache this
|
|
|
|
w.Header().Set("Content-Type", "image/gif")
|
|
|
|
w.Header().Set("Expires", "Mon, 01 Jan 1990 00:00:00 GMT")
|
|
|
|
w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate")
|
|
|
|
w.Header().Set("Pragma", "no-cache")
|
|
|
|
w.WriteHeader(http.StatusOK)
|
2016-12-11 13:50:01 +00:00
|
|
|
|
2017-01-25 21:48:24 +00:00
|
|
|
// 1x1 px transparent GIF
|
|
|
|
b, _ := base64.StdEncoding.DecodeString("R0lGODlhAQABAIAAAAAAAP///yH5BAEAAAAALAAAAAABAAEAAAIBRAA7")
|
|
|
|
w.Write(b)
|
2018-04-25 09:59:30 +00:00
|
|
|
return nil
|
2017-01-25 21:48:24 +00:00
|
|
|
})
|
2016-11-19 21:35:23 +00:00
|
|
|
}
|
2018-05-07 14:05:53 +00:00
|
|
|
|
|
|
|
// runs the aggregate func every 5 mins
|
|
|
|
func aggregate() {
|
|
|
|
counter.Aggregate()
|
|
|
|
timeout := 5 * time.Minute
|
|
|
|
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case <-time.After(timeout):
|
|
|
|
counter.Aggregate()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|