diff --git a/src/main/java/com/profiler/SystemMonitor.java b/src/main/java/com/profiler/SystemMonitor.java index dcb356d77..973a74212 100644 --- a/src/main/java/com/profiler/SystemMonitor.java +++ b/src/main/java/com/profiler/SystemMonitor.java @@ -45,8 +45,8 @@ public class SystemMonitor { public SystemMonitor() { } - public void setDataSender(DataSender udpDataSender) { - this.dataSender = udpDataSender; + public void setDataSender(DataSender dataSender) { + this.dataSender = dataSender; } public void start() { @@ -59,7 +59,7 @@ public class SystemMonitor { executor.shutdown(); } - static class Worker implements Runnable { + private static class Worker implements Runnable { private DataSender dataSender; diff --git a/src/main/java/com/profiler/sender/LoggingDataSender.java b/src/main/java/com/profiler/sender/LoggingDataSender.java new file mode 100644 index 000000000..2035a5120 --- /dev/null +++ b/src/main/java/com/profiler/sender/LoggingDataSender.java @@ -0,0 +1,23 @@ +package com.profiler.sender; + +import org.apache.thrift.TBase; + +import java.util.logging.Logger; + +/** + * + */ +public class LoggingDataSender implements DataSender { + private Logger logger = Logger.getLogger(this.getClass().getName()); + + @Override + public boolean send(TBase data) { + logger.info("send tBase:" + data); + return true; + } + + @Override + public void stop() { + + } +} diff --git a/src/test/java/com/profiler/SystemMonitorTest.java b/src/test/java/com/profiler/SystemMonitorTest.java index ec9b94e23..3efd7f4c1 100644 --- a/src/test/java/com/profiler/SystemMonitorTest.java +++ b/src/test/java/com/profiler/SystemMonitorTest.java @@ -1,5 +1,7 @@ package com.profiler; +import com.profiler.sender.DataSender; +import com.profiler.sender.LoggingDataSender; import org.junit.Test; /** @@ -12,8 +14,15 @@ import org.junit.Test; public class SystemMonitorTest { @Test public void testStart() throws Exception { - SystemMonitor.Worker systemMonitor = new SystemMonitor.Worker(); - systemMonitor.run(); + DataSender loggingDataSender = new LoggingDataSender(); + + SystemMonitor systemMonitor = new SystemMonitor(); + systemMonitor.setDataSender(loggingDataSender); + systemMonitor.start(); + + Thread.sleep(10000); + + systemMonitor.stop(); } @Test