com.predic8.membrane.core.resolver.FileSchemaResolver Maven / Gradle / Ivy
/* Copyright 2013 predic8 GmbH, www.predic8.com
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */
package com.predic8.membrane.core.resolver;
import com.google.common.collect.Lists;
import com.predic8.membrane.core.util.functionalInterfaces.ExceptionThrowingConsumer;
import java.io.*;
import java.net.URLDecoder;
import java.nio.file.*;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
public class FileSchemaResolver implements SchemaResolver {
WatchService watchService;
ConcurrentHashMap watchServiceForFile = new ConcurrentHashMap<>();
ConcurrentHashMap> watchedFiles = new ConcurrentHashMap<>();
int fileWatchIntervalInSeconds = 1;
Runnable fileWatchJob = new Runnable() {
@Override
public void run() {
while(watchedFiles.size() > 0){
for(String url : watchServiceForFile.keySet()){
WatchKey wk = watchServiceForFile.get(url);
List> events = wk.pollEvents();
for(WatchEvent event : events){
Path changedFile = ((Path) event.context());
Path urlPath = Paths.get(url).getFileName();
if(changedFile.toString().equals(urlPath.toString())){
try {
ExceptionThrowingConsumer inputStreamConsumer = watchedFiles.get(url);
watchServiceForFile.remove(url);
watchedFiles.remove(url);
inputStreamConsumer.accept(resolve(url));
} catch (Exception ignored) {
}
}
}
}
try {
Thread.sleep(fileWatchIntervalInSeconds*1000);
} catch (InterruptedException ignored) {
}
}
fileWatcher = null;
}
};
Thread fileWatcher = null;
@Override
public List getSchemas() {
return Lists.newArrayList("file", null);
}
public InputStream resolve(String url) throws ResourceRetrievalException {
try {
return new FileInputStream(normalize(url));
} catch (FileNotFoundException e) {
throw new ResourceRetrievalException(url, e);
}
}
@Override
public void observeChange(String url, ExceptionThrowingConsumer consumer) throws ResourceRetrievalException {
url = Paths.get(normalize(url)).toAbsolutePath().toString();
if(watchService == null){
try {
watchService = FileSystems.getDefault().newWatchService();
} catch (IOException ignored) {
}
}
Path path = Paths.get(url).getParent();
WatchKey watchKey = null;
try {
watchKey = path.register(watchService, StandardWatchEventKinds.ENTRY_MODIFY);
} catch (IOException ignored) {
}
watchServiceForFile.put(url,watchKey);
watchedFiles.put(url,consumer);
if(fileWatcher == null){
fileWatcher = new Thread(fileWatchJob);
}
if(!fileWatcher.isAlive()){
fileWatcher.start();
}
}
public static String normalize(String uri) {
if(uri.startsWith("file:///")) {
if (uri.length() > 9 && uri.charAt(9) == '/')
uri = uri.charAt(8) + ":\\" + URLDecoder.decode(uri.substring(9));
else
uri = "/" + URLDecoder.decode(uri.substring(8));
}
if(uri.startsWith("file://")) {
if (uri.length() > 8 && uri.charAt(8) == '/')
uri = uri.charAt(7) + ":\\" + URLDecoder.decode(uri.substring(9));
else
uri = "/" + URLDecoder.decode(uri.substring(7));
}
if(uri.startsWith("file:")) {
uri = URLDecoder.decode(uri.substring(5));
}
return uri;
}
@Override
public List getChildren(String url) {
String[] children = new File(normalize(url)).list();
if (children == null)
return null;
return Arrays.asList(children);
}
@Override
public long getTimestamp(String url) {
return new File(normalize(url)).lastModified();
}
}