com.eightkdata.mongowp.bson.netty.DefaultNettyBsonLowLevelReader Maven / Gradle / Ivy

Go to download
/*
 * This file is part of MongoWP.
 *
 * MongoWP is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * MongoWP is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with bson-netty. If not, see .
 *
 * Copyright (C) 2016 8Kdata.
 *
 */
package com.eightkdata.mongowp.bson.netty;

import com.eightkdata.mongowp.bson.BsonDocument.Entry;
import com.eightkdata.mongowp.bson.*;
import com.eightkdata.mongowp.bson.impl.*;
import com.eightkdata.mongowp.bson.netty.annotations.Loose;
import com.eightkdata.mongowp.bson.netty.annotations.ModifiesIndexes;
import com.google.common.primitives.UnsignedBytes;
import io.netty.buffer.ByteBuf;
import java.util.ArrayList;
import java.util.EnumSet;
import java.util.LinkedHashMap;
import javax.annotation.concurrent.Immutable;
import javax.inject.Inject;

/**
 *
 */
@Immutable
public class DefaultNettyBsonLowLevelReader extends NettyBsonLowLevelReader {

    @Inject
    public DefaultNettyBsonLowLevelReader(NettyStringReader stringReader) {
        super(stringReader);
    }

    @Override
    BsonArray readArray(@Loose @ModifiesIndexes ByteBuf byteBuf)
            throws NettyBsonReaderException {
        int length = byteBuf.readInt();
        int significantLenght = length - 4 - 1;

        ByteBuf significantSlice = byteBuf.readSlice(significantLenght);

        byte b = byteBuf.readByte();
        assert b == 0x00;

        ArrayList> list = new ArrayList<>();
        while (significantSlice.readableBytes() > 0) {
            list.add(readArrayEntry(significantSlice));
        }
        return new ListBsonArray(list);
    }

    @Override
    BsonBinary readBinary(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        int length = byteBuf.readInt();
        byte subtype = byteBuf.readByte();
        byte[] content = new byte[length];

        byteBuf.readBytes(content);

        return new ByteArrayBsonBinary(ParsingTools.getBinarySubtype(subtype), subtype, content);
    }

    @Override
    BsonDateTime readDateTime(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return new LongBsonDateTime(byteBuf.readLong());
    }

    @Override
    BsonDbPointer readDbPointer(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        String str = getStringReader().readString(byteBuf, false);

        byte[] bytes = new byte[12];
        byteBuf.readBytes(bytes);

        return new DefaultBsonDbPointer(str, new ByteArrayBsonObjectId(bytes));
    }

    @Override
    BsonValue readDeprecated(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return new StringBsonDeprecated(getStringReader().readString(byteBuf, false));
    }

    @Override
    BsonDocument readDocument(@Loose @ModifiesIndexes ByteBuf byteBuf)
            throws NettyBsonReaderException {
        int length = byteBuf.readInt();
        int significantLenght = length - 4 - 1;

        ByteBuf significantSlice = byteBuf.readSlice(significantLenght);

        byte b = byteBuf.readByte();
        assert b == 0x00;

        LinkedHashMap> values = new LinkedHashMap<>();
        while (significantSlice.readableBytes() > 0) {
            Entry entry = readDocumentEntry(significantSlice);
            values.put(entry.getKey(), entry.getValue());
        }
        return new MapBasedBsonDocument(values);
    }

    @Override
    BsonDocument readJavaScriptScope(@Loose @ModifiesIndexes ByteBuf byteBuf)
            throws NettyBsonReaderException {
        return readDocument(byteBuf);
    }

    @Override
    BsonDouble readDouble(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return PrimitiveBsonDouble.newInstance(byteBuf.readDouble());
    }

    @Override
    BsonBoolean readBoolean(@Loose @ModifiesIndexes ByteBuf byteBuf) throws NettyBsonReaderException {
        byte readByte = byteBuf.readByte();
        if (readByte == 0x00) {
            return FalseBsonBoolean.getInstance();
        }
        if (readByte == 0x01) {
            return TrueBsonBoolean.getInstance();
        }
        throw new NettyBsonReaderException("Unexpected boolean byte. 0x00 or "
                + "0x01 was expected, but 0x"
                + UnsignedBytes.toString(readByte, 16) + " was read");
    }

    @Override
    BsonInt32 readInt32(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return PrimitiveBsonInt32.newInstance(byteBuf.readInt());
    }

    @Override
    BsonInt64 readInt64(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return PrimitiveBsonInt64.newInstance(byteBuf.readLong());
    }

    @Override
    BsonJavaScript readJavaScript(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return new DefaultBsonJavaScript(getStringReader().readString(byteBuf, false));
    }

    @Override
    BsonJavaScriptWithScope readJavaScriptWithScope(
            @Loose @ModifiesIndexes ByteBuf byteBuf) throws NettyBsonReaderException {
        byteBuf.readInt();
        String js = getStringReader().readString(byteBuf, false);
        BsonDocument scope = readJavaScriptScope(byteBuf);

        return new DefaultBsonJavaScriptWithCode(js, scope);
    }

    @Override
    BsonMax readMax(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return SimpleBsonMax.getInstance();
    }

    @Override
    BsonMin readMin(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return SimpleBsonMin.getInstance();
    }

    @Override
    BsonNull readNull(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return SimpleBsonNull.getInstance();
    }

    @Override
    BsonObjectId readObjectId(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        byte[] bytes = new byte[12];
        byteBuf.readBytes(bytes);

        return new ByteArrayBsonObjectId(bytes);
    }

    @Override
    BsonRegex readRegex(@Loose @ModifiesIndexes ByteBuf byteBuf)
            throws NettyBsonReaderException {
        String pattern = getStringReader().readCString(byteBuf, false);
        String options = getStringReader().readCString(byteBuf, true);

        EnumSet optionsSet = ParsingTools.parseRegexOptions(options);

        return new DefaultBsonRegex(optionsSet, pattern);
    }

    @Override
    BsonString readString(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return new StringBsonString(getStringReader().readString(byteBuf, false));
    }

    @Override
    BsonTimestamp readTimestamp(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        int ordinal = byteBuf.readInt();
        int seconds = byteBuf.readInt();
        return new DefaultBsonTimestamp(seconds, ordinal);
    }

    @Override
    BsonUndefined readUndefined(@Loose @ModifiesIndexes ByteBuf byteBuf) {
        return SimpleBsonUndefined.getInstance();
    }

}