Ошибка при создании пользовательского mashler / unmashler для Bson.M mongodb - PullRequest
Новая
       62

Ошибка при создании пользовательского mashler / unmashler для Bson.M mongodb

0 голосов
/ 17 апреля 2020

Я получаю сообщение об ошибке WriteValueBytes can only write while positioned on a Element or Value but is positioned on a TopLevel при попытке создать пользовательский mashler / unmashler для bson.M

У меня есть пользовательский тип с именем TransactionId, который представляет UUID, который я хочу преобразовать в строку до сохранение в monbodb и также преобразование его обратно из строки при извлечении значения из mongodb.

Это код, который у меня есть до сих пор 

package main

import (
    "github.com/google/uuid"
    "github.com/pkg/errors"
    "go.mongodb.org/mongo-driver/bson/bsontype"
    "go.mongodb.org/mongo-driver/x/bsonx/bsoncore"
    "go.mongodb.org/mongo-driver/bson"
    "log"
)

// TransactionID is a UUID used to trace a batch of work which is being processed.
type TransactionID uuid.UUID

// String returns the transaction id as a string
func (id TransactionID) String() (result string, err error) {
    val, err := uuid.FromBytes(id[:])
    if err != nil {
        return result, errors.Wrapf(err, "cannot convert transaction ID %s to UUID", string(id[:]))
    }
    return val.String(), nil
}

func (id TransactionID) MarshalBSONValue() (bsontype.Type, []byte, error) {
    idString, err :=  id.String()
    if err != nil {
        return bsontype.String, nil, err
    }
    return bsontype.String, bsoncore.AppendString(nil, idString), nil
}


func (id *TransactionID) UnmarshalBSONValue(bsonType bsontype.Type, bytes []byte) error {
    uid, err := uuid.FromBytes(bytes)
    if err != nil {
        return err
    }

    *id = TransactionID(uid)
    return nil
}


func NewTransactionID() TransactionID {
    return TransactionID(uuid.New())
}


func main() {
    id := NewTransactionID()

    _, err :=  bson.Marshal(id)
    if err != nil {
        log.Fatal(err)
    }
}

Я получаю WriteValueBytes can only write while positioned on a Element or Value but is positioned on a TopLevel в шаг немаршала.

Ссылка: https://play.golang.org/p/_n7VpX-KIyP

1 Ответ

0 голосов
/ 20 апреля 2020

Я получаю, что WriteValueBytes может писать только тогда, когда находится на элементе или значении, но на верхнем уровне на немаршальном шаге.

Функция bson.Marshal () требуется параметр, который можно преобразовать в документ (т. Е. interface{}). Вот почему сообщение об ошибке относится к положению значения. то есть вы не можете иметь строку на верхнем уровне документа, она должна быть элементом документа. Если вам нужно маршалировать одно значение, вы должны использовать bson.MarshalValue () . 

id := NewTransactionID()
vtype, vdata, err := bson.MarshalValue(id)

Ниже приведен пример использования bson.Marshal() (добавление к вашему примеру): 

type Foo struct {
    ID TransactionID
}

func main() {
    id := NewTransactionID()
    foo, err :=  bson.Marshal(&Foo{ID:id})
    if err != nil {
        panic(err)
    }
}
...