Хранение массива с плавающей точкой в ​​MongoDB без преобразования в List - PullRequest
0 голосов
/ 25 апреля 2018

Я пытаюсь сохранить массив с плавающей точкой в ​​MongoDB.Вот пример того, что я пытался сделать:

    float floatArray[] = {1, 2, 3, 4 ..... , 10000};
    MongoCollection<Document> collection = database.getCollection(XMLUtils.getDatabaseCollectionName());

            Document doc = new Document("_id", edgeId)
                  .append("floatArray", floatArray);

            collection.insertOne(doc);

Но это выдает ошибку:

org.bson.codecs.configuration.CodecConfigurationException: Can't find a codec for class [F.

Я провел небольшое исследование и обнаружил следующее в одном из их билетов на JIRA:

В настоящее время класс Document поддерживает только List, а не собственный массив Java

И хотя преобразование в List будет работать, я буду иметь дело с огромными массивами (размер 10k +) и запуск циклов для преобразования их в списки снизит производительность, поскольку у меня будут миллионы массивов.

Есть ли другой способ хранения массивов Java в MongoDB, который не предусматривает преобразование их в список ?

РЕДАКТИРОВАТЬ: Обратите внимание, что вопрос требует решения для сохранения массива с плавающей запятой в MongoDB без преобразования его в список .Предложенные возможные дубликаты списков хранилищ вопросов вместо массивов.

1 Ответ

0 голосов
/ 31 июля 2018

Не отчаивайтесь, так как с большинством связанных с ИЛИ вещей в монго вы всегда можете написать себе кодек. Было бы полезно предоставить некоторые подробности о том, как вы в настоящее время обрабатываете свое сопоставление ИЛИ, но я нашел, что документы или реализация нативных кодеков (особенно DocumentCodec) очень полезны для выяснения того, как их реализовать и использовать.

Это кодек, который я использую для сохранения массива с плавающей точкой без использования коллекции. Обратите внимание, что при этом всегда создается массив float[] фиксированного размера. Если у вас есть другие потребности, вам все равно нужно найти способ определить и предварительно выделить необходимую память. Код ниже потерпит неудачу, если вы попытаетесь сохранить массивы большего размера, чем это ожидается. Во-вторых, afaik mongo не хранит числа с плавающей запятой, но удваивается, поэтому неизбежно произойдет некоторое преобразование.

AbstractCodec - наша собственная реализация часто используемых методов, например readValue(), writeObject() для хранения и чтения различных типов. Они были в значительной степени вдохновлены реализацией драйвера Java, так что это должно быть хорошей отправной точкой, если кодеки являются новыми для вас.

public class WaveformCodec extends AbstractCodec implements Codec<Waveform> {

    public WaveformCodec(CodecRegistry registry, BsonTypeClassMap bsonTypeClassMap) {
        super(registry, bsonTypeClassMap);
    }

    @Override
    public Waveform decode(BsonReader reader, DecoderContext decoderContext) {

        ObjectId id = null;
        float[] data = new float[5120];

        reader.readStartDocument();

        while (reader.readBsonType() != BsonType.END_OF_DOCUMENT) {
            String fieldName = reader.readName();
            switch (fieldName) {
                case "_id":
                    id = (ObjectId) readValue(reader, ObjectId.class, decoderContext);
                    break;
                case "data":
                    reader.readStartArray();
                    int i = 0;
                    while (reader.readBsonType() != BsonType.END_OF_DOCUMENT) {
                        data[i++] = (float) reader.readDouble();
                    }
                    reader.readEndArray();
                    break;
                default:
                    throw new RuntimeException("unknown field in WaveformCodec: " + fieldName);
            }
        }
        reader.readEndDocument();
        return new Waveform(id, data);
    }

    @Override
    public void encode(BsonWriter writer, Waveform waveform, EncoderContext encoderContext) {
        writer.writeStartDocument();

        writeObject(writer, "_id", ObjectId.class, waveform.getId(), encoderContext);

        writer.writeName("data");
        writer.writeStartArray();
        for (float value : waveform.getData()) {
            writer.writeDouble(value);
        }
        writer.writeEndArray();

        writer.writeEndDocument();
    }

    @Override
    public Class<Waveform> getEncoderClass() {
        return Waveform.class;
    }
}

И этот класс мы эффективно сохраняем:

public class Waveform {
    private ObjectId id;
    private float[] data;

    public Waveform(ObjectId id, float[] data) {
        this.id = id;
        this.data = data;
    }

    public ObjectId getId() {
        return id;
    }

    public void setId(ObjectId id) {
        this.id = id;
    }

    public float[] getData() {
        return data;
    }
}

Теперь вам просто нужно зарегистрировать свой кодек в драйвере, и вы сможете пользоваться набранными наборами.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...