diff --git a/.gitignore b/.gitignore
index 81a1846..5551e48 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
.idea
log
.DS_Store
+config
diff --git a/itbaima-monitor-client/pom.xml b/itbaima-monitor-client/pom.xml
index addf445..41aadc0 100644
--- a/itbaima-monitor-client/pom.xml
+++ b/itbaima-monitor-client/pom.xml
@@ -32,6 +32,11 @@
spring-boot-starter-test
test
+
+ com.alibaba.fastjson2
+ fastjson2
+ 2.0.37
+
diff --git a/itbaima-monitor-client/src/main/java/com/example/config/ServerConfiguration.java b/itbaima-monitor-client/src/main/java/com/example/config/ServerConfiguration.java
new file mode 100644
index 0000000..9a03d34
--- /dev/null
+++ b/itbaima-monitor-client/src/main/java/com/example/config/ServerConfiguration.java
@@ -0,0 +1,73 @@
+package com.example.config;
+
+import com.alibaba.fastjson2.JSONObject;
+import com.example.entity.ConnectionConfig;
+import com.example.utils.NetUtils;
+import jakarta.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.util.Scanner;
+
+@Slf4j
+@Configuration
+public class ServerConfiguration {
+
+ @Resource
+ NetUtils net;
+
+ @Bean
+ ConnectionConfig connectionConfig() {
+ log.info("正在加载服务端连接配置...");
+ ConnectionConfig config = this.readConfigurationFromFile();
+ if(config == null)
+ config = this.registerToServer();
+ return config;
+ }
+
+ private ConnectionConfig registerToServer() {
+ Scanner scanner = new Scanner(System.in);
+ String token, address;
+ do {
+ log.info("请输入需要注册的服务端访问地址，地址类似于 'http://192.168.0.22:8080' 这种写法:");
+ address = scanner.nextLine();
+ log.info("请输入服务端生成的用于注册客户端的Token秘钥:");
+ token = scanner.nextLine();
+ } while (!net.registerToServer(address, token));
+ ConnectionConfig config = new ConnectionConfig(address, token);
+ this.saveConfigurationToFile(config);
+ return config;
+ }
+
+ private void saveConfigurationToFile(ConnectionConfig config) {
+ File dir = new File("config");
+ if(!dir.exists() && dir.mkdir())
+ log.info("创建用于保存服务端连接信息的目录已完成");
+ File file = new File("config/server.json");
+ try(FileWriter writer = new FileWriter(file)) {
+ writer.write(JSONObject.from(config).toJSONString());
+ } catch (IOException e) {
+ log.error("保存配置文件时出现问题", e);
+ }
+ log.info("服务端连接信息已保存成功！");
+ }
+
+ private ConnectionConfig readConfigurationFromFile() {
+ File configurationFile = new File("config/server.json");
+ if(configurationFile.exists()) {
+ try (FileInputStream stream = new FileInputStream(configurationFile)){
+ String raw = new String(stream.readAllBytes(), StandardCharsets.UTF_8);
+ return JSONObject.parseObject(raw).to(ConnectionConfig.class);
+ } catch (IOException e) {
+ log.error("读取配置文件时出错", e);
+ }
+ }
+ return null;
+ }
+}
diff --git a/itbaima-monitor-client/src/main/java/com/example/entity/ConnectionConfig.java b/itbaima-monitor-client/src/main/java/com/example/entity/ConnectionConfig.java
new file mode 100644
index 0000000..76eda46
--- /dev/null
+++ b/itbaima-monitor-client/src/main/java/com/example/entity/ConnectionConfig.java
@@ -0,0 +1,11 @@
+package com.example.entity;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+
+@Data
+@AllArgsConstructor
+public class ConnectionConfig {
+ String address;
+ String token;
+}
diff --git a/itbaima-monitor-client/src/main/java/com/example/entity/Response.java b/itbaima-monitor-client/src/main/java/com/example/entity/Response.java
new file mode 100644
index 0000000..f938fd8
--- /dev/null
+++ b/itbaima-monitor-client/src/main/java/com/example/entity/Response.java
@@ -0,0 +1,22 @@
+package com.example.entity;
+
+import com.alibaba.fastjson2.JSONObject;
+
+public record Response(int id, int code, Object data, String message) {
+
+ public boolean success() {
+ return code == 200;
+ }
+
+ public JSONObject asJson() {
+ return JSONObject.from(data);
+ }
+
+ public String asString() {
+ return data.toString();
+ }
+
+ public static Response errorResponse(Exception e) {
+ return new Response(0, 500, null, e.getMessage());
+ }
+}
diff --git a/itbaima-monitor-client/src/main/java/com/example/utils/NetUtils.java b/itbaima-monitor-client/src/main/java/com/example/utils/NetUtils.java
new file mode 100644
index 0000000..cc539bc
--- /dev/null
+++ b/itbaima-monitor-client/src/main/java/com/example/utils/NetUtils.java
@@ -0,0 +1,53 @@
+package com.example.utils;
+
+import com.alibaba.fastjson2.JSONObject;
+import com.example.entity.ConnectionConfig;
+import com.example.entity.Response;
+import jakarta.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.context.annotation.Lazy;
+import org.springframework.stereotype.Component;
+
+import java.net.URI;
+import java.net.http.HttpClient;
+import java.net.http.HttpRequest;
+import java.net.http.HttpResponse;
+
+@Slf4j
+@Component
+public class NetUtils {
+ private final HttpClient client = HttpClient.newHttpClient();
+
+ @Lazy
+ @Resource
+ ConnectionConfig config;
+
+ public boolean registerToServer(String address, String token) {
+ log.info("正在像服务端注册，请稍后...");
+ Response response = this.doGet("/register", address, token);
+ if(response.success()) {
+ log.info("客户端注册已完成！");
+ } else {
+ log.error("客户端注册失败: {}", response.message());
+ }
+ return response.success();
+ }
+
+ private Response doGet(String url) {
+ return this.doGet(url, config.getAddress(), config.getToken());
+ }
+
+ private Response doGet(String url, String address, String token) {
+ try {
+ HttpRequest request = HttpRequest.newBuilder().GET()
+ .uri(new URI(address + "/monitor" + url))
+ .header("Authorization", token)
+ .build();
+ HttpResponse response = client.send(request, HttpResponse.BodyHandlers.ofString());
+ return JSONObject.parseObject(response.body()).to(Response.class);
+ } catch (Exception e) {
+ log.error("在发起服务端请求时出现问题", e);
+ return Response.errorResponse(e);
+ }
+ }
+}
diff --git a/itbaima-monitor-client/src/main/resources/application.properties b/itbaima-monitor-client/src/main/resources/application.yml
similarity index 100%
rename from itbaima-monitor-client/src/main/resources/application.properties
rename to itbaima-monitor-client/src/main/resources/application.yml
diff --git a/itbaima-monitor-server/src/main/java/com/example/service/impl/ClientServiceImpl.java b/itbaima-monitor-server/src/main/java/com/example/service/impl/ClientServiceImpl.java
index 023e0ae..9fc12ae 100644
--- a/itbaima-monitor-server/src/main/java/com/example/service/impl/ClientServiceImpl.java
+++ b/itbaima-monitor-server/src/main/java/com/example/service/impl/ClientServiceImpl.java
@@ -70,6 +70,7 @@ public class ClientServiceImpl extends ServiceImpl impleme
StringBuilder sb = new StringBuilder(24);
for (int i = 0; i < 24; i++)
sb.append(CHARACTERS.charAt(random.nextInt(CHARACTERS.length())));
+ System.out.println(sb);
return sb.toString();
}
}
diff --git a/itbaima-monitor-server/src/main/resources/application-dev.yml b/itbaima-monitor-server/src/main/resources/application-dev.yml
index 134c281..81fac07 100644
--- a/itbaima-monitor-server/src/main/resources/application-dev.yml
+++ b/itbaima-monitor-server/src/main/resources/application-dev.yml
@@ -14,7 +14,7 @@ spring:
password: admin
virtual-host: /
datasource:
- url: jdbc:mysql://localhost:3306/test
+ url: jdbc:mysql://localhost:3306/monitor
username: root
password: 123456
driver-class-name: com.mysql.cj.jdbc.Driver