在MongoDB中存储浮点数组而不转换为List

时间:2018-04-25 18:38:02

标签: java arrays mongodb list

我正在尝试在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 +)并运行for循环将它们转换为Lists会损害性能,因为我将拥有数百万个数组。

有没有其他方法可以在MongoDB中存储java数组不涉及将它们转换为List

编辑: 请注意,该问题要求解决方案在MongoDB中存储一系列浮动而不将其转换为列表。建议可能的重复问题存储列表而不是数组。

1 个答案:

答案 0 :(得分:0)

不要绝望,就像mongo中大多数与OR相关的事情一样,您始终可以编写自己的编解码器。提供一些有关您当前如何处理OR映射的详细信息将很有用,但是我发现文档或本机编解码器(尤其是DocumentCodec)的implementation对弄清楚如何实现和使用它们很有帮助。

这是我用来在不使用Collection的情况下保留浮点数组的编解码器。请注意,这总是创建一个固定大小的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;
    }
}

现在,您只需要在驱动程序中注册编解码器,就可以享受类型化的集合了。