欢迎大家搜索“小猴子的技术笔记”关注我的公众号，有问题可以及时和我交流。
对于一些平台业务对接除了http还有就是socket用的比较多了，特别软件是跟硬件设备做对接。在并发要求不是很高的时候，可以使用BIO的socket，因为它足够简单，足够方便。如果并发量很高的话，推荐使用NIO的socket，因为它能够支持较高的并发。
如果使用的是长连接的话，那么需要注意的是长连接的心跳保持，客户端的断线重连，报文格式的定义等。
如果可以的话，还是建议使用netty这个强大的框架，因为它对各种协议的封装还是很到位的。如果你熟悉它的API的话，那么这里真的强烈推荐netty（后期我会更新学习笔记）。
言归正传，接下来聊一聊BIO-短连接需要注意的事项：如果短连接作为客户端是比较简单的，只用connect->recv/send->close的流程接收数据发送数据，有业务请求来了，进行处理即可，下面给出了代码范例（以下代码均可以在gitee中找到：https://gitee.com/MonkeyBrothers/study/tree/master/springboot-netty/src/main/java/org/hry/socket/bio/shortlink）：
public class SocketClient {
public static void main(String[] args) throws IOException {
Socket socket = new Socket("localhost", 8888);
byte[] bytes = new byte[1024];
InputStream in = socket.getInputStream();
StringBuffer sb = new StringBuffer();
int len;
while ((len = in.read(bytes)) != -1) {
sb.append(new String(bytes, 0, len));
}
// 接收到信息，拿到接收的信息之后就可以处理了。
System.out.println("接收到消息：" + sb.toString());
// 关闭输入流，告诉服务端我读取数据完毕了
socket.shutdownInput();
// 如果需要给socketServer返回处理之后的数据，可以拿到输出流进行输出。
OutputStream out = socket.getOutputStream();
out.write("这里是处理完毕的数据".getBytes(StandardCharsets.UTF_8));
// 关闭输出流，告诉服务端我写完数据了
socket.shutdownOutput();
System.out.println("数据发送完毕");
socket.close();
}
}
连接上服务端，连接上之后，处理数据，处理完毕之后关闭这个socket进行释放资源就可以了。
BIO-服务端：如果是服务端那就 需要注意的东西比较多了，第一个就是并发量很大的很大的时候千万不要使用BIO 的socket更建议使用NIO或者netty框架来帮助你解决业务问题。首先来看一个类似一问一答式BIO的socket服务。
public class SocketServer {
public static void main(String[] args) throws IOException {
ServerSocket server = new ServerSocket(8888);
Socket socket = server.accept();
byte[] bytes = new byte[1024];
InputStream in = socket.getInputStream();
OutputStream out = socket.getOutputStream();
out.write("这里是服务端发送的数据".getBytes(StandardCharsets.UTF_8));
// 告诉客户端，我已经发送完一部分数据了。如果不调用的话，客户端是接受不到输入信号的
socket.shutdownOutput();
StringBuffer sb = new StringBuffer();
int len;
while ((len = in.read(bytes)) != -1) {
sb.append(new String(bytes, 0, len));
}
// 接收到信息，拿到接收的信息之后就可以处理了。
System.out.println("接收到消息：" + sb.toString());
socket.shutdownInput();
server.close();
}
}
这里是有问题的：就是只能有一个socket client来连接，如果是并发的业务是不能够满足需求的，因此需要为每一个socket启动一个线程。但是这样是很消耗资源的。所以并不是并发量特别大的话，还是可以的。再强调一下，超过一百的并发就不要使用BIO了！切记！！！
接下来模拟一下低并发的socket请求，看一下客户端和服务端是怎么处理的。首先客户端使用了while循环，这里“TimeUnit.SEONDS.sleep(1)”进行了休眠1秒，也就是每一秒都来一个请求（如果要每秒请求100的话，可以使用CountDownLatch进行控制，文章最后有实例）。
public class SocketClient {
public static void main(String[] args) throws Exception {
while (true) {
TimeUnit.SECONDS.sleep(1);
Socket socket = new Socket("localhost", 8888);
byte[] bytes = new byte[1024];
InputStream in = socket.getInputStream();
StringBuffer sb = new StringBuffer();
int len;
while ((len = in.read(bytes)) != -1) {
sb.append(new String(bytes, 0, len));
}
// 接收到信息，拿到接收的信息之后就可以处理了。
System.out.println("接收到消息：" + sb.toString());
// 关闭输入流，告诉服务端我读取数据完毕了
socket.shutdownInput();
// 如果需要给socketServer返回处理之后的数据，可以拿到输出流进行输出。
OutputStream out = socket.getOutputStream();
out.write("这里是处理完毕的数据".getBytes(StandardCharsets.UTF_8));
// 关闭输出流，告诉服务端我写完数据了
socket.shutdownOutput();
System.out.println("数据发送完毕");
socket.close();
}
}
}
对应的server服务就需要使用到线程池了，java为我们默认提供了一些线程池。
虽然我在刚开始写socket的时候使用了JDK提供的线程池，但是后来对线程池的深入研究之后发现一些潜在问题。因此还是推荐使用自定的线程池，这样一来可以对参数进行把控，还可以对线程池进行自定义设置名称。
public class SocketServer {
public static void main(String[] args) throws IOException {
Executor executor = Executors.newCachedThreadPool();
ServerSocket server = new ServerSocket(8888);
while(true) {
Socket socket = server.accept();
executor.execute(() ->{
try(InputStream in = socket.getInputStream();OutputStream out = socket.getOutputStream();){
byte[] bytes = new byte[1024];
out.write("这里是服务端发送的数据".getBytes(StandardCharsets.UTF_8));
// 告诉客户端，我已经发送完一部分数据了。如果不调用的话，客户端是接受不到结束信号的
socket.shutdownOutput();
StringBuffer sb = new StringBuffer();
int len;
while ((len = in.read(bytes)) != -1) {
sb.append(new String(bytes, 0, len));
}
// 接收到信息，拿到接收的信息之后就可以处理了。
System.out.println("接收到消息：" + sb.toString());
socket.shutdownInput();
socket.close();
} catch (Exception e) {
e.printStackTrace();
}
});
}
}
}
自定义的线程池，我的生产上在使用，具体的参数可以根据自己的业务进行设置。
public class SocketServer {
public static void main(String[] args) throws IOException {
// 自定义线程池
ThreadPoolExecutor executor = new ThreadPoolExecutor(
10,
10,
5L,
TimeUnit.SECONDS,
new LinkedBlockingQueue<>(10),
new MySocketThreadFactory("自定义socket线程池"),
new ThreadPoolExecutor.DiscardPolicy());
ServerSocket server = new ServerSocket(8888);
while(true) {
Socket socket = server.accept();
executor.execute(() ->{
try(InputStream in = socket.getInputStream();OutputStream out = socket.getOutputStream();){
byte[] bytes = new byte[1024];
out.write("这里是服务端发送的数据".getBytes(StandardCharsets.UTF_8));
// 告诉客户端，我已经发送完一部分数据了。如果不调用的话，客户端是接受不到结束信号的
socket.shutdownOutput();
StringBuffer sb = new StringBuffer();
int len;
while ((len = in.read(bytes)) != -1) {
sb.append(new String(bytes, 0, len));
}
// 接收到信息，拿到接收的信息之后就可以处理了。
System.out.println("接收到消息：" + sb.toString());
socket.shutdownInput();
socket.close();
} catch (Exception e) {
e.printStackTrace();
}
});
}
}
}
自定义线程工厂生产可用。
public class MySocketThreadFactory implements ThreadFactory {
/**
* 线程组
*/
private final ThreadGroup threadGroup;
/**
* 线程数量初始值，这里只是展示使用
*/
private final AtomicInteger threadNumber = new AtomicInteger(1);
/**
* 线程名前缀
*/
public final String namePrefix;
public MySocketThreadFactory(String name) {
SecurityManager s = System.getSecurityManager();
threadGroup = (s != null) ? s.getThreadGroup() : Thread.currentThread().getThreadGroup();
if (null == name || "".equals(name.trim())) name = "pool";
// 线程池数量初始值，这里只是展示使用
AtomicInteger poolNumber = new AtomicInteger(1);
namePrefix = name + "-" + poolNumber.getAndIncrement() + "-socket-thread-";
}
@Override
public Thread newThread(Runnable r) {
Thread t = new Thread(threadGroup, r, namePrefix + threadNumber.getAndIncrement(), 0);
if (t.isDaemon()) t.setDaemon(false);
if (t.getPriority() != Thread.NORM_PRIORITY) t.setPriority(Thread.NORM_PRIORITY);
return t;
}
}
以上的BIO的socket是生产上可以使用的，在并发量不高的情况下可以直接拿去使用。再一次强调，高并发请勿使用线程的socket！
这里给了并发的测试代码：
public class SocketClient {
public static void main(String[] args) throws Exception {
CountDownLatch downLatch = new CountDownLatch(1);
IntStream.range(0,25).forEach(v -> new Thread(() ->{
try {
downLatch.await();
Socket socket = new Socket("localhost", 8888);
byte[] bytes = new byte[1024];
InputStream in = socket.getInputStream();
StringBuffer sb = new StringBuffer();
int len;
while ((len = in.read(bytes)) != -1) {
sb.append(new String(bytes, 0, len));
}
// 接收到信息，拿到接收的信息之后就可以处理了。
System.out.println("接收到消息：" + sb.toString());
// 关闭输入流，告诉服务端我读取数据完毕了
socket.shutdownInput();
// 如果需要给socketServer返回处理之后的数据，可以拿到输出流进行输出。
OutputStream out = socket.getOutputStream();
out.write("这里是处理完毕的数据".getBytes(StandardCharsets.UTF_8));
// 关闭输出流，告诉服务端我写完数据了
socket.shutdownOutput();
System.out.println("数据发送完毕");
socket.close();
} catch (Exception e) {
e.printStackTrace();
}
}).start());
downLatch.countDown();
}
}
并发过来就不行，这个时候就需要调整线程池的设置了。由此可见BIO的socket的中等并发使用是很依赖线程池的。
对于socket阻塞式短连接的使用到此就结束了。需要注意的就是短连接是阻塞式的，因此如果一个业务有问题的话，那么就会阻塞其余的业务。因此开辟了线程池来进行处理这个问题，但是它不适用于高并发的场景，在高并发的场景下建议使用NIO的socket或者netty框架来帮助完成业务！
最后欢迎大家添加我的微信号，如果有问题可以及时和我交流。
7120
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
