package datasaver

import (
	"log"
	"time"

	"openreplay/backend/pkg/db/clickhouse"
	"openreplay/backend/pkg/db/types"
	"openreplay/backend/pkg/env"
	"openreplay/backend/pkg/messages"
)

var finalizeTicker <-chan time.Time

func (si *Saver) InitStats() {
	si.ch = clickhouse.NewConnector(env.String("CLICKHOUSE_STRING"))
	if err := si.ch.Prepare(); err != nil {
		log.Fatalf("Clickhouse prepare error: %v\n", err)
	}
	si.pg.Conn.SetClickHouse(si.ch)
	finalizeTicker = time.Tick(20 * time.Minute)
}

func (si *Saver) InsertStats(session *types.Session, msg messages.Message) error {
	switch m := msg.(type) {
	// Web
	case *messages.SessionEnd:
		return si.ch.InsertWebSession(session)
	case *messages.PerformanceTrackAggr:
		return si.ch.InsertWebPerformanceTrackAggr(session, m)
	case *messages.ClickEvent:
		return si.ch.InsertWebClickEvent(session, m)
	case *messages.InputEvent:
		return si.ch.InsertWebInputEvent(session, m)
	// Unique for Web
	case *messages.PageEvent:
		return si.ch.InsertWebPageEvent(session, m)
	case *messages.ResourceEvent:
		return si.ch.InsertWebResourceEvent(session, m)
	case *messages.ErrorEvent:
		return si.ch.InsertWebErrorEvent(session, m)
	}
	return nil
}

func (si *Saver) CommitStats(optimize bool) error {
	if !optimize {
		return si.ch.Commit()
	}
	select {
	case <-finalizeTicker:
		if err := si.ch.FinaliseSessionsTable(); err != nil {
			log.Printf("Stats: FinaliseSessionsTable returned an error. %v", err)
		}
	default:
	}
	return si.ch.Commit()
}