java Socket输出不稳定问题

java socket编程，当建立连接后，如果连续的几个流去输出到server端，或者建立连接时，就紧接着去写数据，往往会产生 有的第一次的数据可以输出，但是第二次的数据却不能输出。这种情况并不是固定的。第一次运行可以，第二次或许就不可以。
这里引用一下比人的文章。
原文：http://blog.sina.com.cn/s/blog_616e189f0100s3px.html
Socket缓冲区探讨

本文主要探讨java网络套接字传输模型，并对如何将NIO应用于服务端，提高服务端的运行能力和降低服务负载。

1.1 socket套接字缓冲区

Java提供了便捷的网络编程模式，尤其在套接字中，直接提供了与网络进行沟通的输入和输出流，用户对网络的操作就如同对文件操作一样简便。在客户端与服务端建立Socket连接后，客户端与服务端间的写入和写出流也同时被建立，此时即可向流中写入数据，也可以从流中读取数据。在对数据流进行操作时，很多人都会误以为，客户端和服务端的read和write应当是对应的，即：客户端调用一次写入，服务端必然调用了一次写出，而且写入和写出的字节数应当是对应的。为了解释上面的误解，我们提供了Demo-1的示例。

在Demo-1中服务端先向客户端输出了两次，之后刷新了输出缓冲区。客户端先向服务端输出了一次，然后刷新输出缓冲，之后调用了一次接收操作。从Demo-1源码以及后面提供的可能出现的结果可以看出，服务端和客户端的输入和输出并不是对应的，有时一次接收操作可以接收对方几次发过来的信息，并且不是每次输出操作对方都需要接收处理。当然了Demo-1的代码是一种错误的编写方式，没有任何一个程序员希望编写这样的代码。

Demo-1

package com.upc.upcgrid.guan.chapter02;



import java.io.IOException;

import java.net.ServerSocket;

import java.net.Socket;

import java.net.UnknownHostException;



import org.junit.Test;



public class SocketWriteTest {

public static final int PORT = 12123;

public static final int BUFFER_SIZE = 1024;

//服务端代码

@Test

public void server() throws IOException, InterruptedException{

ServerSocket ss = new ServerSocket(PORT);

while(true)

{

Socket s = ss.accept();

//这里向网络进行两次写入

s.getOutputStream().write("hello ".getBytes());

s.getOutputStream().write("guanxinquan ".getBytes());

s.getOutputStream().flush();

s.close();

}

}



//客户端代码

@Test

public void client() throws UnknownHostException, IOException{

byte[] buffer;

Socket s = new Socket("localhost",PORT);//创建socket连接

s.getOutputStream().write(new byte[BUFFER_SIZE]);

s.getOutputStream().flush();

int i = s.getInputStream().read(buffer = new byte[BUFFER_SIZE]);

System.out.println(new String(buffer,0,i));



}

}

Demo-1可能输出的结果：

结果1：

hello

结果2：

hello guanxinquan

为了深入理解网络发送数据的流程，我们需要对Socket的数据缓冲区有所了解。在创建Socket后，系统会为新创建的套接字分配缓冲区空间。这时套接字已经具有了输入缓冲区和输出缓冲区。可以通过Demo-2中的方式来获取和设置缓冲区的大小。缓冲区大小需要根据具体情况进行设置，一般要低于64K（TCP能够指定的最大负重载数据量，TCP的窗口大小是由16bit来确定的），增大缓冲区可以增大网络I/O的性能，而减少缓冲区有助于减少传入数据的backlog（就是缓冲长度，因此提高响应速度）。对于Socket和SeverSocket如果需要指定缓冲区大小，必须在连接之前完成缓冲区的设定。

Demo-2

package com.upc.upcgrid.guan.chapter02;



import java.net.Socket;

import java.net.SocketException;



public class SocketBufferTest {

public static void main(String[] args) throws SocketException {

//创建一个socket

Socket socket = new Socket();

//输出缓冲区大小

System.out.println(socket.getSendBufferSize());

System.out.println(socket.getReceiveBufferSize());

//重置缓冲区大小

socket.setSendBufferSize(1024*32);

socket.setReceiveBufferSize(1024*32);

//再次输出缓冲区大小

System.out.println(socket.getSendBufferSize());

System.out.println(socket.getReceiveBufferSize());

}

}

