在socket InputStream读取数据的问题记录一文中介绍通过input.available()检测是否有可读的字节,上线之后发现服务器load过高,通过jstack发现大量线程都lock在在SocketInputStream.available()上,见截图:
之前以为input.available()会很耗时,通过打点发现原来是调用input.available()过于频繁导致:
解决这个问题有一个方案,不要调用input.available(),通过bufferInputStream不停的尝试read(),通过研究netty的代码发现有更好的方案,先检测input.available(),如果数据没有准备好,则挂起不占用cpu,这种方案是通过PushbackInputStream实现的:
public String readLine(int length) throws IOException {
String buf = "";
PushbackInputStream input = new PushbackInputStream(socket.getInputStream(), 1);
int index = 0;
while (true) {
index = buf.indexOf("\r\n");
if (index >= 0) {
break;
}
int bytesToRead = Math.min(input.available(), length);
if(bytesToRead>0){
byte[] data = new byte[bytesToRead];
input.read(data);
String line = new String(data, Protocol.ENCODE);
buf += line;
}else{
int b = input.read(); //此操作会阻塞,直到有数据被读到
if (b < 0) {
break;
}
input.unread(b);
continue;
}
}
return buf.substring(0, index);
}
修复之后的情况:
后续发现还有问题: