Skip to content

Commit 503b499

Browse files
committed
refactor property action
1 parent 2840fb2 commit 503b499

File tree

2 files changed

+28
-14
lines changed

2 files changed

+28
-14
lines changed

platform-agent/src/main/java/com/flow/platform/agent/CmdManager.java

Lines changed: 1 addition & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -26,19 +26,14 @@
2626
import com.flow.platform.domain.CmdStatus;
2727
import com.flow.platform.domain.CmdType;
2828
import com.flow.platform.domain.Jsonable;
29-
import com.flow.platform.util.ExceptionUtil;
3029
import com.flow.platform.util.Logger;
3130
import com.google.common.collect.Lists;
3231
import com.google.common.collect.Maps;
33-
import java.io.FileInputStream;
34-
import java.io.InputStream;
35-
import java.net.URL;
3632
import java.time.ZonedDateTime;
3733
import java.util.ArrayList;
3834
import java.util.HashMap;
3935
import java.util.List;
4036
import java.util.Map;
41-
import java.util.Properties;
4237
import java.util.concurrent.ExecutorService;
4338
import java.util.concurrent.Executors;
4439
import java.util.concurrent.LinkedBlockingQueue;
@@ -254,15 +249,7 @@ private String collectionAgentInfo() {
254249
dic.put("useMemory", use / kb + "MB");
255250
dic.put("zone", Config.zone());
256251
dic.put("name", Config.name());
257-
URL resource = this.getClass().getClassLoader().getResource("application.properties");
258-
try {
259-
InputStream fileInputStream = new FileInputStream(resource.getFile());
260-
Properties properties = new Properties();
261-
properties.load(fileInputStream);
262-
dic.put("agentVersion", (String) properties.get("version"));
263-
} catch (Throwable e) {
264-
LOGGER.warn("read agent version from application.properties error %s", ExceptionUtil.findRootCause(e));
265-
}
252+
dic.put("agentVersion", Config.getProperty("version"));
266253
return Jsonable.GSON_CONFIG.toJson(dic);
267254
}
268255

platform-agent/src/main/java/com/flow/platform/agent/Config.java

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,12 @@
2121
import com.flow.platform.util.http.HttpClient;
2222
import com.flow.platform.util.http.HttpResponse;
2323
import com.flow.platform.util.zk.ZKClient;
24+
import java.io.FileInputStream;
25+
import java.io.InputStream;
26+
import java.net.URL;
2427
import java.nio.file.Path;
2528
import java.nio.file.Paths;
29+
import java.util.Properties;
2630
import org.apache.curator.utils.ZKPaths;
2731

2832
/**
@@ -50,6 +54,8 @@ public class Config {
5054
public static String ZONE;
5155
public static String NAME;
5256

57+
private static Properties properties;
58+
5359
public static boolean isDebug() {
5460
String boolStr = System.getProperty(PROP_IS_DEBUG, "false");
5561
return Boolean.parseBoolean(boolStr);
@@ -60,6 +66,27 @@ public static int zkTimeout() {
6066
return Integer.parseInt(intStr);
6167
}
6268

69+
/**
70+
* get property from application.properties
71+
* @param name
72+
* @return
73+
*/
74+
public static String getProperty(String name) {
75+
String value;
76+
URL resource = Config.class.getClassLoader().getResource("application.properties");
77+
if (properties == null) {
78+
try (InputStream fileInputStream = new FileInputStream(resource.getFile())) {
79+
properties = new Properties();
80+
properties.load(fileInputStream);
81+
} catch (Throwable e) {
82+
}
83+
}
84+
85+
value = properties.getProperty(name);
86+
87+
return value;
88+
}
89+
6390
/**
6491
* Is delete cmd log after uploaded
6592
*/

0 commit comments

Comments
 (0)