org.hydracache.protocol.data.marshaller.DataMessageMarshaller.java Source code

Introduction

Here is the source code for org.hydracache.protocol.data.marshaller.DataMessageMarshaller.java

Source

/*
 * Copyright 2008 the original author or authors.
 *
 * 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 org.hydracache.protocol.data.marshaller;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;

import org.apache.commons.lang.Validate;
import org.hydracache.io.IoUtils;
import org.hydracache.io.BinaryMarshaller;
import org.hydracache.protocol.data.message.DataMessage;
import org.hydracache.server.data.versioning.Version;

/**
 * {@link BinaryMarshaller} used to handle {@link DataMessage} serialization
 * 
 * @author nzhu
 * 
 */
public class DataMessageMarshaller implements BinaryMarshaller<DataMessage> {

    private BinaryMarshaller<Version> versionMarshaller;

    /**
     * Constructor
     */
    public DataMessageMarshaller(BinaryMarshaller<Version> versionMarshaller) {
        this.versionMarshaller = versionMarshaller;
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.hydracache.io.BinaryMarshaller#readObject(java.io.DataInputStream)
     */
    @Override
    public DataMessage readObject(DataInputStream dataIn) throws IOException {
        DataMessage msg = new DataMessage();

        msg.setVersion(versionMarshaller.readObject(dataIn));

        byte[] bytes = IoUtils.readRemainingBytes(dataIn);

        msg.setBlob(bytes);

        return msg;
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.hydracache.io.BinaryMarshaller#writeObject(java.lang.Object,
     * java.io.DataOutputStream)
     */
    @Override
    public void writeObject(DataMessage msg, DataOutputStream dataOut) throws IOException {
        Validate.notNull(msg.getBlob(), "Data message is empty");

        versionMarshaller.writeObject(msg.getVersion(), dataOut);

        dataOut.write(msg.getBlob());
    }

}