|
|
@@ -0,0 +1,235 @@
|
|
|
+package com.mes.util;
|
|
|
+
|
|
|
+import com.alibaba.fastjson2.JSONObject;
|
|
|
+
|
|
|
+import java.io.*;
|
|
|
+import java.net.*;
|
|
|
+import java.nio.charset.StandardCharsets;
|
|
|
+import java.nio.file.*;
|
|
|
+import java.util.Properties;
|
|
|
+import java.util.concurrent.*;
|
|
|
+
|
|
|
+/**
|
|
|
+ * 客户端运行状态代理。
|
|
|
+ * - 类型由 ClientType.CODE 硬编码自报
|
|
|
+ * - clientId 由服务端根据 (类型, IP) 决定
|
|
|
+ * - 心跳上报本地 config.properties 当前内容（currentConfigJson）
|
|
|
+ * - 服务端下发 desiredConfigJson 直接合并写回 config.properties（备份 .bak）→ 重启生效
|
|
|
+ * - 服务端下发 desiredJarVersion 与本地不一致 → 下载对应 jar → 备份 .bak → 替换 → 重启
|
|
|
+ */
|
|
|
+public final class ClientRuntimeAgent {
|
|
|
+ private static final ScheduledExecutorService EXEC = Executors.newSingleThreadScheduledExecutor(r -> {
|
|
|
+ Thread t = new Thread(r, "client-runtime-agent");
|
|
|
+ t.setDaemon(true);
|
|
|
+ return t;
|
|
|
+ });
|
|
|
+ private static volatile boolean updating;
|
|
|
+
|
|
|
+ private ClientRuntimeAgent() {}
|
|
|
+
|
|
|
+ public static void start(String serverIp, String station, String line) {
|
|
|
+ start(serverIp, station, line, "com.mes.ui.MesClient");
|
|
|
+ }
|
|
|
+
|
|
|
+ public static void start(final String serverIp, final String station, final String line, final String mainClass) {
|
|
|
+ if (serverIp == null || serverIp.trim().isEmpty()) return;
|
|
|
+ send(serverIp, station, line, mainClass);
|
|
|
+ EXEC.scheduleAtFixedRate(() -> send(serverIp, station, line, mainClass), 30, 30, TimeUnit.SECONDS);
|
|
|
+ }
|
|
|
+
|
|
|
+ private static void send(String serverIp, String station, String line, String mainClass) {
|
|
|
+ String type = ClientType.CODE;
|
|
|
+ try {
|
|
|
+ String body = "clientType=" + enc(type)
|
|
|
+ + "&stationCode=" + enc(station)
|
|
|
+ + "&lineSn=" + enc(line)
|
|
|
+ + "&jarVersion=" + version()
|
|
|
+ + "&configVersion=" + configVersion()
|
|
|
+ + "&status=RUNNING"
|
|
|
+ + "&statusMessage=" + enc("客户端运行中")
|
|
|
+ + "¤tConfigJson=" + enc(readLocalConfigAsJson());
|
|
|
+ HttpURLConnection c = (HttpURLConnection) new URL("http://" + serverIp + ":8980/js/a/mes/clientRuntime/heartbeat").openConnection();
|
|
|
+ c.setRequestMethod("POST");
|
|
|
+ c.setConnectTimeout(5000);
|
|
|
+ c.setReadTimeout(10000);
|
|
|
+ c.setDoOutput(true);
|
|
|
+ c.setRequestProperty("Content-Type", "application/x-www-form-urlencoded;charset=UTF-8");
|
|
|
+ byte[] bytes = body.getBytes(StandardCharsets.UTF_8);
|
|
|
+ c.setFixedLengthStreamingMode(bytes.length);
|
|
|
+ try (OutputStream out = c.getOutputStream()) { out.write(bytes); }
|
|
|
+ int code = c.getResponseCode();
|
|
|
+ System.out.println("[ClientRuntime] heartbeat type=" + type + " code=" + code);
|
|
|
+ if (code == 200) {
|
|
|
+ String resp = read(c.getInputStream());
|
|
|
+ System.out.println("[ClientRuntime] response=" + resp);
|
|
|
+ JSONObject data = JSONObject.parseObject(resp).getJSONObject("data");
|
|
|
+ if (data != null) {
|
|
|
+ Long cv = data.getLong("desiredConfigVersion");
|
|
|
+ String cfg = data.getString("desiredConfigJson");
|
|
|
+ if (cv != null && cfg != null && !cfg.trim().isEmpty() && cv > configVersion()) {
|
|
|
+ applyConfig(cv, cfg, mainClass);
|
|
|
+ return; // applyConfig 会重启，不用继续处理 jar
|
|
|
+ }
|
|
|
+ Long jv = data.getLong("desiredJarVersion");
|
|
|
+ if (jv != null && jv > version() && !updating) update(serverIp, type, jv, mainClass);
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ System.err.println("[ClientRuntime] heartbeat rejected code=" + code);
|
|
|
+ }
|
|
|
+ c.disconnect();
|
|
|
+ } catch (Exception e) {
|
|
|
+ System.err.println("[ClientRuntime] heartbeat/update failed: " + e.getMessage());
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 读本地 config/config.properties 转为 JSON 字符串
|
|
|
+ private static String readLocalConfigAsJson() {
|
|
|
+ Path p = Paths.get("config", "config.properties");
|
|
|
+ if (!Files.exists(p)) return "";
|
|
|
+ try {
|
|
|
+ Properties props = new Properties();
|
|
|
+ try (InputStream in = Files.newInputStream(p)) { props.load(in); }
|
|
|
+ StringBuilder sb = new StringBuilder("{");
|
|
|
+ boolean first = true;
|
|
|
+ for (String key : props.stringPropertyNames()) {
|
|
|
+ if (!first) sb.append(",");
|
|
|
+ first = false;
|
|
|
+ sb.append('"').append(esc(key)).append("\":\"").append(esc(props.getProperty(key))).append('"');
|
|
|
+ }
|
|
|
+ sb.append('}');
|
|
|
+ return sb.toString();
|
|
|
+ } catch (Exception e) {
|
|
|
+ return "";
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 应用服务端下发的配置：合并到 config.properties（保留未下发的 key），备份 .bak，然后重启生效。
|
|
|
+ */
|
|
|
+ private static void applyConfig(long ver, String cfg, String mainClass) throws Exception {
|
|
|
+ JSONObject json;
|
|
|
+ try { json = JSONObject.parseObject(cfg); } catch (Exception e) { System.err.println("[ClientRuntime] invalid config json"); return; }
|
|
|
+ if (json == null || json.isEmpty()) return;
|
|
|
+
|
|
|
+ Path dir = Paths.get("config");
|
|
|
+ Files.createDirectories(dir);
|
|
|
+ Path pfile = dir.resolve("config.properties");
|
|
|
+ Properties props = new Properties();
|
|
|
+ if (Files.exists(pfile)) {
|
|
|
+ try (InputStream in = Files.newInputStream(pfile)) { props.load(in); }
|
|
|
+ Files.copy(pfile, dir.resolve("config.properties.bak"), StandardCopyOption.REPLACE_EXISTING);
|
|
|
+ }
|
|
|
+ // 合并：JSON 里的 key 覆盖到 properties
|
|
|
+ for (String key : json.keySet()) {
|
|
|
+ String val = json.getString(key);
|
|
|
+ if (val == null) continue;
|
|
|
+ props.setProperty(key, val);
|
|
|
+ }
|
|
|
+ try (OutputStream out = Files.newOutputStream(pfile)) {
|
|
|
+ props.store(out, "Updated by ClientRuntimeAgent " + new java.util.Date());
|
|
|
+ }
|
|
|
+ // 注意：config-version.txt 不在这里写，交给 restartSelf 的 vbs 在启动新 Java 前写
|
|
|
+ // 这样即使 restart 失败，version 也不会被更新，下次心跳仍会触发同一个 applyConfig
|
|
|
+ System.out.println("[ClientRuntime] config applied v=" + ver + ", restarting");
|
|
|
+ restartSelf(mainClass, ver);
|
|
|
+ }
|
|
|
+
|
|
|
+ /** 触发升级：下载新 jar → 备份 → 替换 → 重启 */
|
|
|
+ private static void update(String serverIp, String type, long target, String mainClass) throws Exception {
|
|
|
+ updating = true;
|
|
|
+ JSONObject root = JSONObject.parseObject(read(new URL("http://" + serverIp + ":8980/js/a/mes/clientVersion/ver?clientType=" + enc(type) + "&version=" + target).openStream()));
|
|
|
+ JSONObject data = root.getJSONObject("data");
|
|
|
+ if (data == null) throw new IOException("target version unavailable");
|
|
|
+ File jar = currentJar();
|
|
|
+ File download = new File(jar.getParentFile(), jar.getName() + ".download");
|
|
|
+ download(data.getString("path"), download);
|
|
|
+ if (download.length() == 0) throw new IOException("empty download");
|
|
|
+ Files.write(Paths.get("config", "client-version.txt"), String.valueOf(target).getBytes(StandardCharsets.UTF_8));
|
|
|
+
|
|
|
+ File vbs = new File(jar.getParentFile(), "client-update-" + System.currentTimeMillis() + ".vbs");
|
|
|
+ String jarPath = jar.getAbsolutePath();
|
|
|
+ String bakPath = jarPath + ".bak";
|
|
|
+ String downloadPath = download.getAbsolutePath();
|
|
|
+ String libGlob = new File(jar.getParentFile(), "lib").getAbsolutePath() + File.separator + "*";
|
|
|
+ String workDir = jar.getParentFile().getAbsolutePath();
|
|
|
+ try (Writer w = new OutputStreamWriter(new FileOutputStream(vbs), StandardCharsets.UTF_8)) {
|
|
|
+ w.write(
|
|
|
+ "Set sh = CreateObject(\"WScript.Shell\")\r\n" +
|
|
|
+ "Set fso = CreateObject(\"Scripting.FileSystemObject\")\r\n" +
|
|
|
+ "WScript.Sleep 2000\r\n" +
|
|
|
+ "If fso.FileExists(\"" + jarPath + "\") Then fso.CopyFile \"" + jarPath + "\", \"" + bakPath + "\", True\r\n" +
|
|
|
+ "fso.MoveFile \"" + downloadPath + "\", \"" + jarPath + "\"\r\n" +
|
|
|
+ "sh.CurrentDirectory = \"" + workDir + "\"\r\n" +
|
|
|
+ "sh.Run \"javaw -cp \"\"" + jarPath + ";" + libGlob + "\"\" " + mainClass + "\", 0, False\r\n" +
|
|
|
+ "fso.DeleteFile WScript.ScriptFullName\r\n"
|
|
|
+ );
|
|
|
+ }
|
|
|
+ new ProcessBuilder("wscript.exe", vbs.getAbsolutePath()).start();
|
|
|
+ System.exit(0);
|
|
|
+ }
|
|
|
+
|
|
|
+ /** 仅重启当前 jar，不替换文件（用于配置生效）；vbs 在启动新 Java 前把 verToWrite 写入 config-version.txt */
|
|
|
+ private static void restartSelf(String mainClass, long verToWrite) throws Exception {
|
|
|
+ File jar = currentJar();
|
|
|
+ File vbs = new File(jar.getParentFile(), "client-restart-" + System.currentTimeMillis() + ".vbs");
|
|
|
+ String jarPath = jar.getAbsolutePath();
|
|
|
+ String libGlob = new File(jar.getParentFile(), "lib").getAbsolutePath() + File.separator + "*";
|
|
|
+ String workDir = jar.getParentFile().getAbsolutePath();
|
|
|
+ String verFile = new File(new File(workDir, "config"), "config-version.txt").getAbsolutePath();
|
|
|
+ try (Writer w = new OutputStreamWriter(new FileOutputStream(vbs), StandardCharsets.UTF_8)) {
|
|
|
+ w.write(
|
|
|
+ "Set sh = CreateObject(\"WScript.Shell\")\r\n" +
|
|
|
+ "Set fso = CreateObject(\"Scripting.FileSystemObject\")\r\n" +
|
|
|
+ "WScript.Sleep 2000\r\n" +
|
|
|
+ "Set f = fso.CreateTextFile(\"" + verFile + "\", True)\r\n" +
|
|
|
+ "f.Write \"" + verToWrite + "\"\r\n" +
|
|
|
+ "f.Close\r\n" +
|
|
|
+ "sh.CurrentDirectory = \"" + workDir + "\"\r\n" +
|
|
|
+ "sh.Run \"javaw -cp \"\"" + jarPath + ";" + libGlob + "\"\" " + mainClass + "\", 0, False\r\n" +
|
|
|
+ "fso.DeleteFile WScript.ScriptFullName\r\n"
|
|
|
+ );
|
|
|
+ }
|
|
|
+ new ProcessBuilder("wscript.exe", vbs.getAbsolutePath()).start();
|
|
|
+ System.exit(0);
|
|
|
+ }
|
|
|
+
|
|
|
+ private static void download(String u, File f) throws IOException {
|
|
|
+ HttpURLConnection c = (HttpURLConnection) new URL(u).openConnection();
|
|
|
+ c.setConnectTimeout(10000);
|
|
|
+ c.setReadTimeout(120000);
|
|
|
+ if (c.getResponseCode() != 200) throw new IOException("download failed");
|
|
|
+ try (InputStream in = c.getInputStream(); OutputStream out = new FileOutputStream(f)) {
|
|
|
+ byte[] b = new byte[8192];
|
|
|
+ int n;
|
|
|
+ while ((n = in.read(b)) != -1) out.write(b, 0, n);
|
|
|
+ } finally { c.disconnect(); }
|
|
|
+ }
|
|
|
+
|
|
|
+ private static File currentJar() throws Exception {
|
|
|
+ File f = new File(ClientRuntimeAgent.class.getProtectionDomain().getCodeSource().getLocation().toURI());
|
|
|
+ if (!f.isFile()) throw new IOException("not running from jar");
|
|
|
+ return f;
|
|
|
+ }
|
|
|
+
|
|
|
+ private static long version() { return readLong("client-version.txt", 1); }
|
|
|
+ private static long configVersion() { return readLong("config-version.txt", 0); }
|
|
|
+ private static long readLong(String n, long d) {
|
|
|
+ try { return Long.parseLong(new String(Files.readAllBytes(Paths.get("config", n)), StandardCharsets.UTF_8).trim()); }
|
|
|
+ catch (Exception e) { return d; }
|
|
|
+ }
|
|
|
+ private static String enc(String s) {
|
|
|
+ try { return URLEncoder.encode(s == null ? "" : s, "UTF-8"); } catch (Exception e) { return ""; }
|
|
|
+ }
|
|
|
+ private static String esc(String s) {
|
|
|
+ if (s == null) return "";
|
|
|
+ return s.replace("\\", "\\\\").replace("\"", "\\\"").replace("\n", "\\n").replace("\r", "\\r").replace("\t", "\\t");
|
|
|
+ }
|
|
|
+ private static String read(InputStream in) throws IOException {
|
|
|
+ try (BufferedReader r = new BufferedReader(new InputStreamReader(in, StandardCharsets.UTF_8))) {
|
|
|
+ StringBuilder b = new StringBuilder();
|
|
|
+ String l;
|
|
|
+ while ((l = r.readLine()) != null) b.append(l);
|
|
|
+ return b.toString();
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
wangxichen