From a9afea0603edbd7639ef0eb827046abceacfff66 Mon Sep 17 00:00:00 2001
From: xking <git@x47.cn>
Date: Sun, 1 Dec 2024 16:50:43 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=20=20=E6=9C=8D=E5=8A=A1?=
 =?UTF-8?q?=E7=AB=AF=20=E5=AE=A2=E6=88=B7=E7=AB=AF=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/cn/x47/RIPService.java    | 41 +++++++++++++------------
 src/main/java/cn/x47/config/Config.java |  4 ++-
 2 files changed, 25 insertions(+), 20 deletions(-)

diff --git a/src/main/java/cn/x47/RIPService.java b/src/main/java/cn/x47/RIPService.java
index b638462..0cc64f4 100644
--- a/src/main/java/cn/x47/RIPService.java
+++ b/src/main/java/cn/x47/RIPService.java
@@ -18,26 +18,29 @@ public class RIPService {
 
     public static void main(String[] args) throws Exception {
         // 启动服务器
-//        RIPServer server = new RIPServer();
-//        new Thread(() -> {
-//            try {
-//                server.start();
-//            } catch (InterruptedException e) {
-//                e.printStackTrace();
-//            }
-//        }).start();
+        if (Config.startServer) {
+            RIPServer server = new RIPServer();
+            new Thread(() -> {
+                try {
+                    server.start();
+                } catch (InterruptedException e) {
+                    e.printStackTrace();
+                }
+            }).start();
+            System.out.println("服务端已经启动");
+        }
 
-        // 启动客户端
-        RIPClient client = new RIPClient();
-
-        // 定期发送路由更新
-        ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
-        scheduler.scheduleAtFixedRate(() -> {
-            System.out.println("发送 开始");
-            RIPPacket packet = createRipResponsePacket(Config.RIP_VERSION);
-            client.sendRipPacket(packet);
-            System.out.println("发送 结束");
-        }, 0, 10, TimeUnit.SECONDS);
+        if (Config.startClient) {
+            // 启动客户端
+            RIPClient client = new RIPClient();
+            // 定期发送路由更新
+            ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
+            scheduler.scheduleAtFixedRate(() -> {
+                RIPPacket packet = createRipResponsePacket(Config.RIP_VERSION);
+                client.sendRipPacket(packet);
+            }, 0, 10, TimeUnit.SECONDS);
+            System.out.println("客户端已经启动");
+        }
 
         // 主线程等待
         Thread.currentThread().join();
diff --git a/src/main/java/cn/x47/config/Config.java b/src/main/java/cn/x47/config/Config.java
index bfd55d7..5cc6abb 100644
--- a/src/main/java/cn/x47/config/Config.java
+++ b/src/main/java/cn/x47/config/Config.java
@@ -2,5 +2,7 @@ package cn.x47.config;
 
 public class Config {
     // 设置协议版本，1 表示 RIP v1，2 表示 RIP v2
-    public static final byte RIP_VERSION = 1;
+    public static final byte RIP_VERSION = 2;
+    public static final boolean startServer = true;
+    public static final boolean startClient = true;
 }