package db

import (
	"log"
	"os"

	"github.com/gocql/gocql"
)

var Session *gocql.Session

func InitScyllaDB() {
	cluster := gocql.NewCluster(os.Getenv("SCYLLA_CLUSTER")) // Replace with your ScyllaDB cluster IPs
	cluster.Keyspace = os.Getenv("SCYLLA_KEYSPACE")          // Replace with your keyspace
	cluster.Consistency = gocql.Quorum

	session, err := cluster.CreateSession()
	if err != nil {
		log.Fatalf("Failed to connect to ScyllaDB: %v", err)
	}
	Session = session
	log.Println("Connected to ScyllaDB")
}

func CloseScyllaDB() {
	if Session != nil {
		Session.Close()
	}
}