internal/database/database.go (view raw)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
package database
import (
"fmt"
"log"
"os"
"github.com/gocql/gocql"
_ "github.com/joho/godotenv/autoload"
)
func New() Service {
// Load environment variables
cassandraHost := os.Getenv("CASSANDRA_HOST")
cassandraUsername := os.Getenv("CASSANDRA_USERNAME")
cassandraPassword := os.Getenv("CASSANDRA_PASSWORD")
cassandraKeyspace := os.Getenv("CASSANDRA_KEYSPACE")
cluster := gocql.NewCluster(cassandraHost)
cluster.Port = 9142
cluster.SslOpts = nil
cluster.Authenticator = gocql.PasswordAuthenticator{
Username: cassandraUsername,
Password: cassandraPassword,
}
cluster.SslOpts = &gocql.SslOptions{
CaPath: "cassandra_ca.crt",
EnableHostVerification: false,
}
cluster.Consistency = gocql.LocalQuorum
cluster.Keyspace = cassandraKeyspace
session, err := cluster.CreateSession()
if err != nil {
log.Fatal("Failed to create Cassandra session:", err)
}
return &service{session: session}
}
func (s *service) Health() map[string]string {
stats := make(map[string]string)
if err := s.session.Query("SELECT release_version FROM system.local").Exec(); err != nil {
stats["status"] = "down"
stats["error"] = fmt.Sprintf("Cassandra down: %v", err)
return stats
}
stats["status"] = "up"
stats["message"] = "Cassandra is healthy"
return stats
}
func (s *service) Close() error {
s.session.Close()
return nil
}
|