🎨 node权重

FishPiOffical · Dec 20, 2024 · 9eebd92 · 9eebd92
1 parent 94a62c1
commit 9eebd92
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/src/main/java/org/b3log/symphony/processor/ChatroomProcessor.java b/src/main/java/org/b3log/symphony/processor/ChatroomProcessor.java
@@ -317,6 +317,7 @@ public void getNode(final RequestContext context) {
             JSONObject node = new JSONObject();
             node.put("node", entry.getKey());
             node.put("name", NodeUtil.nodeNickNames.get(entry.getKey()));
+            node.put("weight", NodeUtil.nodeWeight.get(entry.getKey()));
             node.put("online", entry.getValue());
             data.put(node);
         }

diff --git a/src/main/java/org/b3log/symphony/util/NodeUtil.java b/src/main/java/org/b3log/symphony/util/NodeUtil.java
@@ -64,6 +64,8 @@ public class NodeUtil {
 
     public static HashMap<String, String> nodeNickNames = new HashMap<>();
 
+    public static HashMap<String, String> nodeWeight = new HashMap<>();
+
     public static void init() {
         LOGGER.log(Level.INFO, "Loading nodes");
         for (WebSocket i : wsNodes) {
@@ -73,9 +75,11 @@ public static void init() {
         wsNodes = new ArrayList<>();
         uriNodes = new ArrayList<>();
         nodeNickNames.clear();
+        nodeWeight.clear();
         String[] nodes = Symphonys.get("chatroom.node.url").split(";");
         for (String i : nodes) {
             nodeNickNames.put(i.split(",")[0], i.split(",")[1]);
+            nodeWeight.put(i.split(",")[0], i.split(",")[2]);
             i = i.split(",")[0];
             String serverUri = i + "?apiKey=" + Symphonys.get("chatroom.node.adminKey");
             try {

diff --git a/src/main/resources/symphony.properties b/src/main/resources/symphony.properties
@@ -355,6 +355,6 @@ pay.wechat.mch_id=
 pay.wechat.key=
 
 # Rhyus
-chatroom.node.url=ws://127.0.0.1:10831,\u6cb3\u5317\u4e00\u533a
+chatroom.node.url=ws://127.0.0.1:10831,\u6cb3\u5317\u4e00\u533a,1
 #ws://121.62.31.42:10831
 chatroom.node.adminKey=123456