package com.yx.demo.filemonitor;
import java.io.File;
import java.lang.ref.WeakReference;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Timer;
import java.util.TimerTask;
/**
* FileMonitor
* 文件监控器
*
* @author yx
* @date 2019/12/21 0:59
*/
public class FileMonitor {
/**
* 每2秒更新的默认监控器
*/
private static FileMonitor defaultFileMonitor = new FileMonitor(2 * 1000);
private Timer timer_;
private HashMap files_; // File -> Long
private List fileEntrys = new java.util.concurrent.CopyOnWriteArrayList();
private Collection> listeners_; // of WeakReference(FileListener)
private long pollingInterval = 10000;
public static FileMonitor getDefaultFileMonitor() {
return defaultFileMonitor;
}
/**
* Create a file monitor instance with specified polling interval.
*
* @param pollingInterval Polling interval in milli seconds.
*/
public FileMonitor(long pollingInterval) {
this.pollingInterval = pollingInterval;
files_ = new HashMap();
listeners_ = new ArrayList>();
timer_ = new Timer("FileMonitor", true);
timer_.schedule(new FileMonitorNotifier(), 0, pollingInterval);
}
/**
* Stop the file monitor polling.
*/
public void stop() {
timer_.cancel();
timer_ = null;
}
public void start() {
if (timer_ == null) {
timer_ = new Timer(true);
timer_.schedule(new FileMonitorNotifier(), 0, pollingInterval);
}
}
/**
* Add file to listen for. File may be any java.io.File (including a
* directory) and may well be a non-existing file in the case where the
* creating of the file is to be trepped.
*
* More than one file can be listened for. When the specified file is
* created, modified or deleted, listeners are notified.
*
* @param file File to listen for.
*/
public void addFile(String id, File file) {
if (!files_.containsKey(file)) {
FileEntry entry = new FileEntry(id, file, file.exists() ? file.lastModified() : -1);
files_.put(file, entry);
}
}
/**
* 添加监控文件实体。
*
* @param fileEntry
*/
public void addFileEntry(FileEntry fileEntry) {
if (!fileEntrys.contains(fileEntry)) {
fileEntrys.add(fileEntry);
}
}
/**
* 通过文件实体的标识判断监控文件实体是否存在。
*
* @param id
* @return
*/
public boolean fileEntryExists(String id) {
if (id == null) {
return false;
}
for (int i = 0; i < fileEntrys.size(); i++) {
if (id.equals(fileEntrys.get(i).getId())) {
return true;
}
}
return false;
}
/**
* 通过文件实体标识删除一个监控文件实体。
*
* @param id
*/
public void removeFileEntry(String id) {
if (id == null) {
return;
}
for (int i = 0; i < fileEntrys.size(); i++) {
if (id.equals(fileEntrys.get(i).getId())) {
fileEntrys.remove(i);
return;
}
}
}
/**
* Remove specified file for listening.
*
* @param file File to remove.
*/
public void removeFile(File file) {
files_.remove(file);
}
/**
* Add listener to this file monitor.
*
* @param fileListener Listener to add.
*/
public void addListener(FileListener fileListener) {
// Don't add if its already there
for (Iterator> i = listeners_.iterator(); i.hasNext(); ) {
WeakReference reference = i.next();
FileListener listener = (FileListener) reference.get();
if (listener == fileListener) {
return;
}
}
// Use WeakReference to avoid memory leak if this becomes the
// sole reference to the object.
listeners_.add(new WeakReference(fileListener));
}
/**
* Remove listener from this file monitor.
*
* @param fileListener Listener to remove.
*/
public void removeListener(FileListener fileListener) {
for (Iterator> i = listeners_.iterator(); i.hasNext(); ) {
WeakReference reference = (WeakReference) i.next();
FileListener listener = (FileListener) reference.get();
if (listener == fileListener) {
i.remove();
break;
}
}
}
/**
* This is the timer thread which is executed every n milliseconds according
* to the setting of the file monitor. It investigates the file in question
* and notify listeners if changed.
*/
private class FileMonitorNotifier extends TimerTask {
@Override
public void run() {
try {
for (Iterator i = fileEntrys.iterator(); i.hasNext(); ) {
try {
FileEntry entry = i.next();
if (entry == null || !entry.check()) {
i.remove();
}
} catch (Throwable t) {
t.printStackTrace();
System.out.println("执行文件监控发生错误:" + t.getMessage());
}
}
// Loop over the registered files and see which have changed.
// Use a copy of the list in case listener wants to alter the
// list within its fileChanged method.
Collection files = new ArrayList(files_.keySet());
for (Iterator i = files.iterator(); i.hasNext(); ) {
File file = i.next();
try {
FileEntry fileEntry = files_.get(file);
long lastModifiedTime = fileEntry.getLastModified();
long newModifiedTime = file.exists() ? file.lastModified() : -1;
//logger.debug(file.getAbsolutePath());
//logger.debug(" {}=>{}", lastModifiedTime, newModifiedTime);
// Chek if file has changed
if (newModifiedTime != lastModifiedTime) {
//logger.debug("file changed {})", file.getAbsolutePath());
fileEntry.setLastModified(newModifiedTime);
// Register new modified time
files_.put(file, fileEntry);
if (fileEntry.getFileListener() != null) {
fileEntry.getFileListener().fileChanged(fileEntry);
} else {
// Notify listeners
for (Iterator> j =
listeners_.iterator(); j.hasNext(); ) {
WeakReference reference =
(WeakReference) j.next();
FileListener listener = (FileListener) reference.get();
// Remove from list if the back-end object has been GC'd
if (listener == null) {
j.remove();
} else {
listener.fileChanged(fileEntry);
}
}
}
}
} catch (Throwable t) {
if (file != null) {
t.printStackTrace();
System.out.println(
"file monitor execute error, file=" + file.getAbsolutePath() +
t.getMessage());
} else {
System.out.println(
"file monitor execute error, file=null" + t.getMessage());
}
}
}
} catch (Throwable t) {
System.out.println("执行文件监控发生错误" + t.getMessage());
}
}
}
}
package com.yx.demo.filemonitor;
/**
* FileListener
*
* @author yx
* @date 2019/12/21 0:55
*/
public interface FileListener {
/**
*
* @param fileEntry
*/
public void fileChanged(FileEntry fileEntry);
}
