Go+MongoDB票务系统：如何区分UserID与TicketID并解决解码错误

问题核心在于：你通过type UserID primitive.ObjectID定义的自定义类型是全新类型，MongoDB Go驱动无法自动将primitive.ObjectID解码到该自定义类型中，因此触发了cannot decode ObjectID into an array错误。解决思路是让自定义ID类型实现MongoDB驱动要求的BSON编解码接口。

解决方案：实现BSON编解码接口 #

为UserID和TicketID分别实现bson.ValueMarshaler和bson.ValueUnmarshaler接口，驱动就能正确处理这两个类型的编解码逻辑。

1. 定义自定义ID类型并实现编解码方法 #

import (
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/bson/primitive"
)

type UserID primitive.ObjectID
type TicketID primitive.ObjectID

// UserID的BSON解码实现
func (u *UserID) UnmarshalBSONValue(t bson.Type, data []byte) error {
    var oid primitive.ObjectID
    err := oid.UnmarshalBSONValue(t, data)
    if err != nil {
        return err
    }
    *u = UserID(oid)
    return nil
}

// UserID的BSON编码实现
func (u UserID) MarshalBSONValue() (bson.Type, []byte, error) {
    oid := primitive.ObjectID(u)
    return oid.MarshalBSONValue()
}

// TicketID的BSON解码实现
func (t *TicketID) UnmarshalBSONValue(tp bson.Type, data []byte) error {
    var oid primitive.ObjectID
    err := oid.UnmarshalBSONValue(tp, data)
    if err != nil {
        return err
    }
    *t = TicketID(oid)
    return nil
}

// TicketID的BSON编码实现
func (t TicketID) MarshalBSONValue() (bson.Type, []byte, error) {
    oid := primitive.ObjectID(t)
    return oid.MarshalBSONValue()
}

2. 补充实用工具方法（可选） #

为了更便捷使用自定义ID，可以添加生成新ID、字符串转换等方法：

// 生成新的UserID
func NewUserID() UserID {
    return UserID(primitive.NewObjectID())
}

// 将UserID转为字符串
func (u UserID) String() string {
    return primitive.ObjectID(u).String()
}

// 生成新的TicketID
func NewTicketID() TicketID {
    return TicketID(primitive.NewObjectID())
}

// 将TicketID转为字符串
func (t TicketID) String() string {
    return primitive.ObjectID(t).String()
}

3. 结构体与查询代码无需修改 #

你的User结构体和GetAllUsers函数保持原样即可，现在驱动能正确将_id字段解码到UserID类型：

type User struct {
    ID UserID `json:"id" bson:"_id"`
    // 其他业务字段...
}

func (coll *mongo.Collection) GetAllUsers(ctx context.Context) ([]User, error) {
    cursor, err := coll.Find(ctx, bson.D{})
    if err != nil {
        return []User{}, err
    }
    users := []User{}
    err = cursor.All(ctx, &users)
    if err != nil {
        return []User{}, err
    }
    return users, nil
}

额外说明 #

如果需要支持JSON编解码（比如接口返回JSON数据场景），可以为自定义ID类型实现json.Marshaler和json.Unmarshaler接口，逻辑参考BSON编解码，直接复用primitive.ObjectID的JSON处理方法即可。

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