1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
package com.main;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.math.RoundingMode;
import java.net.ServerSocket;
import java.net.Socket;
import java.text.DecimalFormat;
 
/**
 * Îļþ´«ÊäServer¶Ë<br>
 * ¹¦ÄÜ˵Ã÷£º
 *
 * @author Â³ÐÇΰ
 * @Date 2019Äê02ÔÂ14ÈÕ
 * @version 1.0
 */
public class FileTransferServer extends ServerSocket {
 
    private static final int SERVER_PORT = 8899; // ·þÎñ¶Ë¶Ë¿Ú
 
    private static DecimalFormat df = null;
 
    static {
        // ÉèÖÃÊý×Ö¸ñʽ£¬±£ÁôһλÓÐЧСÊý
        df = new DecimalFormat("#0.0");
        df.setRoundingMode(RoundingMode.HALF_UP);
        df.setMinimumFractionDigits(1);
        df.setMaximumFractionDigits(1);
    }
 
    public FileTransferServer() throws Exception {
        super(SERVER_PORT);
    }
 
    /**
     * Ê¹ÓÃÏ̴߳¦Àíÿ¸ö¿Í»§¶Ë´«ÊäµÄÎļþ
     * @throws Exception
     */
    public void load() throws Exception {
        while (true) {
            // server³¢ÊÔ½ÓÊÕÆäËûSocketµÄÁ¬½ÓÇëÇó£¬serverµÄaccept·½·¨ÊÇ×èÈûʽµÄ
            Socket socket = this.accept();
            /**
             * ÎÒÃǵķþÎñ¶Ë´¦Àí¿Í»§¶ËµÄÁ¬½ÓÇëÇóÊÇͬ²½½øÐеģ¬ Ã¿´Î½ÓÊÕµ½À´×Ô¿Í»§¶ËµÄÁ¬½ÓÇëÇóºó£¬
             * ¶¼ÒªÏȸúµ±Ç°µÄ¿Í»§¶ËͨÐÅÍêÖ®ºó²ÅÄÜÔÙ´¦ÀíÏÂÒ»¸öÁ¬½ÓÇëÇó¡£ ÕâÔÚ²¢·¢±È½Ï¶àµÄÇé¿öÏ»áÑÏÖØÓ°Ïì³ÌÐòµÄÐÔÄÜ£¬
             * Îª´Ë£¬ÎÒÃÇ¿ÉÒÔ°ÑËü¸ÄΪÈçÏÂÕâÖÖÒì²½´¦ÀíÓë¿Í»§¶ËͨÐŵķ½Ê½
             */
            // Ã¿½ÓÊÕµ½Ò»¸öSocket¾Í½¨Á¢Ò»¸öеÄÏß³ÌÀ´´¦ÀíËü
            new Thread(new Task(socket)).start();
        }
    }
 
    /**
     * ´¦Àí¿Í»§¶Ë´«Êä¹ýÀ´µÄÎļþÏß³ÌÀà
     */
    class Task implements Runnable {
 
        private Socket socket;
 
        private DataInputStream dis;
 
        private FileOutputStream fos;
 
        public Task(Socket socket) {
            this.socket = socket;
        }
 
        @Override
        public void run() {
            try {
                dis = new DataInputStream(socket.getInputStream());
 
                // ÎļþÃûºÍ³¤¶È
                String fileName = dis.readUTF();
                long fileLength = dis.readLong();
                File directory = new File("D:\\FTCache");
                if(!directory.exists()) {
                    directory.mkdir();
                }
                File file = new File(directory.getAbsolutePath() + File.separatorChar + fileName);
                fos = new FileOutputStream(file);
 
                // ¿ªÊ¼½ÓÊÕÎļþ
                byte[] bytes = new byte[1024];
                int length = 0;
                while((length = dis.read(bytes, 0, bytes.length)) != -1) {
                    fos.write(bytes, 0, length);
                    fos.flush();
                }
                System.out.println("======== Îļþ½ÓÊճɹ¦ [File Name£º" + fileName + "] [Size£º" + getFormatFileSize(fileLength) + "] ========");
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                try {
                    if(fos != null)
                        fos.close();
                    if(dis != null)
                        dis.close();
                    socket.close();
                } catch (Exception e) {}
            }
        }
    }
 
    /**
     * ¸ñʽ»¯Îļþ´óС
     * @param length
     * @return
     */
    private String getFormatFileSize(long length) {
        double size = ((double) length) / (1 << 30);
        if(size >= 1) {
            return df.format(size) + "GB";
        }
        size = ((double) length) / (1 << 20);
        if(size >= 1) {
            return df.format(size) + "MB";
        }
        size = ((double) length) / (1 << 10);
        if(size >= 1) {
            return df.format(size) + "KB";
        }
        return length + "B";
    }
 
    /**
     * Èë¿Ú
     * @param args
     */
    public static void main(String[] args) {
        try {
            FileTransferServer server = new FileTransferServer(); // Æô¶¯·þÎñ¶Ë
            server.load();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}