在Go语言中高效连接与操作Apache Cassandra：实用教程

go get github.com/gocql/gocql

package main

import (
    "fmt"
    "log"
    "time"

    "github.com/gocql/gocql"
)

func main() {
    // 1. 创建集群配置
    cluster := gocql.NewCluster("127.0.0.1") // 替换为你的Cassandra节点IP地址

    // 2. 配置Keyspace
    cluster.Keyspace = "mykeyspace" // 替换为你的Keyspace名称

    // 3. 配置身份认证（如果需要）
    // cluster.Authenticator = gocql.PasswordAuthenticator{
    //     Username: "cassandra",
    //     Password: "cassandra",
    // }

    // 4. 配置其他参数
    cluster.Consistency = gocql.Quorum        // 默认一致性级别为Quorum
    cluster.ProtoVersion = 4                  // 推荐使用最新协议版本
    cluster.Timeout = 5 * time.Second         // 连接超时时间
    cluster.ConnectTimeout = 10 * time.Second // 建立连接的超时时间
    cluster.Port = 9042                       // Cassandra默认端口

    // 5. 创建会话
    session, err := cluster.CreateSession()
    if err != nil {
        log.Fatalf("无法连接到Cassandra集群: %v", err)
    }
    defer session.Close() // 确保会话在使用完毕后关闭，释放资源

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

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

CREATE KEYSPACE IF NOT EXISTS mykeyspace WITH replication = {'class': 'SimpleStrategy', 'replication_factor': 1};

CREATE TABLE IF NOT EXISTS mykeyspace.users (
    id UUID PRIMARY KEY,
    name text,
    email text,
    age int
);

func insertUser(session *gocql.Session, id gocql.UUID, name, email string, age int) error {
    query := `INSERT INTO users (id, name, email, age) VALUES (?, ?, ?, ?)`
    if err := session.Query(query, id, name, email, age).Exec(); err != nil {
        return fmt.Errorf("插入用户失败: %w", err)
    }
    fmt.Printf("用户 %s 插入成功。\n", name)
    return nil
}

// 在 main 函数中调用
// userID := gocql.MustParseUUID("a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11")
// if err := insertUser(session, userID, "张三", "zhangsan@example.com", 30); err != nil {
//     log.Println(err)
// }

type User struct {
    ID    gocql.UUID
    Name  string
    Email string
    Age   int
}

func getUserByID(session *gocql.Session, id gocql.UUID) (*User, error) {
    query := `SELECT id, name, email, age FROM users WHERE id = ? LIMIT 1`
    var user User
    if err := session.Query(query, id).Scan(&user.ID, &user.Name, &user.Email, &user.Age); err != nil {
        if err == gocql.ErrNotFound {
            return nil, fmt.Errorf("用户 ID %s 未找到", id.String())
        }
        return nil, fmt.Errorf("查询用户失败: %w", err)
    }
    return &user, nil
}

func getAllUsers(session *gocql.Session) ([]User, error) {
    query := `SELECT id, name, email, age FROM users`
    iter := session.Query(query).Iter()
    var users []User
    var user User
    for iter.Scan(&user.ID, &user.Name, &user.Email, &user.Age) {
        users = append(users, user)
    }
    if err := iter.Close(); err != nil {
        return nil, fmt.Errorf("关闭迭代器失败: %w", err)
    }
    return users, nil
}

// 在 main 函数中调用
// user, err := getUserByID(session, userID)
// if err != nil {
//     log.Println(err)
// } else {
//     fmt.Printf("查询到用户: %+v\n", user)
// }
//
// allUsers, err := getAllUsers(session)
// if err != nil {
//     log.Println(err)
// } else {
//     fmt.Printf("所有用户: %+v\n", allUsers)
// }

func updateUserEmail(session *gocql.Session, id gocql.UUID, newEmail string) error {
    query := `UPDATE users SET email = ? WHERE id = ?`
    if err := session.Query(query, newEmail, id).Exec(); err != nil {
        return fmt.Errorf("更新用户邮件失败: %w", err)
    }
    fmt.Printf("用户 ID %s 的邮件更新成功。\n", id.String())
    return nil
}

// 在 main 函数中调用
// if err := updateUserEmail(session, userID, "new.email@example.com"); err != nil {
//     log.Println(err)
// }

func deleteUser(session *gocql.Session, id gocql.UUID) error {
    query := `DELETE FROM users WHERE id = ?`
    if err := session.Query(query, id).Exec(); err != nil {
        return fmt.Errorf("删除用户失败: %w", err)
    }
    fmt.Printf("用户 ID %s 删除成功。\n", id.String())
    return nil
}

// 在 main 函数中调用
// if err := deleteUser(session, userID); err != nil {
//     log.Println(err)
// }