Demo-2的输出：

8192

8192

32768

32768

了解了Socket缓冲区的概念后，需要探讨一下Socket的可写状态和可读状态。当输出缓冲区未满时，Socket是可写的（注意，不是对方启用接收操作后，本地才能可写，这是错误的理解），因此，当套接字被建立时，即处于可写如的状态。对于可读，则是指缓冲区中有接收到的数据，并且这些数据未完成处理。在socket创建时，并不处于可读状态，仅当连接的另一方向本套接字的通道写入数据后，本套接字方能处于可读状态（注意，如果对方套接字已经关闭，那么本地套接字将处于可读状态，并且每次调用read后，返回的都是-1）。

现在应用前面的讨论，重新分析一下Demo-1的执行流程，服务端与客户端建立连接后，服务器端先向缓冲区写入两条信息，在第一条信息写入时，缓冲区并未写满，因此在第二条信息输入时，第一条信息很可能还未发送，因此两条信息可能同时被传送到客户端。另一方面，如果在第二条信息写入时，第一条已经发送出去，那么客户端的接收操作仅会获得第一条信息，因为客户端没有继续接收的操作，因此第二条信息在缓冲区中，将不会被读取，当socket关闭时，缓冲区将被释放，未被读取的数据也就变的无效了。如果对方的socket已经关闭，本地再次调用读取方法，则读取方法直接返回-1，表示读到了文件的尾部。

对于缓冲区空间的设定，要根据具体情况来定，如果存在大量的长信息（比如文件传输），将缓冲区定义的大些，可能更好的利用网络资源，如果更多的是短信息（比如聊天消息），使用小的缓冲区可能更好些，这样刷新的速度会更快。一般系统默认的缓冲大小是8*1024。除非对自己处理的情况很清晰，否则请不要随意更改这个设置。

由于可读状态是在对方写入数据后或socket关闭时才能出现，因此如果客户端和服务端都停留在read时，如果没有任何一方，向对方写入数据，这将会产生一个死锁。

此外，在本地接收操作发起之前，很可能接收缓冲区中已经有数据了，这是一种异步。不要误以为，本地调用接收操作后，对方才会发送数据，实际数据何时到达，本地不能做出任何假设。

如果想要将多条输入的信息区分开，可以使用一些技巧，在文件操作中使用-1表示EOF，就是文件的结束，在网络传输中，也可以使用-1表示一条传输语句的结束。Demo-3中给出了一个读取和写入操作，在客户端和服务端对称的使用这两个类，可以将每一条信息分析出来。Demo-3中并不是将网络的传输同步，而是分析出缓冲中的数据，将以-1为结尾进行数据划分。如果写聊天程序可以使用类似的模式。

Demo-3

package com.upc.upcgrid.guan.chapter02;



import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.net.ServerSocket;

import java.net.Socket;

import java.net.UnknownHostException;

import java.nio.ByteBuffer;

import java.util.ArrayList;

import java.util.List;



import org.junit.Test;



