From 0b7bb2339d2cf867fce4443ea8a4357bdabc50ba Mon Sep 17 00:00:00 2001
From: Alexander Zavorotynskiy <alexander@openreplay.com>
Date: Mon, 30 May 2022 17:41:45 +0200
Subject: [PATCH] fix(backend/datasaver): changed postgres on clickhouse and
 added missed imports

---
 ee/backend/internal/datasaver/stats.go | 30 ++++++++++++++------------
 1 file changed, 16 insertions(+), 14 deletions(-)

diff --git a/ee/backend/internal/datasaver/stats.go b/ee/backend/internal/datasaver/stats.go
index 501a861aa..2017dcd20 100644
--- a/ee/backend/internal/datasaver/stats.go
+++ b/ee/backend/internal/datasaver/stats.go
@@ -5,7 +5,9 @@ import (
 	"time"
 
 	"openreplay/backend/pkg/db/clickhouse"
+	. "openreplay/backend/pkg/db/types"
 	"openreplay/backend/pkg/env"
+	. "openreplay/backend/pkg/messages"
 )
 
 var ch *clickhouse.Connector
@@ -25,39 +27,39 @@ func (si *Saver) InsertStats(session *Session, msg Message) error {
 	switch m := msg.(type) {
 	// Web
 	case *SessionEnd:
-		return si.pg.InsertWebSession(session)
+		return ch.InsertWebSession(session)
 	case *PerformanceTrackAggr:
-		return si.pg.InsertWebPerformanceTrackAggr(session, m)
+		return ch.InsertWebPerformanceTrackAggr(session, m)
 	case *ClickEvent:
-		return si.pg.InsertWebClickEvent(session, m)
+		return ch.InsertWebClickEvent(session, m)
 	case *InputEvent:
-		return si.pg.InsertWebInputEvent(session, m)
+		return ch.InsertWebInputEvent(session, m)
 		// Unique for Web
 	case *PageEvent:
-		si.pg.InsertWebPageEvent(session, m)
+		ch.InsertWebPageEvent(session, m)
 	case *ResourceEvent:
-		return si.pg.InsertWebResourceEvent(session, m)
+		return ch.InsertWebResourceEvent(session, m)
 	case *ErrorEvent:
-		return si.pg.InsertWebErrorEvent(session, m)
+		return ch.InsertWebErrorEvent(session, m)
 	case *LongTask:
-		return si.pg.InsertLongtask(session, m)
+		return ch.InsertLongtask(session, m)
 
 	// IOS
 	case *IOSSessionEnd:
-		return si.pg.InsertIOSSession(session)
+		return ch.InsertIOSSession(session)
 	case *IOSPerformanceAggregated:
-		return si.pg.InsertIOSPerformanceAggregated(session, m)
+		return ch.InsertIOSPerformanceAggregated(session, m)
 	case *IOSClickEvent:
-		return si.pg.InsertIOSClickEvent(session, m)
+		return ch.InsertIOSClickEvent(session, m)
 	case *IOSInputEvent:
-		return si.pg.InsertIOSInputEvent(session, m)
+		return ch.InsertIOSInputEvent(session, m)
 	// Unique for Web
 	case *IOSScreenEnter:
 		//ch.InsertIOSView(session, m)
 	case *IOSCrash:
-		return si.pg.InsertIOSCrash(session, m)
+		return ch.InsertIOSCrash(session, m)
 	case *IOSNetworkCall:
-		return si.pg.InsertIOSNetworkCall(session, m)
+		return ch.InsertIOSNetworkCall(session, m)
 	}
 	return nil
 }