nioutils.java

利用java实现的多人游戏的服务器系统

package com.hypefiend.javagamebook.common;

import java.nio.*;
import java.nio.channels.*;

/**
 * NIOUtils.java
 *
 * Misc utility functions to simplify dealing w/NIO channels and buffers
 *
 * @author <a href="mailto:[e-mail]">bret barker</a>
 * @version 1.0
 */
public class NIOUtils {

    /** 
     * first, writes the header, then the 
     * event into the given ByteBuffer
     * in preparation for the channel write
     */
    public static void prepBuffer(GameEvent event, ByteBuffer writeBuffer) {
	// write header
	writeBuffer.clear();
	writeBuffer.putInt(0); // todo: clientId
	if (event.getGameName() != null)
	    writeBuffer.putInt(event.getGameName().hashCode()); 
	else
	    writeBuffer.putInt(0); 
	int sizePos = writeBuffer.position();
	writeBuffer.putInt(0);// placeholder for payload size
	// write event
	int payloadSize = event.write(writeBuffer);

	// insert the payload size in the placeholder spot 	
	writeBuffer.putInt(sizePos, payloadSize); 
	
	// prepare for a channel.write
	writeBuffer.flip(); 
    }
    

    /** 
     * write the contents of a ByteBuffer to the given SocketChannel
     */
    public static void channelWrite(SocketChannel channel, ByteBuffer writeBuffer) {
	long nbytes = 0;
	long toWrite = writeBuffer.remaining();

	// loop on the channel.write() call since it will not necessarily
	// write all bytes in one shot
	try {
	    while (nbytes != toWrite) {
		nbytes += channel.write(writeBuffer);
		
		try {
		    Thread.sleep(Globals.CHANNEL_WRITE_SLEEP);
		}
		catch (InterruptedException e) {}
	    }
	}
	catch (ClosedChannelException cce) {
	}
	catch (Exception e) {
	} 
	
	// get ready for another write if needed
	writeBuffer.rewind();
    }

    /**
     * write a String to a ByteBuffer, 
     * prepended with a short integer representing the length of the String
     */
    public static void putStr(ByteBuffer buff, String str) {
	if (str == null) {
	    buff.putShort((short)0);
	}
	else {
	    buff.putShort((short)str.length());
	    buff.put(str.getBytes());
	}
    }

    /**
     * read a String from a ByteBuffer 
     * that was written w/the putStr method
     */
    public static String getStr(ByteBuffer buff) {
	short len = buff.getShort();
	if (len == 0) {
	    return null;
	}
	else {
	    byte[] b = new byte[len];
	    buff.get(b);
	    return new String(b);
	}
    }


}