通过socket编程掌握IO流 —— BIO

一、本次目标

  1. 编写最简单的1:1的server:client,感受socket通信编程;
  2. 修改client为多线程,模拟多个client请求server,观测server响应;
  3. 封装server处理client请求方法,另开线程处理;
  4. 处理线程加入线程池管理,减轻server负荷;

二、动手实践

1、编写配置接口,包含默认服务端地址、端口等

package com.cjt.io;

public interface Config {

  String DEFAULT_ENCODE = "UTF-8";

  String DEFAULT_ADDR = "127.0.0.1";

  int DEFAULT_PORT = 6666;
}

2、编写时间工具类DateUtil,方便输出日志打印

package com.cjt.io;

import java.text.SimpleDateFormat;
import java.util.Date;

public class DateUtil {

  private final static String DEFAULT_PATTERN = "HH:mm:ss";

  public static String getCurTimeStr(){
    SimpleDateFormat sdf = new SimpleDateFormat(DEFAULT_PATTERN);
    return sdf.format(new Date());
  }
}

3、编写服务端代码

package com.cjt.io.bio;

import com.cjt.io.Config;
import com.cjt.io.DateUtil;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.ServerSocket;
import java.net.Socket;

public class ServerBIO implements Config{

  private ServerSocket server;

  private ServerBIO(int port){
    try {
      server = new ServerSocket(port);
    } catch (IOException e) {
      e.printStackTrace();
    }
  }


  private void start() {
    System.out.println("[" + DateUtil.getCurTimeStr() + "]:服务端已经启动");
    try {
      while (true) {
        System.out.println("[" + DateUtil.getCurTimeStr() + "]:server循环获取client请求开始");
        // 阻塞,直到有客户端发起请求
        Socket socket = server.accept();
        System.out.println("[" + DateUtil.getCurTimeStr() + "]:有新的client连接啦");
        try (BufferedReader reader = new BufferedReader(new InputStreamReader(socket.getInputStream(), DEFAULT_ENCODE))){
          String line;
          StringBuilder builder = new StringBuilder();
          // 普通IO流会阻塞,所以这里的readLine()时间会根据客户端操作的时间而定
          while ((line = reader.readLine()) != null) {
            builder.append(line);
            builder.append(System.getProperty("line.separator"));
          }
          System.out.println("[" + DateUtil.getCurTimeStr() + "]:client传来消息");
          System.out.println(builder.toString());
        } catch (IOException e) {
          e.printStackTrace();
          break;
        }
      }
    } catch (IOException e) {
      e.printStackTrace();
    }
  }

  public static void main(String[] args){
    new ServerBIO(DEFAULT_PORT).start();
  }
}

4、编写客户端代码

package com.cjt.io.bio;

import com.cjt.io.Config;
import com.cjt.io.DateUtil;

import java.io.*;
import java.net.Socket;

public class ClientBIO implements Config {

  private String serverIp;

  private int port;

  private String msg;

  private int second;

  private ClientBIO(String serverIp, int port) {
    this.serverIp = serverIp;
    this.port = port;
  }

  private ClientBIO second(int second) {
    this.second = second;
    return this;
  }

  private ClientBIO msg(String msg) {
    this.msg = msg;
    return this;
  }

  private void sleep(int second) {
    try {
      Thread.sleep(second * 1000);
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
  }

  public void send() {
    try (Socket socket = new Socket(serverIp, port)) {
      System.out.println("[" + DateUtil.getCurTimeStr() + "]:连接server成功");
      BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(), DEFAULT_ENCODE));
      writer.write(msg);
      writer.flush();
      sleep(second);
      System.out.println("[" + DateUtil.getCurTimeStr() + "]:client写入消息" + msg);
    } catch (IOException e) {
      e.printStackTrace();
    }
  }


  public static void main(String[] args) {
    ClientBIO client_1 = new ClientBIO(DEFAULT_ADDR, DEFAULT_PORT).second(3).msg("我是第一个客户端");
    client_1.send();
  }
}

5、测试IO流的阻塞性

基于最简单的单线程方式我们分别实现了server和client的编写,这里不会详细阐述怎么读流,写数据,buffered流怎么用,主要是研究IO流的特性,更好地学会怎么处理高并发中的流处理。

首先启动ServerBIO,查看控制台:

[17:06:54]:服务端已经启动
[17:06:54]:server循环获取client请求开始

