org.mockserver.persistence.FileWatcher Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mockserver-core Show documentation

Functionality used by all MockServer modules for matching and expectations

There is a newer version: 5.15.0

package org.mockserver.persistence;

import org.mockserver.scheduler.Scheduler;

import java.io.File;
import java.nio.file.*;
import java.util.function.Consumer;

import static java.util.concurrent.TimeUnit.MILLISECONDS;

public class FileWatcher {

    private boolean running = true;

    public FileWatcher(String filePath, Runnable updatedHandler, Consumer errorHandler) throws Exception {
        WatchService watchService;
        Path directoryPath = Paths.get(filePath);
        Path fileName = directoryPath.getFileName();
        watchService = FileSystems.getDefault().newWatchService();
        Path parent = directoryPath.getParent() != null ? directoryPath.getParent() : new File(".").toPath();
        parent
            .register(
                watchService,
                StandardWatchEventKinds.OVERFLOW,
                StandardWatchEventKinds.ENTRY_CREATE,
                StandardWatchEventKinds.ENTRY_MODIFY
            );

        new Scheduler.SchedulerThreadFactory(FileWatcher.class.getSimpleName()).newThread(() -> {
            while (isRunning()) {
                try {
                    WatchKey key = watchService.take();

                    if (isRunning()) {
                        for (WatchEvent event : key.pollEvents()) {
                            if (event.context() instanceof Path && ((Path) event.context()).getFileName().equals(fileName)) {
                                // ensure file has been committed to file system
                                MILLISECONDS.sleep(100);
                                updatedHandler.run();
                                break;
                            }
                        }

                        key.reset();
                    }
                } catch (Throwable throwable) {
                    errorHandler.accept(throwable);
                }
            }
        }).start();
    }

    public boolean isRunning() {
        return running;
    }

    public void setRunning(boolean running) {
        this.running = running;
    }
}