go-couchbase库使用指南

go-couchbase是Couchbase官方提供的Golang客户端库，用于与Couchbase Server进行交互。下面我将详细介绍其使用方法。

安装

首先安装go-couchbase库：

go get github.com/couchbase/go-couchbase

基本连接

package main

import (
	"fmt"
	"github.com/couchbase/go-couchbase"
)

func main() {
	// 连接到Couchbase集群
	client, err := couchbase.Connect("http://localhost:8091")
	if err != nil {
		fmt.Printf("连接失败: %v\n", err)
		return
	}

	// 获取集群信息
	info, err := client.Info()
	if err != nil {
		fmt.Printf("获取集群信息失败: %v\n", err)
		return
	}
	fmt.Printf("集群信息: %+v\n", info)
}

操作Bucket

// 获取Bucket池
pool, err := client.GetPool("default")
if err != nil {
    fmt.Printf("获取池失败: %v\n", err)
    return
}

// 获取Bucket
bucket, err := pool.GetBucket("default")
if err != nil {
    fmt.Printf("获取Bucket失败: %v\n", err)
    return
}

// 写入数据
err = bucket.Set("key1", 0, "value1")
if err != nil {
    fmt.Printf("写入数据失败: %v\n", err)
}

// 读取数据
var value string
err = bucket.Get("key1", &value)
if err != nil {
    fmt.Printf("读取数据失败: %v\n", err)
} else {
    fmt.Printf("读取到的值: %s\n", value)
}

// 删除数据
err = bucket.Delete("key1")
if err != nil {
    fmt.Printf("删除数据失败: %v\n", err)
}

批量操作

// 批量写入
items := []couchbase.BulkItem{
    {Key: "key2", Value: "value2"},
    {Key: "key3", Value: "value3"},
    {Key: "key4", Value: "value4"},
}
err = bucket.Bulk(items)
if err != nil {
    fmt.Printf("批量写入失败: %v\n", err)
}

// 批量读取
keys := []string{"key2", "key3", "key4"}
results := make(map[string]interface{})
err = bucket.GetBulk(keys, &results)
if err != nil {
    fmt.Printf("批量读取失败: %v\n", err)
} else {
    fmt.Printf("批量读取结果: %+v\n", results)
}

视图查询

// 执行视图查询
var result interface{}
err = bucket.View("design_doc", "view_name", map[string]interface{}{
    "key":       "some_key",
    "reduce":    false,
    "limit":     10,
}, &result)
if err != nil {
    fmt.Printf("视图查询失败: %v\n", err)
} else {
    fmt.Printf("视图查询结果: %+v\n", result)
}

N1QL查询

// 执行N1QL查询
var queryResult []interface{}
err = bucket.Query("SELECT * FROM `default` WHERE type = $1", map[string]interface{}{
    "$1": "user",
}, &queryResult)
if err != nil {
    fmt.Printf("N1QL查询失败: %v\n", err)
} else {
    fmt.Printf("N1QL查询结果: %+v\n", queryResult)
}

高级功能

乐观锁(CAS)

// 获取CAS值
var cas uint64
err = bucket.Gets("key2", &value, &cas)
if err != nil {
    fmt.Printf("获取CAS失败: %v\n", err)
    return
}

// 使用CAS更新
err = bucket.Cas("key2", 0, cas, "new_value")
if err != nil {
    fmt.Printf("CAS更新失败: %v\n", err)
}

过期时间设置

// 设置10秒后过期
err = bucket.Set("temp_key", 10, "temporary_value")
if err != nil {
    fmt.Printf("设置过期数据失败: %v\n", err)
}

连接池管理

// 创建自定义连接池
poolConfig := &couchbase.PoolConfig{
    MaxIdleConns:        10,
    MaxActiveConns:      100,
    ConnWaitTimeout:     30 * time.Second,
    ConnectTimeout:      10 * time.Second,
    BucketConnectDelay:  100 * time.Millisecond,
}

client, err := couchbase.ConnectWithConfig("http://localhost:8091", poolConfig)
if err != nil {
    fmt.Printf("自定义连接池连接失败: %v\n", err)
    return
}

错误处理

Couchbase操作可能返回多种错误，建议进行专门处理：

if err != nil {
    switch err := err.(type) {
    case *couchbase.Error:
        if err.IsKeyNotFound() {
            fmt.Println("键不存在")
        } else if err.IsTemporary() {
            fmt.Println("临时错误，可以重试")
        } else {
            fmt.Printf("Couchbase错误: %v\n", err)
        }
    default:
        fmt.Printf("其他错误: %v\n", err)
    }
    return
}

最佳实践

1. 复用Bucket对象，避免频繁创建和销毁
2. 对大文档考虑使用压缩
3. 合理设置过期时间
4. 使用CAS操作处理并发更新
5. 监控连接池状态，避免连接泄漏

go-couchbase提供了丰富的功能来操作Couchbase数据库，以上示例涵盖了大部分常用场景。更详细的信息可以参考官方文档。