Java的TCP/IP编程学习--基于长度的成帧

基于长度的成帧

package framer;

import java.io.*;

/**
* @ClassName LengthFramer
* @Description TODO
* LengthFramer.java类实现了基于长度的成帧方法，适用于长度小于65535（216-1）
* 字节的消息。发送者首先给出指定消息的长度，并将长度信息以b1g-endian顺序存入两
* 个字节的整数中，再将这两个字节放在完整的消息内容前，连同消息一起写入输出流。
* 在接收端，我们使用DatalnputStream以读取整型的长度信息；readFully（）方法将阻塞
* 等待，直到给定的数组完全填满，这正是我们需要的。值得注意的是，使用这种成帧方
* 法，发送者不需要检查要成帧的消息内容，而只需要检查消息的长度是否超出了限制。
* @Author Cays
* @Date 2019/3/17 8:45
* @Version 1.0
**/
public class LengthFramer implements Framer{
public static final int MAXMESSAGELENGTH = 65535;
public static final int BYTEMASK = 0xff;
public static final int SHORTMASK = 0xffff;
public static final int BYTESHIFT=8;
private DataInputStream in;

/**
* 获取赖消息源的输入流，并将其包裹在一个DataInputStream
* @param in
* @throws IOException
*/
public LengthFramer(InputStream in) throws IOException{
this.in = new DataInputStream(in);
}

@Override
public void frameMsg(byte[] message, OutputStream out) throws IOException {
if (message.length>MAXMESSAGELENGTH){
throw new IOException("message too long");
}
/**
* 由于我们用的是长为两个字节的字段，因此消息的长度不能超过65535。
* （注意该值太大而不能存入一个short型整数中，
* 因此我们每次只向输出流写一个字节）
*/
out.write((message.length>>BYTESHIFT)&BYTEMASK);
out.write(message.length&BYTEMASK);
out.write(message);
out.flush();
}

@Override
public byte[] nextMsg() throws IOException {
int length;
try {
//readUnsignedShort（）方法读取两个字节，
// 将它们作为big-endian整数进行解释，并
//以int型整数返回它们的值。
length = in.readUnsignedShort();
}catch (EOFException e){
return null;
}
byte[] msg=new byte[length];
//阻塞等待，直到接收到足够的字节来填满指定的字节数组
in.readFully(msg);
return msg;
}
}