如何使用gocql实现Go类型到Scylla/Cassandra类型的转换？[]string转set<text>是否需要自定义映射器？

嘿，这个问题问得很到位！我来帮你拆解清楚：

1. []string 转 set：原生支持，无需自定义映射器 #

好消息是，gocql本身就内置了这种类型映射的支持，完全不需要你自己写自定义编解码器。你可以直接用Go的[]string来对应Scylla/Cassandra里的set<text>类型，下面是个简单的示例：

首先定义对应表结构的Go结构体：

import (
    "log"
    "github.com/gocql/gocql"
    "github.com/google/uuid"
)

type User struct {
    ID   uuid.UUID `cql:"id"`
    Tags []string  `cql:"tags"` // 这里的[]string直接对应数据库中的set<text>
}

然后是插入数据的代码：

// 初始化session（根据你的集群配置调整）
session, err := gocql.NewSession(gocql.SessionConfig{
    Consistency: gocql.Quorum,
    Hosts:       []string{"127.0.0.1"},
    Keyspace:    "my_keyspace",
})
if err != nil {
    log.Fatal(err)
}
defer session.Close()

// 构造数据并插入到数据库
newUser := User{
    ID:   uuid.New(),
    Tags: []string{"golang", "scylla", "database"},
}

err = session.Query(`INSERT INTO users (id, tags) VALUES (?, ?)`, newUser.ID, newUser.Tags).Exec()
if err != nil {
    log.Fatal(err)
}

查询的时候也一样，直接用[]string接收即可：

var retrievedUser User
err = session.Query(`SELECT id, tags FROM users WHERE id = ?`, newUser.ID).Scan(&retrievedUser.ID, &retrievedUser.Tags)
if err != nil {
    log.Fatal(err)
}

// 这里retrievedUser.Tags就是[]string类型，和数据库里的set<text>完全对应
fmt.Println("Retrieved tags:", retrievedUser.Tags)

2. 获取对应类型而非字节数组 #

你提到担心拿到的是字节数组，其实只要你用匹配的Go类型来接收查询结果，gocql的编解码器会自动帮你完成字节到对应类型的转换：

• 如果你查询的是set<text>列，用[]string接收，得到的就是切片类型，不是字节数组
• 如果你用MapScan方法获取结果，返回的map中对应key的值也会是[]string，比如：

var result map[string]interface{}
err = session.Query(`SELECT tags FROM users WHERE id = ?`, newUser.ID).MapScan(result)
if err != nil {
    log.Fatal(err)
}

// 类型断言后就能拿到[]string类型的tags
tags, ok := result["tags"].([]string)
if ok {
    fmt.Println("Tags from MapScan:", tags)
}

只有当你刻意用[]byte或者gocql.RawBytes类型去接收时，才会拿到原始的字节数组。所以只要类型匹配，完全不用担心这个问题。

什么时候需要自定义映射器？ #

只有在处理非常特殊的类型映射场景时（比如把自定义的Go结构体映射到Cassandra的UDT，或者一些小众的类型组合），才需要去实现gocql.Marshaler和gocql.Unmarshaler接口。像[]string转set<text>这种常见的场景，gocql的原生支持已经足够好用了。

内容的提问来源于stack exchange，提问作者Sculas