Java 監控本地某路徑下是否有數據生成並進行相應處理

背景:本地某路徑下中有數據生成,需要對其進行監控,並進行相應處理
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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章