(Socket、ServerSocket、多线程)模拟文件上传至服务器

  之前整理了一篇文件上传下载的博客和线程池相关文章，这次搞一篇使用Socket和ServerSocket结合线程池创建多线程来模拟上传文件至指定服务器。

Client客户端

  此篇文章在创建demo的时候不需要额外引入其他依赖，直接Next创建一个Client客户端就可以了。

  使用Socket套接字来创建客户端

public class UploadClient {

	public static void main(String[] args) throws IOException {
		//创建一个指定host和port的客户端套接字
		Socket socket = new Socket("127.0.0.1", 8083);
		//指定本地文件File对象
		File file = new File("E:" + File.separator + "1.jpg");
		//创建本地FileInputStream流对象来读取文件
		FileInputStream fileInputStream = new FileInputStream(file);
		//通过socket获取outputStream流
		OutputStream outputStream = socket.getOutputStream();
		//使用outputStream的write方法读取本地文件
		byte[] bytes = new byte[(int) file.length()];
		int len = 0;
		while ((len = fileInputStream.read(bytes)) != -1) {
			outputStream.write(bytes,0,len);
		}
		//使用socket的shutdownOutStream()方法禁用此套接字的输出流
		socket.shutdownOutput();
		//读取服务器返回的信息
		InputStream inputStream = socket.getInputStream();
		while ((len = inputStream.read(bytes)) != -1) {
			//将读取的服务器信息打印出来
			System.out.println(new String(bytes, 0, len));
		}

		//关闭流
		fileInputStream.close();
		socket.close();
	}
}

Server服务器端

   同样新建一个项目，在项目中使用ServerSocket套接字来创建Server服务器端：

public class UploadServer {

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

		//1.创建一个服务器对象
		ServerSocket serverSocket = new ServerSocket(8083);
		//2.监听服务器，看是否有客户端连接，这里使用while死循环，一直开启服务器端监听
		while (true) {
			Socket socket = serverSocket.accept();
			ThreadPoolExecutor threadPoolExecutor;
			threadPoolExecutor = new ThreadPoolExecutor(5, 10, 500, TimeUnit.MILLISECONDS, new ArrayBlockingQueue<Runnable>(5));
			threadPoolExecutor.execute(new Runnable() {
				@Override
				public void run() {
					FileOutputStream fileOutputStream = null;
					//虽然已经声明抛出了异常，但是因为run()方法没有抛异常，这里重写之后也不能抛异常，只能捕获异常
					try {
						//3.获取网络流inputStream对象
						InputStream inputStream = socket.getInputStream();
						//4.创建本地outputStream对象
						File folder = new File("D:" + File.separator + "upload");
						if (!folder.exists()) {
							folder.mkdirs();
						}
						fileOutputStream = new FileOutputStream(folder + File.separator + System.currentTimeMillis() + ".jpg");
						//5.将上传的文件读出来
						int len = 0;
						byte[] bytes = new byte[1024];
						while ((len = inputStream.read(bytes)) != -1) {
							//6.将读出来的数据再写到服务器本地磁盘中
							fileOutputStream.write(bytes, 0, len);
						}
						//7.写完之后，再给客户端一个回应。
						OutputStream outputStream = socket.getOutputStream();
						outputStream.write("服务器响应：上传成功！".getBytes());
						//8.释放资源
						fileOutputStream.close();
						socket.close();
					} catch (IOException e) {
						e.printStackTrace();
					}
				}
			});
		}
	}
}

测试

   先启动Server服务器端，然后再启动Client客户端，可以从客户端上面看到从服务器端返回的响应信息。再看D盘上的upload文件夹中，确实有了一张.jpg格式的图片。

   通过上面代码，就可以简单模拟多线程模拟文件上传至服务器端。这篇对于案例中的FileInputStream和InputStream之间的区分和FileOutputStream和OutputStream之间的区分，我是简单的理解为带File的就是直接对接磁盘的本地流，（栗子：比如你要从C盘中读取一个txt文件，那就先使用FileInputStream的read()方法读取出来，再使用OutputStream（不带file）的网络流对象的write()方法写到服指定服务器上去）。