GoCQL - Golang高性能Cassandra驱动使用指南

GoCQL是Go语言中最流行的高性能Cassandra数据库驱动，它提供了简单高效的API来与Cassandra集群交互。

安装

go get github.com/gocql/gocql

基本使用

1. 创建集群连接

package main

import (
	"fmt"
	"log"
	"github.com/gocql/gocql"
)

func main() {
	// 创建集群配置
	cluster := gocql.NewCluster("127.0.0.1") // Cassandra节点地址
	cluster.Keyspace = "example"             // 指定keyspace
	cluster.Consistency = gocql.Quorum       // 设置一致性级别

	// 创建会话
	session, err := cluster.CreateSession()
	if err != nil {
		log.Fatal(err)
	}
	defer session.Close()

	fmt.Println("Connected to Cassandra cluster!")
}

2. 执行查询

// 插入数据
if err := session.Query(
	`INSERT INTO users (id, name, email) VALUES (?, ?, ?)`,
	"user1", "John Doe", "john@example.com").Exec(); err != nil {
	log.Fatal(err)
}

// 查询数据
var id, name, email string

iter := session.Query(`SELECT id, name, email FROM users WHERE id = ?`, "user1").Iter()
for iter.Scan(&id, &name, &email) {
	fmt.Printf("User: %s, Name: %s, Email: %s\n", id, name, email)
}
if err := iter.Close(); err != nil {
	log.Fatal(err)
}

高级特性

1. 批量操作

batch := session.NewBatch(gocql.LoggedBatch)
batch.Query("INSERT INTO users (id, name) VALUES (?, ?)", "user2", "Alice")
batch.Query("INSERT INTO users (id, name) VALUES (?, ?)", "user3", "Bob")

if err := session.ExecuteBatch(batch); err != nil {
	log.Fatal(err)
}

2. 预编译语句

// 准备语句
stmt := session.Query(
	`INSERT INTO users (id, name, email) VALUES (?, ?, ?)`).Prepared()

// 执行预编译语句
if err := stmt.Bind("user4", "Eve", "eve@example.com").Exec(); err != nil {
	log.Fatal(err)
}

3. 分页查询

var pageState []byte

for {
	iter := session.Query(`SELECT id, name FROM users`).
		PageSize(100).            // 每页100条
		PageState(pageState).     // 设置分页状态
		Iter()

	var id, name string
	for iter.Scan(&id, &name) {
		fmt.Printf("User: %s, Name: %s\n", id, name)
	}

	// 获取下一页状态
	pageState = iter.PageState()
	if len(pageState) == 0 {
		break // 没有更多数据
	}

	if err := iter.Close(); err != nil {
		log.Fatal(err)
	}
}

性能优化技巧

1. 连接池配置:

cluster.NumConns = 5 // 每个主机的连接数
cluster.PoolConfig.HostSelectionPolicy = gocql.TokenAwareHostPolicy(gocql.RoundRobinHostPolicy())

1. 一致性级别选择:

cluster.Consistency = gocql.LocalQuorum // 根据需求选择合适的一致性级别

1. 批量操作:

// 使用非日志批量操作提高性能
batch := session.NewBatch(gocql.UnloggedBatch)

1. 异步查询:

ch := make(chan *gocql.Query, 100)

// 生产者
go func() {
	for i := 0; i < 100; i++ {
		ch <- session.Query("INSERT INTO data (id, value) VALUES (?, ?)", i, fmt.Sprintf("value-%d", i))
	}
	close(ch)
}()

// 消费者
for q := range ch {
	q.ExecAsync()
}

错误处理

if err := session.Query("INSERT INTO invalid_table (id) VALUES (?)", "id1").Exec(); err != nil {
	switch e := err.(type) {
	case gocql.RequestError:
		fmt.Printf("Request error: %v\n", e)
	case gocql.HostError:
		fmt.Printf("Host error: %v\n", e)
	default:
		fmt.Printf("Other error: %v\n", e)
	}
}

最佳实践

1. 重用Session对象，不要频繁创建和关闭
2. 使用预编译语句提高性能
3. 合理设置一致性级别
4. 使用TokenAwareHostPolicy减少网络跳数
5. 监控查询性能，优化慢查询

GoCQL提供了丰富的功能和良好的性能，是Go语言与Cassandra交互的首选驱动。