diff options
Diffstat (limited to 'cmd/site/main.go')
| -rw-r--r-- | cmd/site/main.go | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/cmd/site/main.go b/cmd/site/main.go index bad297e..7dbe9bc 100644 --- a/cmd/site/main.go +++ b/cmd/site/main.go @@ -17,6 +17,7 @@ import ( "github.com/gorilla/feeds" blackfriday "github.com/russross/blackfriday" "github.com/tj/front" + "gopkg.in/segmentio/analytics-go.v3" ) var port = os.Getenv("PORT") @@ -47,11 +48,24 @@ type Site struct { templates map[string]*template.Template tlock sync.RWMutex + + segment analytics.Client } func (s *Site) ServeHTTP(w http.ResponseWriter, r *http.Request) { ln.Log(r.Context(), ln.F{"action": "Site.ServeHTTP", "user_ip_address": r.RemoteAddr, "path": r.RequestURI}) + s.mux.ServeHTTP(w, r) + + if s.segment != nil { + if !strings.HasPrefix(r.RequestURI, "/blog/") { + s.segment.Enqueue(&analytics.Track{ + UserId: Hash(r.RemoteAddr, r.Header.Get("X-Forwarded-For")), + Event: "Page Viewed", + Properties: analytics.NewProperties().SetURL(r.RequestURI), + }) + } + } } // Build creates a new Site instance or fails. @@ -88,6 +102,10 @@ func Build() (*Site, error) { templates: map[string]*template.Template{}, } + if wk := os.Getenv("SEGMENT_WRITE_KEY"); wk != "" { + s.segment = analytics.New(wk) + } + err := filepath.Walk("./blog/", func(path string, info os.FileInfo, err error) error { if err != nil { return err |