From 1ff51c44cb7a7109c5d3891f07d36c3023309810 Mon Sep 17 00:00:00 2001
From: Alexander Zavorotynskiy <zavorotynskiy@pm.me>
Date: Thu, 1 Sep 2022 08:40:01 +0200
Subject: [PATCH] fix(backend): fixed bug in db saver (ee)

---
 ee/backend/internal/db/datasaver/saver.go | 10 ++++++----
 ee/backend/internal/db/datasaver/stats.go |  2 --
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/ee/backend/internal/db/datasaver/saver.go b/ee/backend/internal/db/datasaver/saver.go
index 2799ec087..7b48fbaa0 100644
--- a/ee/backend/internal/db/datasaver/saver.go
+++ b/ee/backend/internal/db/datasaver/saver.go
@@ -3,13 +3,15 @@ package datasaver
 import (
 	"openreplay/backend/pkg/db/cache"
 	"openreplay/backend/pkg/db/clickhouse"
+	"openreplay/backend/pkg/queue/types"
 )
 
 type Saver struct {
-	pg *cache.PGCache
-	ch clickhouse.Connector
+	pg       *cache.PGCache
+	ch       clickhouse.Connector
+	producer types.Producer
 }
 
-func New(pg *cache.PGCache) *Saver {
-	return &Saver{pg: pg}
+func New(pg *cache.PGCache, producer types.Producer) *Saver {
+	return &Saver{pg: pg, producer: producer}
 }
diff --git a/ee/backend/internal/db/datasaver/stats.go b/ee/backend/internal/db/datasaver/stats.go
index 084187ef2..ecf418090 100644
--- a/ee/backend/internal/db/datasaver/stats.go
+++ b/ee/backend/internal/db/datasaver/stats.go
@@ -25,10 +25,8 @@ func (si *Saver) InsertStats(session *types.Session, msg messages.Message) error
 	switch m := msg.(type) {
 	// Web
 	case *messages.SessionEnd:
-		// TODO: get issue_types and base_referrer before session end
 		return si.ch.InsertWebSession(session)
 	case *messages.PerformanceTrackAggr:
-		// TODO: page_path
 		return si.ch.InsertWebPerformanceTrackAggr(session, m)
 	case *messages.ClickEvent:
 		return si.ch.InsertWebClickEvent(session, m)