根据server的代码,可以发现目前就阻塞在server.accept()这个位置,然后我们再启动client的main方法,模拟了两个client一前一后发送请求,观察client的控制台:

[17:06:57]:连接server成功
[17:07:00]:client写入消息我是第一个客户端

同时切到server的控制台:

[17:06:54]:服务端已经启动
[17:06:54]:server循环获取client请求开始
[17:06:57]:有新的client连接啦
[17:07:00]:client传来消息
我是第一个客户端

[17:07:00]:server循环获取client请求开始

57秒client连接server成功,同时server打印“有新的client连接”,由于手动sleep的原因,直到00秒时client请求写入数据才完成,而server便一直阻塞在reader.readLine()这个位置,直接阻塞到00秒读取到client的消息,然后继续循环获取client阻塞在server.accept()这个位置。

这只是一个最最简单的socket通信模型,只有一个client请求,那么我们修改client实现Runnable接口,开启线程发送数据,模拟多个client并行访问server:

package com.cjt.io.bio;

import com.cjt.io.Config;
import com.cjt.io.DateUtil;

import java.io.*;
import java.net.Socket;

public class ClientBIO implements Config, Runnable {

  private String serverIp;

  private int port;

  private String msg;

  private int second;

  private ClientBIO(String serverIp, int port) {
    this.serverIp = serverIp;
    this.port = port;
  }

  private ClientBIO second(int second) {
    this.second = second;
    return this;
  }

  private ClientBIO msg(String msg) {
    this.msg = msg;
    return this;
  }

  private void sleep(int second) {
    try {
      Thread.sleep(second * 1000);
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
  }

  @Override
  public void run() {
    try (Socket socket = new Socket(serverIp, port)) {
      System.out.println("[" + DateUtil.getCurTimeStr() + "]:连接server成功");
      BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(), DEFAULT_ENCODE));
      writer.write(msg);
      writer.flush();
      sleep(second);
      System.out.println("[" + DateUtil.getCurTimeStr() + "]:client写入消息" + msg);
    } catch (IOException e) {
      e.printStackTrace();
    }
  }

  public static void main(String[] args) {
    ClientBIO client_1 = new ClientBIO(DEFAULT_ADDR, DEFAULT_PORT).second(3).msg("我是第一个客户端");
    new Thread(client_1).start();
    ClientBIO client_2 = new ClientBIO(DEFAULT_ADDR, DEFAULT_PORT).second(1).msg("我是第二个客户端");
    new Thread(client_2).start();
  }
}

运行client的main方法,观察client的控制台:

[17:21:44]:连接server成功
[17:21:44]:连接server成功
[17:21:45]:client写入消息我是第二个客户端
[17:21:47]:client写入消息我是第一个客户端

很明显多线程并行发送请求到server,由于client_2的sleep为1秒的原因,较client_1先打印,然后回到server的控制台:

[17:21:37]:服务端已经启动
[17:21:37]:server循环获取client请求开始
[17:21:44]:有新的client连接啦
[17:21:47]:client传来消息
我是第一个客户端

[17:21:47]:server循环获取client请求开始
[17:21:47]:有新的client连接啦
[17:21:47]:client传来消息
我是第二个客户端

[17:21:47]:server循环获取client请求开始

server这里控制台输出完全相反,44秒提示“有新的client连接啦”,这里是client_1,通过后面的msg也可以知道,然后在47秒读取完client_1后立即获取到client_2的连接,虽然client_2的sleep有1秒的时间,但是由于client_2的连接时间在44秒,所以数据早已准备就绪,但是server的线程阻塞在读取client_1了。

三、补充完善

1、封装server处理client请求,单独新开线程:

package com.cjt.io.bio;

import com.cjt.io.Config;
import com.cjt.io.DateUtil;

import java.io.*;
import java.net.Socket;

public class ClientHandler implements Runnable, Config {

  private Socket socket;

  ClientHandler(Socket socket) {
    this.socket = socket;
  }

  @Override
  public void run() {
    try {
      BufferedReader reader = new BufferedReader(new InputStreamReader(socket.getInputStream(), DEFAULT_ENCODE));
      String line;
      StringBuilder builder = new StringBuilder();
      // 普通IO流会阻塞,所以这里的readLine()时间会根据客户端操作的时间而定
      while ((line = reader.readLine()) != null) {
        builder.append(line);
        builder.append(System.getProperty("line.separator"));
      }
      System.out.println("[" + DateUtil.getCurTimeStr() + "]:client传来消息");
      System.out.println(builder.toString());
    } catch (IOException e) {
      e.printStackTrace();
    }
  }
}

