BinaryHighwaterRowMarshaller.java

/*
 * Copyright 2013 Jive Software, Inc
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.jivesoftware.os.amza.service.storage.binary;

import com.jivesoftware.os.amza.api.AmzaInterner;
import com.jivesoftware.os.amza.api.filer.UIO;
import com.jivesoftware.os.amza.api.ring.RingMember;
import com.jivesoftware.os.amza.api.wal.WALHighwater;
import com.jivesoftware.os.amza.api.wal.WALHighwater.RingMemberHighwater;
import com.jivesoftware.os.amza.service.storage.HighwaterRowMarshaller;
import java.util.ArrayList;
import java.util.List;

public class BinaryHighwaterRowMarshaller implements HighwaterRowMarshaller<byte[]> {

    private final AmzaInterner amzaInterner;

    public BinaryHighwaterRowMarshaller(AmzaInterner amzaInterner) {
        this.amzaInterner = amzaInterner;
    }

    @Override
    public byte[] toBytes(WALHighwater highwater) throws Exception {
        if (highwater == null) {
            return null;
        }
        byte[] bytes = new byte[sizeInBytes(highwater)];
        int o = 0;
        for (RingMemberHighwater rmh : highwater.ringMemberHighwater) {
            int ringMemberSizeInBytes = rmh.ringMember.sizeInBytes();
            bytes[o] = (byte) 1; // hasMore
            o++;
            UIO.intBytes(ringMemberSizeInBytes, bytes, o);
            o += 4;
            rmh.ringMember.toBytes(bytes, o);
            o += ringMemberSizeInBytes;
            UIO.longBytes(rmh.transactionId, bytes, o);
            o += 8;
        }
        bytes[o] = (byte) 0; // hasMore
        return bytes;
    }

    @Override
    public int sizeInBytes(WALHighwater highwater) {
        int length = 0;
        for (RingMemberHighwater rmh : highwater.ringMemberHighwater) {
            length += 1 + 4 + rmh.ringMember.sizeInBytes() + 8;
        }
        length += 1;
        return length;
    }

    @Override
    public WALHighwater fromBytes(byte[] row) throws Exception {
        List<RingMemberHighwater> rmh = new ArrayList<>();
        int o = 0;
        while (row[o] != 0) { // hasMore
            o++;
            int ringMemberLength = UIO.bytesInt(row, o);
            o += 4;
            RingMember ringMember = amzaInterner.internRingMember(row, o, ringMemberLength);
            o += ringMemberLength;
            long transactionId = UIO.bytesLong(row, o);
            o += 8;
            rmh.add(new RingMemberHighwater(ringMember, transactionId));
        }
        return new WALHighwater(rmh);
    }
}