V1.0

Author: Dantence

.gitignore

@@ -1,4 +1,2 @@
 
-settings
-
 target

README.md

@@ -0,0 +1,9 @@
+## 局域网文件传输系统
+
+### 简介
+
+基于Java socket api的P2P文件传输系统，支持文件一键共享、P2P文件传输、文件共享管理、文件下载管理以及用户管理等功能。
+
+### 使用方法
+
+配置好局域网环境后，使用命令java -jar 运行jar包(或者启动Main类中的main方法)

settings/setting.properties

@@ -0,0 +1,8 @@
+# application port
+port=11223
+
+# share file root directory
+share_root=D:/test
+
+# download file root directory
+download_root=D:/test/download

src/main/java/cache/Cache.java

@@ -4,17 +4,12 @@
  * @date 2023/3/23
  */
 
-import common.util.FileScanner;
-import common.util.PropertyParser;
 import thread.BroadcastReceiverThread;
 import thread.Server;
-import thread.ServerThread;
 
-import java.io.IOException;
 import java.net.InetAddress;
 import java.net.NetworkInterface;
 import java.net.Socket;
-import java.net.UnknownHostException;
 import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;
 
@@ -30,9 +25,6 @@
 public class Cache {
     public static ConcurrentHashMap<String, List<String>> ipRecourceMap = new ConcurrentHashMap<>();
     public static HashSet<String> ipTable = new HashSet<>();
-
-    public static HashMap<String, Socket> socketPool = new HashMap<>();
-
     public static String localHost;
 
     static {
@@ -54,7 +46,6 @@ public static String getLocalHostExactAddress() {
                         continue;
                     }
                     if (currentAddr.getHostAddress().contains(":")) {
-                        // IPv6 address
                         continue;
                     }
                     if (currentAddr.isSiteLocalAddress()) {
@@ -69,16 +60,6 @@ public static String getLocalHostExactAddress() {
         return null;
     }
 
-    public static Socket getSocketByIp(String ip) throws IOException {
-        if (socketPool.containsKey(ip)) {
-            return socketPool.get(ip);
-        } else {
-            Socket socket = new Socket(ip, 11223);
-            socketPool.put(ip, socket);
-            return socket;
-        }
-    }
-
     public static BroadcastReceiverThread broadcastReceiverThread;
 
     public static Server serverThread;
@@ -89,23 +70,6 @@ public static boolean checkThread(String ip) {
         return serverThreadMap.containsKey(ip);
     }
 
-    public static void stopAllThread() {
-        broadcastReceiverThread.stop();
-        for(String ip : serverThreadMap.keySet()) {
-            try {
-                serverThreadMap.get(ip).close();
-            } catch (Exception e) {
-                e.printStackTrace();
-            }
-
-        }
-
-        try {
-            serverThread.stop();
-        } catch (IOException e) {
-            e.printStackTrace();
-        }
-    }
 
     public static void add(String ip, List<String> resources) {
         ipTable.add(ip);
@@ -117,22 +81,4 @@ public static void clear(String ip) {
         ipRecourceMap.remove(ip);
     }
 
-    public static void printResources() {
-        HashSet<String> fileSet = new HashSet<>();
-        for (String ip : ipRecourceMap.keySet()) {
-            fileSet.addAll(ipRecourceMap.get(ip));
-        }
-        for (String file : fileSet) {
-            System.out.println(file);
-        }
-    }
-
-    public static String getAvailableIp(String file) {
-        for (String ip : ipRecourceMap.keySet()) {
-            if (ipRecourceMap.get(ip).contains(file)) {
-                return ip;
-            }
-        }
-        return null;
-    }
 }

src/main/java/cache/CacheUpdate.java

@@ -0,0 +1,41 @@
+package cache;
+
+import common.message.MessageType;
+import common.util.FileScanner;
+import common.util.PropertyParser;
+import service.BroadcastSenderService;
+
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+
+public class CacheUpdate implements Runnable {
+    private final ScheduledExecutorService executor;
+
+    public CacheUpdate() {
+        this.executor = Executors.newSingleThreadScheduledExecutor(runnable -> {
+            Thread thread = new Thread(runnable);
+            thread.setDaemon(true);
+            return thread;
+        });
+    }
+
+    public void start() {
+        executor.scheduleAtFixedRate(this, 0, 5, TimeUnit.SECONDS);
+    }
+
+    @Override
+    public void run() {
+        Cache.add(Cache.localHost, FileScanner.getAllFiles(PropertyParser.getShareRoot()));
+        try {
+            BroadcastSenderService.sendMessage(MessageType.HEART_BEAT_REQ);
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+
+    public void stop() {
+        executor.shutdown();
+    }
+
+}

src/main/java/common/message/MessageType.java

@@ -18,10 +18,15 @@ public interface MessageType {
     String FILE_DOWNLOAD_REQ = "5";
     // 文件下载响应
     String FILE_DOWNLOAD_RESP = "6";
+    // 退出登录响应
     String BROADCAST_LOGOUT_RESP = "7";
-
+    // 关闭程序请求
     String CLOSE_REQ = "8";
-
+    // 关闭程序响应
     String CLOSE_RESP = "9";
+    // 心跳检测请求
+    String HEART_BEAT_REQ = "10";
+    // 心跳检测响应
+    String HEART_BEAT_RESP = "11";
 
 }

src/main/java/common/util/SocketPool.java

@@ -21,8 +21,6 @@ public static synchronized Socket getSocket(String host, int port) throws Except
         for (int i = 0; i < pool.size(); i++) {
 
             Socket socket = pool.get(i);
-            System.out.println("host: " + socket.getInetAddress().getHostAddress());
-            System.out.println("port: " + socket.getPort());
 
             if (socket.getInetAddress().getHostAddress().equals(host) && socket.getPort() == port && !socket.isClosed()) {
                 pool.remove(i);

src/main/java/main/Main.java

@@ -1,33 +1,37 @@
 package main;
 
 import cache.Cache;
+import cache.CacheUpdate;
 import common.message.MessageType;
-import thread.BroadcastSenderThread;
-import thread.BroadcastReceiverThread;
-import thread.Server;
-import thread.ServerThread;
-import window.FileUploader;
-
-import java.io.File;
+import service.BroadcastSenderService;
+import thread.*;
+import window.ClientWindow;
 
 public class Main {
     public static void main(String[] args) throws Exception {
         // 启动广播包接收线程
         Cache.broadcastReceiverThread = new BroadcastReceiverThread();
-        Thread thread1 = new Thread(Cache.broadcastReceiverThread);
-        thread1.setDaemon(true);
-        thread1.start();
+        Thread broadcastReceiverThread = new Thread(Cache.broadcastReceiverThread);
+        broadcastReceiverThread.setDaemon(true);
+        broadcastReceiverThread.start();
 
-        // 启动服务线程
+        // 启动服务端线程
         Cache.serverThread = new Server();
-        Thread thread = new Thread(Cache.serverThread);
-        thread.setDaemon(true);
-        thread.start();
+        Thread serverThread = new Thread(Cache.serverThread);
+        serverThread.setDaemon(true);
+        serverThread.start();
+
+        // 启动客户端线程
+        // new Thread(new ClientThread()).start();
 
         // 节点上线发送广播包
-        new Thread(new BroadcastSenderThread(MessageType.BROADCAST_LOGIN_REQ)).start();
+        BroadcastSenderService.sendMessage(MessageType.BROADCAST_LOGIN_REQ);
 
         // 启动图形界面
-        new FileUploader();
+        new ClientWindow();
+
+        // 异步缓存更新及心跳监测线程
+        CacheUpdate cacheUpdate = new CacheUpdate();
+        cacheUpdate.start();
     }
 }

src/main/java/service/BroadcastSenderService.java

@@ -0,0 +1,47 @@
+package service;/**
+ * @author Dantence
+ * @description:
+ * @date 2023/3/29
+ */
+
+import cache.Cache;
+import common.dto.BroadcastLoginDTO;
+import common.message.Message;
+import common.util.FileScanner;
+import common.util.MessageUtil;
+import common.util.PropertyParser;
+
+import java.net.DatagramSocket;
+import java.net.InetAddress;
+
+/**
+ * @projectName: p2pFileSystem
+ * @package: service
+ * @className: BroadcastSenderService
+ * @author: Dantence
+ * @description: 广播服务
+ * @date: 2023/3/29 15:15
+ * @version: 1.0
+ */
+public class BroadcastSenderService {
+    public static void sendMessage(String messageType) {
+        DatagramSocket socket = null;
+        try {
+            socket = new DatagramSocket();
+            socket.setBroadcast(true);
+            Message message = new Message();
+            BroadcastLoginDTO broadcastLoginDTO = new BroadcastLoginDTO();
+            broadcastLoginDTO.setIp(Cache.localHost);
+            broadcastLoginDTO.setResources(FileScanner.getAllFiles(PropertyParser.getShareRoot()));
+            message.setBroadcastLoginDTO(broadcastLoginDTO);
+            message.setMessageType(messageType);
+            MessageUtil.sendBroadcastMessage(socket, message, InetAddress.getByName("255.255.255.255"), PropertyParser.getPort());
+        } catch (Exception e) {
+            e.printStackTrace();
+        } finally {
+            if(socket != null) {
+                socket.close();
+            }
+        }
+    }
+}

src/main/java/service/DownloadFileService.java

@@ -17,7 +17,7 @@
  * @package: service
  * @className: DownloadFileService
  * @author: Dantence
- * @description: TODO
+ * @description: 文件下载服务
  * @date: 2023/3/26 1:01
  * @version: 1.0
  */