ByteBufferUtils.java

Go to the documentation of this file.

/* *********************************************************************** *
 * project: org.matsim.*
 * ByteBufferUtils.java
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * copyright       : (C) 2009 by the members listed in the COPYING,        *
 *                   LICENSE and WARRANTY file.                            *
 * email           : info at matsim dot org                                *
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *   See also COPYING, LICENSE and WARRANTY file                           *
 *                                                                         *
 * *********************************************************************** */

package org.matsim.core.utils.misc;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutput;
import java.io.ObjectOutputStream;
import java.io.Serializable;
import java.nio.ByteBuffer;

public class ByteBufferUtils {

    public static void putString(final ByteBuffer buffer, final String string) {
        buffer.putInt(string.length());
        for (int i = 0; i < string.length(); i++) {
            buffer.putChar(string.charAt(i));
        }
    }

    public static String getString(final ByteBuffer buffer) {
        int length = buffer.getInt();
        char[] chBuffer = new char[length];
        for (int i = 0; i < length; i++) {
            chBuffer[i] = buffer.getChar();
        }
        return new String(chBuffer);
    }

    public static void putObject(final ByteBuffer buffer, Serializable o){
        try (ByteArrayOutputStream bos = new ByteArrayOutputStream()) {
            try (ObjectOutput oout = new ObjectOutputStream(bos)) {
                oout.writeObject(o);
                byte[] laneBytes = bos.toByteArray();
                buffer.putInt(laneBytes.length);
                for (int i = 0; i < laneBytes.length; i++) {
                    buffer.put(laneBytes[i]);
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static Object getObject(ByteBuffer buffer) {
        int length = buffer.getInt();
        byte[] bytes = new byte[length];
        for (int i = 0; i < length; i++) {
            bytes[i] = buffer.get();
        }
        try (ByteArrayInputStream bis = new ByteArrayInputStream(bytes);
                 ObjectInputStream oin = new ObjectInputStream(bis)) {
            return oin.readObject();
        } catch (IOException | ClassNotFoundException e) {
            e.printStackTrace();
            return null;
        }
    }

}