mongodb - 根据类型处理MongoDB错误

Question

我正在使用其ID获取帐户的数据。当前，当我进行此查询时，mongo-go-driver给出了一个错误，我想根据其类型来不同地处理此错误。例如，如果文档不存在，我想返回404，但是假设持有mongodb的实例掉了，在这种情况下，我想返回500。如何处理错误类型:

func (dao MongoDAO) Get(ctx *gin.Context, filter bson.M, entity interface{}) error {
    context, _ := context.WithTimeout(context.Background(), 5*time.Second)
    if err := dao.Collection.FindOne(context, filter).Decode(entity); err != nil {
        return err
    }
    return nil
}

Answer 1

您可以使用mongo-go-driver文档(https://pkg.go.dev/go.mongodb.org/mongo-driver/mongo?tab=doc#pkg-variables)中定义的Error变量。有了这些，您也许可以执行以下操作:

func (dao MongoDAO) Get(ctx *gin.Context, filter bson.M, entity interface{}) error {
    context, _ := context.WithTimeout(context.Background(), 5*time.Second)
    if err := dao.Collection.FindOne(context, filter).Decode(entity); err != nil {
        if err == mongo.ErrNoDocuments {
            // Return the 404
        }
        return err
    }
    return nil
}