Как эффективно конвертировать BSON в JSON с Mongo-Go-драйвера? - PullRequest
Новая
       25

Как эффективно конвертировать BSON в JSON с Mongo-Go-драйвера?

0 голосов
/ 18 июня 2019

Я хочу преобразовать bson в mongo-go-driver в json эффективно.

Я должен позаботиться о обработке NaN, потому что json.Marshal не удастся, если существует NaNв данных.

Например, я хочу преобразовать данные ниже bson в json. 

b, _ := bson.Marshal(bson.M{"a": []interface{}{math.NaN(), 0, 1}})
// How to convert b to json?

Сбой ниже. 

// decode
var decodedBson bson.M
bson.Unmarshal(b, &decodedBson)
_, err := json.Marshal(decodedBson)
if err != nil {
    panic(err) // it will be invoked
    // panic: json: unsupported value: NaN
}

1 Ответ

2 голосов
/ 18 июня 2019

Если вы знаете структуру вашего BSON, вы можете создать пользовательский тип, который реализует интерфейсы json.Marshaler и json.Unmarshaler и обрабатывает NaN по вашему желанию. Пример: 

type maybeNaN struct{
    isNan  bool
    number float64
}

func (n maybeNaN) MarshalJSON() ([]byte, error) {
    if n.isNan {
        return []byte("null"), nil // Or whatever you want here
    }
    return json.Marshal(n.number)
}

func (n *maybeNan) UnmarshalJSON(p []byte) error {
    if string(p) == "NaN" {
        n.isNan = true
        return nil
    }
    return json.Unmarshal(p, &n.number)
}

type myStruct struct {
    someNumber maybeNaN `json:"someNumber" bson:"someNumber"`
    /* ... */
}

Если у вас произвольная структура вашего BSON, единственный вариант - обойти структуру, используя отражение, и преобразовать любые вхождения NaN в тип (возможно, пользовательский тип, как описано выше)

...