Go语言与Cassandra：构建高效数据存储客户端

go get github.com/gocql/gocql

package main

import (
    "fmt"
    "log"
    "time"

    "github.com/gocql/gocql"
)

func main() {
    // 配置Cassandra集群信息
    // 替换为您的Cassandra节点IP地址
    cluster := gocql.NewCluster("127.0.0.1")
    cluster.Keyspace = "mykeyspace" // 指定要连接的键空间
    cluster.Consistency = gocql.Quorum // 设置默认一致性级别
    cluster.ProtoVersion = 4 // 建议使用Cassandra协议版本4或更高
    cluster.Timeout = time.Second * 5 // 查询超时时间
    cluster.ConnectTimeout = time.Second * 5 // 连接建立超时时间
    cluster.Port = 9042 // 默认端口

    // 创建会话
    session, err := cluster.CreateSession()
    if err != nil {
        log.Fatalf("无法连接到Cassandra集群: %v", err)
    }
    defer session.Close() // 确保在函数退出时关闭会话

    fmt.Println("成功连接到Cassandra集群！")

    // 接下来可以执行数据操作
}

// ... (之前的连接代码) ...

// 尝试创建键空间（如果不存在）
err = session.Query(`CREATE KEYSPACE IF NOT EXISTS mykeyspace WITH replication = {'class': 'SimpleStrategy', 'replication_factor': 1}`).Exec()
if err != nil {
    log.Fatalf("无法创建或连接到键空间: %v", err)
}
fmt.Println("键空间 'mykeyspace' 准备就绪。")

// 重新创建会话以确保连接到正确的键空间
// 注意：如果cluster.Keyspace在NewCluster时已经指定，则不需要重新创建会话
// 但如果是在连接后动态创建Keyspace并切换，则需要重新创建session
// 这里的示例是为了确保Keyspace存在后，session能正确指向它
session.Close() // 关闭旧会话
cluster.Keyspace = "mykeyspace"
session, err = cluster.CreateSession()
if err != nil {
    log.Fatalf("无法切换到键空间 'mykeyspace': %v", err)
}
defer session.Close() // 确保关闭新会话

// 创建表（如果不存在）
err = session.Query(`
    CREATE TABLE IF NOT EXISTS users (
        id UUID PRIMARY KEY,
        name text,
        email text
    )`).Exec()
if err != nil {
    log.Fatalf("无法创建表 'users': %v", err)
}
fmt.Println("表 'users' 准备就绪。")

// ... (后续数据操作) ...

// 插入数据
id := gocql.MustParseUUID("a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11")
name := "张三"
email := "zhangsan@example.com"

if err := session.Query(`INSERT INTO users (id, name, email) VALUES (?, ?, ?)`,
    id, name, email).Exec(); err != nil {
    log.Fatalf("插入数据失败: %v", err)
}
fmt.Println("数据插入成功！")

// 查询数据
var userId gocql.UUID
var userName, userEmail string

iter := session.Query(`SELECT id, name, email FROM users WHERE id = ?`, id).Iter()
for iter.