多线程TCP文件上传

多线程TCP文件上传

学习案例

Client.java

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;

/**
 * 客户端
 * Created by Aongi on 2017/5/4.
 * Version 1.0
 */
public class Client {
    //main方法
    public static void main(String[] args) throws IOException {
        Socket socket = new Socket("127.0.0.1",8888);
        //发送数据给服务器端
        OutputStream os = socket.getOutputStream();
        FileInputStream fis = new FileInputStream("d:\\df.jpg");
        byte[] bytes = new byte[1024];
        int i=0;
        while ((i=fis.read(bytes))!=-1){
            os.write(bytes,0,i);
        }
        //给服务器写终止序列
        socket.shutdownOutput();
        //接受服务器端返回的数据
        InputStream is = socket.getInputStream();
        i =  is.read(bytes);
        System.out.println(new String(bytes,0,i));

        fis.close();
        //关闭socket对象
        socket.close();
    }
}

Server.java

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

/**
 * Created by Aongi on 2017/5/4.
 * Version 1.0
 */
public class Server {
    //main方法
    public static void main(String[] args) throws IOException {
        ServerSocket server = new ServerSocket(8888);
        while (true) {
            // 获取到一个客户端,必须开启新线程,为这个客户端服务
            Socket socket = server.accept();
            new Thread(new TCPRunnable(socket)).start();
        }
    }

}

TCPRunnable.java

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.Socket;
import java.util.Random;

/**
 * Created by Aongi on 2017/5/4.
 * Version 1.0
 */
public class TCPRunnable implements Runnable{
    private Socket socket;
    public TCPRunnable(Socket socket){
        this.socket=socket;
    }
    @Override
    public void run() {
        try {
            //通过客户端连接对象,获取字节输入流,读取客户端图片
            InputStream in = socket.getInputStream();
            //将目的文件夹封装到File对象
            File upload = new File("E:\\tup");
            if (!upload.exists())
                upload.mkdirs();

            //防止文件同名被覆盖,从新定义文件名字
            //规则:  域名+毫秒值+6位随机数
            String filename = "df" + System.currentTimeMillis() + new Random().nextInt(999999) + ".jpg";
            //创建字节输出流,将图片写入到目的文件夹中
            FileOutputStream fos = new FileOutputStream(upload + File.separator + filename);
            //读写字节数组
            byte[] bytes = new byte[1024];
            int len = 0;
            while ((len = in.read(bytes)) != -1) {
                fos.write(bytes, 0, len);
            }
            //通过客户端连接对象获取字节输出流
            //上传成功写回客户端
            socket.getOutputStream().write("上传成功".getBytes());

            fos.close();
            socket.close();
        }catch (IOException e){
            throw new RuntimeException("文件上传失败！");
        }
    }
}