背景:本地某路徑下中有數據生成,需要對其進行監控,並進行相應處理
CODE的分析:
啓動文件服務,數據生成處理
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.io.File;
import java.io.FileNotFoundException;
import java.nio.file.Path;
@Service
public class HandleService implements InitializingBean {
private static final Logger LOGGER = LoggerFactory.getLogger(HandleService.class);
private UserConfiguration userConfiguration;
@Override
public void afterPropertiesSet() throws Exception {
String path = argumentNotNullAndNotEmpty(userConfiguration.getMonitorPath(),
"test.user.monitor-path");
File file = new File(path);
if (!file.exists() || !file.isDirectory()) {
throw new FileNotFoundException("沒有找到文件夾" + file.getAbsolutePath());
}
String extension = userConfiguration.getExtension();
if (extension == null) {
extension = "";
} else {
extension = extension.trim();
}
Path monitorPath = file.toPath();
FileMonitor monitor = new FileMonitor(monitorPath, LOGGER, extension);
monitor.start(this::handleEvent, true); //調用文件監控服務,並啓動
}
private void handleEvent(String params) {
//對新數據的生成進行相應的處理
//TO DO Something
}
private String argumentNotNullAndNotEmpty(String value, String name) {
if (value == null) {
throw new NullPointerException(name);
}
value = value.trim();
if (value.isEmpty()) {
throw new IllegalArgumentException(name);
}
return value;
}
@Autowired
public void setUserConfiguration(UserConfiguration userConfiguration) {
this.userConfiguration = userConfiguration;
}
}
啓動文件服務,數據生成處理類
import org.slf4j.Logger;
import java.nio.file.*;
import java.util.function.Consumer;
public class FileMonitor {
private final Path monitorPath;
private final Logger logger;
private final String extension;
public FileMonitor(Path monitorPath, Logger logger, String extension) {
this.monitorPath = monitorPath;
this.logger = logger;
this.extension = extension;
}
public void start(Consumer<String> consumer, boolean autoRestart) {
new Thread(() -> {
try {
start(consumer);
} catch (Exception e) {
if (autoRestart) {
logger.info("嘗試重啓文件監視服務");
start(consumer, true);
}
}
}, "FileMonitor").start();
}
private void start(Consumer<String> consumer) throws Exception {
try (WatchService watchService = FileSystems.getDefault().newWatchService()) {
monitorPath.register(watchService, StandardWatchEventKinds.ENTRY_CREATE);
logger.info("文件監視服務開始服務:監視文件夾{}", monitorPath.toString());
WatchKey key;
while ((key = watchService.take()) != null) {
for (WatchEvent<?> event : key.pollEvents()) {
Path path = monitorPath.resolve((Path) event.context());
String filename = path.toString();
if (!Files.isDirectory(path) && filename.endsWith(extension) && !filename.endsWith(".temp")) {
logger.info("{}文件發生了{}事件!", getFileName(filename), event.kind());
consumer.accept(filename);
}
}
boolean valid = key.reset();
if (!valid) {
throw new Exception("watchKey重置失敗");
}
}
} finally {
logger.error("文件監視服務停止服務");
}
}
private String getFileName(String fullPath) {
int index = fullPath.lastIndexOf('\\');
if (index > 0 && index < fullPath.length()) {
return fullPath.substring(index + 1);
}
return fullPath;
}
}
配置類
appilication.properties文件
#文件路徑
test.user.monitor-path=D://data
test.user.extension=.zip
配置類
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;
import java.util.List;
import java.util.Map;
@Component
@Data
@ConfigurationProperties(prefix = "test.user")
public class UserConfiguration {
private String monitorPath;
private String extension;
}