如何使用 Go 连接器将 Tarantool select 结果序列化为现成的结构

How to serialize Tarantool select results to ready structure with Go connector

请描述一种方法,如何将 select 结果序列化为 go-tarantool 连接器,使其能够访问字段 tuple.key1.key2

我们总是在我们的项目中使用 conn.*Typed() 查询来做到这一点。

首先,您需要在 Tarantool 中定义表示元组的结构。 然后你需要为它实现两个接口,msgpack.CustomDecoder and msgpack.CustomEncoder .

你应该可以这样做:

type Session struct {
    ID         string
    UserID     int64 
}

func (s *Session) EncodeMsgpack(e *msgpack.Encoder) error {
    if err := e.EncodeArrayLen(2); err != nil {
        return err
    }
  
    if err := e.EncodeString(s.ID); err != nil {
        return err
    }
  
    if err := e.EncodeInt64(s.UserID); err != nil {
        return err
    }

    return nil
}

func (s *Session) DecodeMsgpack(d *msgpack.Decoder) error {
    l, err := d.DecodeArrayLen()
    if err != nil {
        return err
    }
  
    decodedFields := 1
    if s.ID, err = d.DecodeString(); err != nil || decodedFields == l {
        return err
    }

    decodedFields++
    if s.UserID, err = d.DecodeInt64(); err != nil || decodedFields == l {
        return err
    }
 
    for i := 0; i < l-decodedFields; i++ {
        _ = d.Skip()
    }
  
    return nil
}

注意解码器。 它包含计数字段。这对于 non-breaking 迁移是必需的。

例如,如果 msgpack 数组的字段少于我们尝试解码的字段,则不会出现任何问题。

来自 select 查询的响应是 msgpack 元组的顺序数组,因此如果我们不跳过未知字段,则结构的下一个实例的解码将不会从 next 的开头开始元组。

那你可以试试查询:

func() ([]Session, error) {
    const userID = 822
    
    var sessions []Session
    err := conn.SelectTyped("session", "user", 0, 10, tarantool.IterEq, []interface{}{userID}, &resp)
    if err != nil {
        return nil, err
    }

    if len(resp) == 0 {
        return nil, nil
    }
  
    return sessions, nil
}

在我看来,这是最好的方法,因为反射、类型转换和类型断言最少,这些反射、类型转换和类型断言可能会在生产中因使用不当而导致恐慌。这也是一种更高效的方式。