Java简单的TCP通信模拟文件上传

一时失言乱红尘 2021-11-26 13:40 255阅读 0赞

## 概述 ##

客户端上传文件到服务器端，服务器接收文件，并且传回“上传成功”字样。

## 客户端 ##

首先，客户端要绑定服务器端的地址以及端口号。  
要想将文件（此处是图片）上传到服务器，那么需要先将文件读入到内存中来，使用FileInputStream  
用Socket的网络输出流接受文件，写到服务器上去。  
同时还要接受来自服务器的文字“上传成功”,因此需要过去socket的网络输入流  
最后释放资源。

代码：

import java.io.*;
    import java.net.Socket;
    
    public class TCPFileClient {
        public static void main(String[] args) throws IOException {
            FileInputStream fis = new FileInputStream(new File("OutData/img.png"));
            Socket socket = new Socket("127.0.0.1",8888);
            OutputStream os = socket.getOutputStream();
            int len = 0;
            byte[] bytes = new byte[1024];
            while((len = fis.read(bytes))!=-1){
                os.write(bytes,0,len);
            }
            os.flush();
            socket.shutdownOutput();
            InputStream is = socket.getInputStream();
            int l = 0;
            byte[] bytes1 = new byte[1024];
            while((l = is.read(bytes1))!=-1){
                System.out.println(new String(bytes1,0,l));
            }
            os.close();
            fis.close();
            is.close();
            socket.close();
        }
    
    }

另外，需要有shudown来结束，因为while循环中的条件不会得到满足。

## 服务器端 ##

服务器要绑定一个端口  
服务器要一直开启，所以while循环的条件应该是true  
接受客户端传递的文件  
为了提高并发度，我们可以创建线程  
除此之外，给上传的文件命名，为了避免重复，可以用随机数  
当文件上传成功之后，还要给客户端传递一个“上传成功”的字样

代码：

import java.io.FileOutputStream;
    import java.io.IOException;
    import java.io.InputStream;
    import java.net.ServerSocket;
    import java.net.Socket;
    import java.util.Random;
    
    public class TCPFIleServer {
        public static void main(String[] args) throws IOException {
            ServerSocket serverSocket = new ServerSocket(8888);
            while(true){
                final Socket socket = serverSocket.accept();
    
                new Thread(new Runnable() {//上传作为一个线程，提高效率。
                    @Override
                    public void run() {
                        try{
                            InputStream is = socket.getInputStream();
                            String fileName =  "img"+System.currentTimeMillis()+new Random().nextInt(999999)+".jpg";
                            FileOutputStream fos = new FileOutputStream(fileName);
                            int len = 0;
                            byte[] bytes =new byte[1024];
                            while((len = is.read(bytes))!=-1){
                                fos.write(bytes,0,len);
                            }
                            socket.getOutputStream().write("上传成功".getBytes());
                            fos.close();
                            socket.close();
                        }catch(IOException e){
                            System.out.println(e);
                        }
                    }
                }).start();
                //服务器就不用关闭了
    //            serverSocket.close();
            }
        }
    }

## 总结 ##

要把握好网络流的使用，同时要熟练使用本地流