29 lines
544 B
Go
29 lines
544 B
Go
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()
|
|
}
|
|
}
|