同时修改server的start调用代码:

System.out.println("[" + DateUtil.getCurTimeStr() + "]:服务端已经启动");
while (true) {
  try {
    System.out.println("[" + DateUtil.getCurTimeStr() + "]:server循环获取client请求开始");
    // 阻塞,直到有客户端发起请求
    Socket socket = server.accept();
    System.out.println("[" + DateUtil.getCurTimeStr() + "]:有新的client连接啦");
    new Thread(new ClientHandler(socket)).start();
  } catch (IOException e) {
    e.printStackTrace();
    break;
  }
}

不需要修改client代码,直接启动server后,在启动client的main方法,观测这次server的控制台:

[17:37:53]:服务端已经启动
[17:37:53]:server循环获取client请求开始
[17:38:00]:有新的client连接啦
[17:38:00]:server循环获取client请求开始
[17:38:00]:有新的client连接啦
[17:38:00]:server循环获取client请求开始
[17:38:02]:client传来消息
我是第二个客户端

[17:38:04]:client传来消息
我是第一个客户端

对比上面,很明显server新开了client处理线程,在处理client消息方面不会造成阻塞了。server收到client连接后新开线程处理并立即轮询接下来连接的client,“我是第二个客户端”较先打印很好的印证了这点。

2、将ClientHandler处理线程加入到线程池

Java创建的线程属于宝贵的系统资源,若是在大量的client访问时,则会创建大量的client消息处理线程,这样必将导致系统性能急剧下降,最终宕掉。所以我们可以将client消息处理线程加入到线程池中进行管理:

··· ···
// 定长线程池,超出数量的线程会在队列中等待
private ExecutorService threadPool = Executors.newFixedThreadPool(20);
··· ···
// 创建一个新的线程加入到线程池中管理并立即提交执行
threadPool.execute(new Thread(new ClientHandler(socket)));
··· ···

这样就不用担心由于高并发产生巨大的线程从而使系统崩溃,运行结果与上面相差无异。这里就不再重复,感兴趣的可以自己试试。

3、增加server响应

修改ClientHandler,简单回应client(复述消息):

··· ···
// 在写之前必须关闭读
socket.shutdownInput();

BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(), DEFAULT_ENCODE));
writer.write(builder.toString());
writer.flush();

socket.close();
··· ···

那么client的发送消息也得需要读取server的响应了,修改client的run方法:

  @Override
  public void run() {
    try (Socket socket = new Socket(serverIp, port)) {
      System.out.println("[" + DateUtil.getCurTimeStr() + "]:连接server成功");
      BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(), DEFAULT_ENCODE));
      writer.write(msg);
      writer.flush();
      sleep(second);
      System.out.println("[" + DateUtil.getCurTimeStr() + "]:client写入消息");
      System.out.println(msg);

      // 在写之前必须关闭写
      socket.shutdownOutput();

      BufferedReader reader = new BufferedReader(new InputStreamReader(socket.getInputStream(), DEFAULT_ENCODE));
      StringBuilder builder = new StringBuilder();
      while ((msg = reader.readLine()) != null){
        builder.append(msg);
        builder.append(System.getProperty("line.separator"));
      }
      System.out.println("[" + DateUtil.getCurTimeStr() + "]:server传来消息");
      System.out.println(builder.toString());
    } catch (IOException e) {
      e.printStackTrace();
    }
  }

同样的先运行server后,再执行client的main方法,观测client的控制台:

[18:08:07]:连接server成功
[18:08:07]:连接server成功
[18:08:08]:client写入消息
我是第二个客户端
[18:08:08]:server传来消息
我是第二个客户端

Disconnected from the target VM, address: '127.0.0.1:61618', transport: 'socket'
[18:08:10]:client写入消息
我是第一个客户端
[18:08:10]:server传来消息
我是第一个客户端

这说明已经server成功做出了响应,并且client也成功读取到了server的消息。这里需要强调的是:

  1. jdk 1.7开始,try后面可直接初始化某些流(implements AutoCloseable),则会自动关闭;
  2. socket关闭后对应的输入/输出流也会被关闭;
  3. socket 一次I/O操作既有读也有写的话,中间一定要加上shutup流;

你可能感兴趣的:(通过socket编程掌握IO流 —— BIO)