WebSocket传输超过126字节数据的方法

之前用Java的Socket模拟HTTP传输实现了WebSocket的服务端，并成功与客户端握手
http://shuxiayeshou.blog.51cto.com/4452347/1762152

服务端和客户端握手成功后就可以开始交互了。客户端发来个信息，服务端接收，再反回个信息。
服务端向客户端发数据用byte[]型，且UTF-8编码，比如字符串就是 str.getBytes("UTF-8");
但是有个要求，在发送数据之前要发一个头数据。这个头数据当然也是byte[]型数组。
头数据包含啥呢？
比如这个头数据我们定义

byte[] pushHead=new byte[2];

长度为2的byte数组。第一个元素pushHead[0]存的是接收到来自客户端byte[]的第一个元素
比如客户端传来的数据我们用一个

byte[] buff=new byte[3072];
socket.getInputStream.read(buff);

那么

pushHead[0]=buff[0];

而头数据的第二个元素pushHead[1]存的是服务端要发给客户端数据的长度

但是有个问题，就是从Server传输数据超过126字节就不行，连接自动断开还报错

InputStream in=socket.getInputStream();
OutputStream out=socket.getOutputStream();
byte[] buff=new byte[3072];
int hasRead=0;
while((hasRead=in.read(buff)>0){
byte[] pushHead=new byte[2];    //在传递数据前要传递这个数据头
String x="要传输的数据......";
pushHead[0]=buff[0];            //数据头数组的第一个字节是接收到的数据的第一个字节
//第二个字节是要传输数据的字节数组长度转化成的字节
pushHead[1]=(byte)x.getBytes("UTF-8").length;
out.write(pushHead);            //先传递数据头
out.write(x.getBytes("UTF-8");  //再传递数据
}

上面的代码在传递的数据转化成的字节数组长度小于126时是OK的，但是>=126时就报错。

因为字节的范围是-128~127，所以传输数据长度大于125就报错

那就要另外的处理方法
传输跟据数据长度分三种情况，即有三种范围
小于126，那么上面的代码就OK了

大于125 小于65536 那么数据头的第二个字节存的是(byte)126，第三第四个字节存的是数据长度

大于65535 数据头第二个字节存的是(byte)127，后面再有8个字节存数据长度

这样说过于平面，看代码吧（代码也是平面的好不 (╯°Д°)╯（ ┻━┻ ）

InputStream in=socket.getInputStream();
OutputStream out=socket.getOutputStream();
byte[] buff=new byte[3072];
int hasRead=0;
while((hasRead=in.read(buff)>0){
byte[] pushHead;    //在传递数据前要传递这个数据头
String x="要传输的数据......";

int length=x.getBytes("UTF-8").length;

if(length<126){
pushHead=new byte[2];
pushHead[0]=buff[0];    //数据头数组的第一个字节是接收到的数据的第一个字节
//第二个字节是要传输数据的字节数组长度转化成的字节
pushHead[1]=(byte)x.getBytes("UTF-8").length;
out.write(pushHead);            //先传递数据头
}
else if(length>125 && length<65536){
pushHead=new byte[4];
pushHead[0]=buff[0];
pushHead[1]=(byte)126;
pushHead[2]=(byte)((length>>8) & 0xFF);
pushHead[3]=(byte)(length & 0xFF);
out.write(pushHead);
}
else if(length>65535){
pushHead=new byte[10];
pushHead[0]=buff[0];
pushHead[1]=(byte)127;
pushHead[2]=(byte)((length>>56) & 0xFF);
pushHead[3]=(byte)((length>>48) & 0xFF);
pushHead[4]=(byte)((length>>40) & 0xFF);
pushHead[5]=(byte)((length>>32) & 0xFF);
pushHead[6]=(byte)((length>>24) & 0xFF);
pushHead[7]=(byte)((length>>16) & 0xFF);
pushHead[8]=(byte)((length>>8) & 0xFF);
pushHead[9]=(byte)(length & 0xFF);
out.write(pushHead);
}

out.write(x.getBytes("UTF-8");  //再传递数据
}

这样大一些的数据也能一次传了，原先只能传不超过126字节数据时，要把数据分割分批传递，客户端接收后再组装或其它方式处理。不过目前只试了字符串的传输，还没试过图片，回头试试