track a user as recurring only if he's not new

dawanda · Apr 8, 2016 · 81e30fd · 81e30fd
1 parent 75066c0
commit 81e30fd
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -40,7 +40,10 @@ func (service *Service) httpTrack(w http.ResponseWriter, r *http.Request) {
 	} else {
 		cookie.Expires = time.Now().Add(service.CookieTTL)
 		http.SetCookie(w, cookie)
-		service.RecurringSessions.Touch(cookie.Value)
+
+		if !service.NewSessions.Contains(cookie.Value) {
+			service.RecurringSessions.Touch(cookie.Value)
+		}
 	}
 
 	service.writeEmptyGif(w, r)

diff --git a/tracker.go b/tracker.go
@@ -64,6 +64,13 @@ func (tracker *Tracker) Debugf(msg string, args ...interface{}) {
 	}
 }
 
+func (tracker *Tracker) Contains(sessionID string) bool {
+	tracker.bucketMutex.Lock()
+	_, ok := tracker.buckets[sessionID]
+	tracker.bucketMutex.Unlock()
+	return ok
+}
+
 func (tracker *Tracker) Touch(sessionID string) {
 	tracker.bucketMutex.Lock()