Go语言与Cassandra数据库客户端连接指南

go get github.com/gocql/gocql

CREATE KEYSPACE IF NOT EXISTS mykeyspace WITH replication = {'class': 'SimpleStrategy', 'replication_factor': 1};
USE mykeyspace;
CREATE TABLE IF NOT EXISTS users (
    id UUID PRIMARY KEY,
    name text,
    email text
);

package main

import (
    "fmt"
    "log"
    "time"

    "github.com/gocql/gocql" // 导入gocql库
)

func main() {
    // 1. 配置Cassandra集群连接参数
    // NewCluster函数接受一个或多个Cassandra节点IP地址作为参数
    cluster := gocql.NewCluster("127.0.0.1") // 替换为你的Cassandra节点IP地址
    cluster.Keyspace = "mykeyspace"          // 指定要连接的Keyspace
    cluster.Consistency = gocql.Quorum       // 设置默认一致性级别为Quorum
    cluster.Port = 9042                      // Cassandra的默认CQL端口
    cluster.Timeout = 5 * time.Second        // 每个请求的超时时间
    cluster.ConnectTimeout = 10 * time.Second // 初始连接到集群的超时时间
    cluster.NumConns = 2                     // 每个主机维护的连接数，默认为2

    // 2. 创建Session
    // Session是与Cassandra集群交互的主要接口，它管理着连接池和请求路由。
    session, err := cluster.CreateSession()
    if err != nil {
        log.Fatalf("无法连接到Cassandra集群: %v", err)
    }
    defer session.Close() // 确保在main函数退出时关闭Session，释放资源

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

    // 3. 示例操作：插入数据
    fmt.Println("\n--- 插入数据 ---")
    // Query方法用于构建一个查询。Exec方法执行非查询操作（INSERT, UPDATE, DELETE）。
    // gocql.TimeUUID() 生成一个基于时间的UUID，适合作为主键。
    if err := session.Query("INSERT INTO users (id, name, email) VALUES (?, ?, ?)",
        gocql.TimeUUID(), "张三", "zhangsan@example.com").Exec(); err != nil {
        log.Printf("插入数据失败: %v", err)
    } else {
        fmt.Println("数据插入成功。")
    }

    // 4. 示例操作：查询数据
    fmt.Println("\n--- 查询数据 ---")
    var id gocql.UUID
    var name, email string
    // Iter() 方法返回一个迭代器，用于遍历查询结果集。
    iter := session.Query("SELECT id, name, email FROM users WHERE name = ?", "张三").Iter()
    // Scan方法将当前行的数据扫描到指定的变量中。
    for iter.Scan(&id, &name, &email) {
        fmt.Printf("ID: %s, 姓名: %s, 邮箱: %s\n", id.String(), name, email)
    }
    // 遍历结束后，务必关闭迭代器以释放资源。
    if err := iter.Close(); err != nil {
        log.Printf("查询迭代器关闭失败: %v", err)
    }

    // 5. 示例操作：使用预处理语句插入数据 (推荐实践)
    // 预处理语句可以提高性能，减少解析开销，并防止SQL注入。
    fmt.Println("\n--- 使用预处理语句插入数据 ---")
    insertStmt := `INSERT INTO users (id, name, email) VALUES (?, ?, ?)`
    preparedInsert := session.Query(insertStmt) // 预处理查询
    if err := preparedInsert.Bind(gocql.TimeUUID(), "李四", "lisi@example.com").Exec(); err != nil {
        log.Printf("使用预处理语句插入数据失败: %v", err)
    } else {
        fmt.Println("使用预处理语句插入数据成功。")
    }

    // 6. 示例操作：更新数据
    fmt.Println("\n--- 更新数据 ---")
    updateStmt := `UPDATE users SET email = ? WHERE name = ?`
    if err := session.Query(updateStmt, "zhangsan_new@example.com", "张三").Exec(); err != nil {
        log.Printf("更新数据失败: %v", err)
    } else {
        fmt.Println("数据更新成功。")
    }

    // 再次查询以验证更新
    fmt.Println("\n--- 再次查询以验证更新 ---")
    iter = session.Query("SELECT id, name, email FROM users WHERE name = ?", "张三").Iter()
    for iter.Scan(&id, &name, &email) {
        fmt.Printf("ID: %s, 姓名: %s, 邮箱: %s\n", id.String(), name, email)
    }
    if err := iter.Close(); err != nil {
        log.Printf("查询迭代器关闭失败: %v", err)
    }

    // 7. 示例操作：删除数据
    fmt.Println("\n--- 删除数据 ---")
    deleteStmt := `DELETE FROM users WHERE name = ?`
    if err := session.Query(deleteStmt, "李四").Exec(); err != nil {
        log.Printf("删除数据失败: %v", err)
    } else {
        fmt.Println("数据删除成功。")
    }

    // 验证删除
    fmt.Println("\n--- 验证删除 ---")
    count := 0
    // COUNT(*) 查询返回匹配行的数量
    if err := session.Query("SELECT COUNT(*) FROM users WHERE name = ?", "李四").Scan(&count); err != nil {
        log.Printf("查询计数失败: %v", err)
    } else {
        fmt.Printf("姓名 '李四' 的剩余记录数: %d\n", count)
    }
}