2022-12-21 16:12:15 +00:00
|
|
|
package sysmon.plugins.base;
|
2021-05-28 13:52:33 +00:00
|
|
|
|
|
|
|
import org.pf4j.Extension;
|
|
|
|
import org.slf4j.Logger;
|
|
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import oshi.hardware.HWDiskStore;
|
|
|
|
import oshi.hardware.HardwareAbstractionLayer;
|
|
|
|
import sysmon.shared.Measurement;
|
|
|
|
import sysmon.shared.MetricExtension;
|
|
|
|
import sysmon.shared.MetricResult;
|
|
|
|
|
2022-11-07 12:33:35 +00:00
|
|
|
import java.util.*;
|
2021-05-28 13:52:33 +00:00
|
|
|
|
|
|
|
@Extension
|
|
|
|
public class BaseDiskExtension implements MetricExtension {
|
|
|
|
|
|
|
|
private static final Logger log = LoggerFactory.getLogger(BaseDiskExtension.class);
|
|
|
|
|
2021-09-10 10:15:33 +00:00
|
|
|
// Extension details
|
|
|
|
private final String name = "base_disk";
|
|
|
|
private final String description = "Base Disk Metrics";
|
|
|
|
|
|
|
|
// Configuration / Options
|
|
|
|
private boolean enabled = true;
|
2021-11-06 19:32:54 +00:00
|
|
|
private boolean threaded = false;
|
2022-11-07 12:33:35 +00:00
|
|
|
private String interval = "10s";
|
2021-09-10 10:15:33 +00:00
|
|
|
|
2021-05-28 13:52:33 +00:00
|
|
|
private HardwareAbstractionLayer hardwareAbstractionLayer;
|
2022-11-16 11:21:33 +00:00
|
|
|
private List<HWDiskStore> diskStores;
|
|
|
|
private int refreshCounter = 0;
|
2021-05-28 13:52:33 +00:00
|
|
|
|
|
|
|
|
2021-09-10 10:15:33 +00:00
|
|
|
@Override
|
|
|
|
public boolean isEnabled() {
|
|
|
|
return enabled;
|
|
|
|
}
|
|
|
|
|
2021-11-06 19:32:54 +00:00
|
|
|
@Override
|
|
|
|
public boolean isThreaded() {
|
|
|
|
return threaded;
|
|
|
|
}
|
|
|
|
|
2021-05-28 13:52:33 +00:00
|
|
|
@Override
|
|
|
|
public boolean isSupported() {
|
2021-06-11 00:21:32 +00:00
|
|
|
hardwareAbstractionLayer = BasePlugin.getHardwareAbstractionLayer();
|
|
|
|
return hardwareAbstractionLayer != null;
|
2021-05-28 13:52:33 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public String getName() {
|
2021-09-10 10:15:33 +00:00
|
|
|
return name;
|
2021-05-28 13:52:33 +00:00
|
|
|
}
|
|
|
|
|
2022-09-06 13:08:52 +00:00
|
|
|
@Override
|
|
|
|
public String getInterval() {
|
2022-11-07 12:33:35 +00:00
|
|
|
return interval;
|
2022-09-06 13:08:52 +00:00
|
|
|
}
|
|
|
|
|
2021-05-28 13:52:33 +00:00
|
|
|
@Override
|
|
|
|
public String getDescription() {
|
2021-09-10 10:15:33 +00:00
|
|
|
return description;
|
2021-05-28 13:52:33 +00:00
|
|
|
}
|
|
|
|
|
2021-09-10 10:15:33 +00:00
|
|
|
@Override
|
|
|
|
public void setConfiguration(Map<String, Object> map) {
|
|
|
|
if (map.containsKey("enabled")) {
|
|
|
|
enabled = (boolean) map.get("enabled");
|
|
|
|
}
|
2021-11-06 19:32:54 +00:00
|
|
|
if(map.containsKey("threaded")) {
|
|
|
|
threaded = (boolean) map.get("threaded");
|
|
|
|
}
|
2022-11-07 12:33:35 +00:00
|
|
|
if(map.containsKey("interval")) {
|
|
|
|
interval = (String) map.get("interval");
|
|
|
|
}
|
2021-09-10 10:15:33 +00:00
|
|
|
}
|
2021-05-28 13:52:33 +00:00
|
|
|
|
|
|
|
@Override
|
|
|
|
public MetricResult getMetrics() {
|
|
|
|
|
2021-09-11 19:55:48 +00:00
|
|
|
ArrayList<Measurement> measurementList = new ArrayList<>();
|
2022-11-16 11:21:33 +00:00
|
|
|
if(diskStores == null || refreshCounter++ > 360) {
|
|
|
|
log.info("getMetrics() - refreshing list of disk stores");
|
|
|
|
diskStores = hardwareAbstractionLayer.getDiskStores();
|
|
|
|
refreshCounter = 0;
|
|
|
|
}
|
2021-09-11 19:55:48 +00:00
|
|
|
|
2021-05-28 13:52:33 +00:00
|
|
|
for(HWDiskStore store : diskStores) {
|
2021-09-11 19:55:48 +00:00
|
|
|
|
2022-11-16 11:21:33 +00:00
|
|
|
store.updateAttributes();
|
2021-05-28 13:52:33 +00:00
|
|
|
String name = store.getName();
|
2022-11-07 12:33:35 +00:00
|
|
|
if (name.matches("h?disk[0-9]+") ||
|
|
|
|
//name.matches("/dev/dm-[0-9]+") ||
|
|
|
|
name.matches("/dev/x?[sv]d[a-z]") ||
|
|
|
|
name.matches("/dev/nvme[0-9]n[0-9]") ||
|
|
|
|
name.startsWith("\\\\.\\PHYSICALDRIVE")
|
|
|
|
) {
|
|
|
|
|
|
|
|
TreeMap<String, String> tagsMap = new TreeMap<String, String>() {{
|
2021-09-11 19:55:48 +00:00
|
|
|
put("name", name);
|
|
|
|
}};
|
|
|
|
|
2022-11-07 12:33:35 +00:00
|
|
|
TreeMap<String, Object> fieldsMap = new TreeMap<String, Object>() {{
|
2021-09-11 19:55:48 +00:00
|
|
|
put("read", store.getReadBytes());
|
|
|
|
put("write", store.getWriteBytes());
|
|
|
|
put("iotime", store.getTransferTime());
|
|
|
|
put("queue", store.getCurrentQueueLength());
|
|
|
|
}};
|
|
|
|
|
2022-03-01 19:07:43 +00:00
|
|
|
log.debug("getMetrics() - tags: {}, fields: {}", tagsMap, fieldsMap);
|
2021-09-11 19:55:48 +00:00
|
|
|
measurementList.add(new Measurement(tagsMap, fieldsMap));
|
2022-03-30 11:36:54 +00:00
|
|
|
} else {
|
2022-11-07 16:44:14 +00:00
|
|
|
log.debug("getMetrics() - skipping device: {}", name);
|
2021-05-28 13:52:33 +00:00
|
|
|
}
|
2021-09-11 19:55:48 +00:00
|
|
|
|
2021-05-28 13:52:33 +00:00
|
|
|
}
|
|
|
|
|
2021-09-11 19:55:48 +00:00
|
|
|
return new MetricResult(name, measurementList);
|
2021-05-28 13:52:33 +00:00
|
|
|
}
|
|
|
|
|
2022-09-06 13:08:52 +00:00
|
|
|
}
|