package db

import (
	"log"

	"github.com/gocql/gocql"
)

var Session *gocql.Session

func InitScyllaDB() {
	cluster := gocql.NewCluster("127.0.0.1") // Replace with your ScyllaDB cluster IPs
	cluster.Keyspace = "chatservice"         // 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()
	}
}