public class SocketWriteTest {

public static final int PORT = 12123;

public static final int BUFFER_SIZE = 1024;



//读取一条传入的，以-1为结尾的数据

public class ReadDatas{

//数据临时缓冲用

private List buffers = new ArrayList();

private Socket socket;//数据的来源

public ReadDatas(Socket socket) throws IOException {

this.socket = socket;

}



public void read() throws IOException

{

buffers.clear();//清空上次的读取状态

InputStream in = socket.getInputStream();//获取输入流

int k = 0;

byte r = 0;

while(true)

{

ByteBuffer buffer = ByteBuffer.allocate(BUFFER_SIZE);//新分配一段数据区

//如果新数据区未满，并且没有读到-1，则继续读取

for(k = 0 ; k < BUFFER_SIZE ; k++)

{

r = (byte) in.read();//读取一个数据

if(r != -1)//数据不为-1，简单放入缓冲区

buffer.put(r);

else{//读取了一个-1，表示这条信息结束

buffer.flip();//翻转缓冲，以备读取操作

buffers.add(buffer);//将当前的buffer添加到缓冲列表

return;

}

}

buffers.add(buffer);//由于缓冲不足，直接将填满的缓冲放入缓冲列表



}



}





public String getAsString()

{

StringBuffer str = new StringBuffer();

for(ByteBuffer buffer: buffers)//遍历缓冲列表

{

str.append(new String(buffer.array(),0,buffer.limit()));//组织字符串

}

return str.toString();//返回生成的字符串

}

}



//将一条信息写出给接收端

public class WriteDatas{

public Socket socket;//数据接收端

public WriteDatas(Socket socket,ByteBuffer[] buffers) throws IOException {

this.socket = socket;

write(buffers);

}



public WriteDatas(Socket socket) {

this.socket = socket;

}



public void write(ByteBuffer[] buffers) throws IOException

{

OutputStream out = socket.getOutputStream();//获取输出流

for(ByteBuffer buffer:buffers)

{

out.write(buffer.array());//将数据输出到缓冲区

}

out.write(new byte[]{-1});//输出终结符

out.flush();//刷新缓冲区



}



}



//服务端代码

@Test

public void server() throws IOException, InterruptedException{

ServerSocket ss = new ServerSocket(PORT);

while(true)

{

Socket s = ss.accept();



//从网络连续读取两条信息

ReadDatas read = new ReadDatas(s);

read.read();

System.out.println(read.getAsString());

read.read();

System.out.println(read.getAsString());

//向网络中输出一条信息

WriteDatas write = new WriteDatas(s);

write.write(new ByteBuffer[]{ByteBuffer.wrap("welcome to us ! ".getBytes())});

//关闭套接字

s.close();



}

}





//客户端代码

@Test

public void client() throws UnknownHostException, IOException{

Socket s = new Socket("localhost",PORT);//创建socket连接

//连续向服务端写入两条信息

WriteDatas write = new WriteDatas(s,new ByteBuffer[]{ByteBuffer.wrap("ni hao guan xin quan ! ".getBytes())} );

write.write(new ByteBuffer[]{ByteBuffer.wrap("let's study java network !".getBytes())});

//从服务端读取一条信息

ReadDatas read = new ReadDatas(s);

read.read();

System.out.println(read.getAsString());

//关闭套接字

s.close();

}

}

在Demo-3中的这种消息处理方式过于复杂，需要理解java底层的缓冲区的知识，还需要编程人员完成消息的组合（在消息末尾添加-1），在Java中可以使用一种简单的方式完成上述的操作，就是使用java DataInputStream和DataOutputStream提供的方法。Demo-4给出了使用java相关流类完成同步的消息的方法（估计他们与我们Demo-3使用的方式是相似的）。你可以查阅java其它API，可以找到其他的方式。

Demo-4

package com.upc.upcgrid.guan.chapter02;



import java.io.DataInputStream;

import java.io.DataOutputStream;

import java.io.IOException;

import java.net.ServerSocket;

import java.net.Socket;

import java.net.UnknownHostException;





import org.junit.Test;



public class SocketDataStream {

public static final int PORT = 12123;

@Test

public void server() throws IOException

{

ServerSocket ss = new ServerSocket(PORT);

while(true)

{

Socket s = ss.accept();

DataInputStream in = new DataInputStream(s.getInputStream());

DataOutputStream out = new DataOutputStream(s.getOutputStream());



out.writeUTF("hello guan xin quan ! ");

out.writeUTF("let's study java togethor! ");



System.out.println(in.readUTF());

s.close();

}

}



@Test

public void client() throws UnknownHostException, IOException

{

Socket s = new Socket("localhost",PORT);

DataInputStream in = new DataInputStream(s.getInputStream());

DataOutputStream out = new DataOutputStream(s.getOutputStream());



System.out.println(in.readUTF());

System.out.println(in.readUTF());

out.writeUTF("welcome to java net world ! ");

s.close();

}

}



简单总结：

上面主要介绍了java Socket通信的缓冲区机制，并通过几个示例让您对java Socket的工作原理有了简单了解。这里需要注意的是可读状态和可写状态，因为这两个概念将对下一节的内容理解至关重要。下一节将描述java NIO提高服务端的并发性。