MongoDB
 sql >> Baza danych >  >> NoSQL >> MongoDB

Kodek MongoDB BSON nie jest używany podczas kodowania obiektu

Po kilku dniach badań znalazłem rozwiązanie.

DutyBlockCodec zależy od LocalDateCodec (którą stworzyłem) w celu kodowania/dekodowania. Ta zależność nie jest spełniona po prostu przez dodanie dwóch kodeków do tego samego rejestru kodeków. Rozwiązaniem jest przekazanie CodecRegistry obiekt zawierający kodeki, które DutyBlockCodec zależy od (np. CodecRegistry zawierające w sobie LocalDateCodec ) do DutyBlockCodec konstruktora, który jest przechowywany jako zmienna składowa. Aby użyć LocalDateCodec do kodowania używam EncoderContext.encodeWithChildContext() metody, przekazując kodek, program piszący i element do zakodowania. Dodatkowo piszę poszczególne pola zamiast pisać Document jako String (jak w moim oryginalnym kodzie). Tak więc DutyBlock kodek wygląda tak:

public class DutyBlockCodec implements Codec<DutyBlock> {
    private final CodecRegistry codecRegistry;

    public DutyBlockCodec(final CodecRegistry codecRegistry) {
        this.codecRegistry = codecRegistry;
    }

    @Override
    public void encode(BsonWriter writer, DutyBlock t, EncoderContext ec) {
        writer.writeStartDocument();
            Codec dateCodec = codecRegistry.get(LocalDate.class);
            writer.writeName("startDate");
            ec.encodeWithChildContext(dateCodec, writer, t.getStartDate());
            writer.writeName("endDate");
            ec.encodeWithChildContext(dateCodec, writer, t.getEndDate());
            writer.writeName("blockLength");
            writer.writeInt32(t.getBlockLength());
            writer.writeName("pointValue");
            writer.writeDouble(t.getPointValue());

            //Writing ArrayList of RAs
            writer.writeName("assigned");
            writer.writeStartArray();
                for (Ra ra : t.getRasOnDuty()) {
                    Codec raCodec = codecRegistry.get(Ra.class);
                    ec.encodeWithChildContext(raCodec, writer, ra);
                }
            writer.writeEndArray();
        writer.writeEndDocument();
    }

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

    @Override
    public DutyBlock decode(BsonReader reader, DecoderContext dc) {
        reader.readStartDocument();
            Codec<LocalDate> dateCodec = codecRegistry.get(LocalDate.class);
            reader.readName();
            LocalDate startDate = dateCodec.decode(reader, dc);
            reader.readName();
            LocalDate endDate = dateCodec.decode(reader, dc);
            reader.readName();
            int blockLength = reader.readInt32();
            reader.readName();
            double pointValue = reader.readDouble();

            //Reading ArrayList of RAs
            reader.readName();
            Codec<Ra> raCodec = codecRegistry.get(Ra.class);
            ArrayList<Ra> rasOnDuty = new ArrayList<>();
            reader.readStartArray();
                while (reader.readBsonType() != BsonType.END_OF_DOCUMENT) {
                    rasOnDuty.add(raCodec.decode(reader, dc));
                }
            reader.readEndArray();
        reader.readEndDocument();

        return new DutyBlock(startDate, endDate, blockLength, pointValue, rasOnDuty);
    }

}

DutyBlockCodec zależy od innego kodeka, dlatego wymaga CodecRegistry do przekazania na jego konstruktora. Chociaż wierzę, że możliwe jest utworzenie CodecRegistry z LocalDateCodec , a następnie przekaż to jako argument do DutyBlockCodec , a następnie utwórz kolejny CodecRegistry zawierające oba LocalDateCodec i DutyBlockCodec , jest to dość mylące, a MongoDB udostępnia funkcję CodecProvider aby ułatwić ten proces.

Korzystanie z CodecProvider interfejs, napisałem DutyBlockCodecProvider

public class DutyBlockCodecProvider implements CodecProvider {
    @Override
    public <T> Codec<T> get(Class<T> type, CodecRegistry cr) {
        if (type == DutyBlock.class) {
            return (Codec<T>) new DutyBlockCodec(cr);
        }
        return null;
    }
}

Dodałem te CodecProviders do klienta MongoDB za pomocą CodecRegistries.fromProviders() metoda.

CodecRegistry codecRegistry = CodecRegistries.fromRegistries(
            CodecRegistries.fromCodecs(new LocalDateCodec()),
            CodecRegistries.fromProviders(
                    new RaCodecProvider(),
                    new DutyBlockCodecProvider(),
                    new ScheduledDutyCodecProvider()),
            MongoClient.getDefaultCodecRegistry());  
    MongoClientOptions options = MongoClientOptions.builder()
            .codecRegistry(codecRegistry).build();
    mongoClient = new MongoClient(new ServerAddress(), options);
    db = mongoClient.getDatabase("DutySchedulerDB");

Mój kod źródłowy tego projektu można znaleźć pod adresem https://github.com/desrepair/DutyScheduler. Chętnie odpowiem na wszelkie pytania, jakie mogą mieć ludzie.



  1. Redis
  2.   
  3. MongoDB
  4.   
  5. Memcached
  6.   
  7. HBase
  8.   
  9. CouchDB
  1. Mongoose - wyszukiwanie poddokumentów według kryteriów

  2. Importuj dane CSV jako tablicę w MongoDB za pomocą mongoimport

  3. Sortowanie według wirtualnego pola w mongoDB (mongoose)

  4. Dlaczego moje identyfikatory MongooseJS ObjectId nie przechodzą testu równości?

  5. MongoDB $setIsSubset