模擬C/S服務器上傳文件

使用多線程技術實現只要有一個客戶上傳文件，就開啓一個線程，完成文件的上傳。

TCP通信的客戶端代碼：

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

//模擬C/S服務器上傳文件
//文件上傳到客戶端
public class TCPClient {
    public static void main(String[] args) throws IOException {
        FileInputStream fis = new FileInputStream("d:\\1.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);
        }
        socket.shutdownOutput();   //上傳完文件，給服務器寫一個結束標記,不然會發生阻塞狀態
        InputStream is = socket.getInputStream();
        while ((len = is.read(bytes)) != -1){
            System.out.println(new String(bytes, 0, len));
        }
        fis.close();
        os.close();
    }
}

TCP通信的服務器端代碼：

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Random;

//文件上傳到服務器端
public class TCPServer {
    public static void main(String[] args) throws IOException {
        ServerSocket server = new ServerSocket(8888);

        //讓服務器一直處於監聽狀態(死循環)
        while (true) {
            Socket socket = server.accept();

            //使用多線程技術，有一個客戶上傳文件，就開啓一個線程，完成文件上傳
            new Thread(new Runnable() {

                //完成文件上傳
                @Override
                public void run() {
                    try {
                        InputStream is = socket.getInputStream();
                        File file = new File("d:\\upload");
                        if (!file.exists()){
                            file.mkdirs();
                        }

                        //自定義文件命名規則：域名 + 毫秒值 + 隨機數
                        String filename = "hucm" + System.currentTimeMillis() + new Random().nextInt(999999) + ".jpg";
                        FileOutputStream fos = new FileOutputStream(file + "\\" + 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();
        }
        //server.close();服務器就沒必要關閉
    }
}