public void testSimpleFileObject() throws IOException {
FileSystemManager fs = new FileSystemManager();
//
String pathFile = "com/sinosoft/one/mvc/scanning/vfs/SimpleFileObject.class";
URL urlFile = loader.getResource(pathFile);
assertNotNull(urlFile);
FileObject fileObjectFile = fs.resolveFile(urlFile);
assertEquals(SimpleFileObject.class, fileObjectFile.getClass());
//
String pathDir = new File(urlFile.getPath()).getParent().replace('\\', '/');
pathDir = StringUtils.removeEnd(pathDir, "/");
URL urlDir = ResourceUtils.getURL(pathDir);
assertNotNull(urlDir);
File fileDir = new File(urlDir.getFile());
assertTrue(fileDir.exists());
FileObject fileObjectDir = fs.resolveFile(urlDir);
assertEquals(SimpleFileObject.class, fileObjectDir.getClass());
File dirFile = ResourceUtils.getFile(urlDir);
assertTrue(urlDir.toString().endsWith("/"));
assertTrue(urlDir.getPath().endsWith("/"));
assertFalse(dirFile.getPath().endsWith("/"));
assertTrue(fileObjectDir.toString().endsWith("/"));
// exists
assertTrue(fileObjectFile.exists());
assertTrue(fileObjectDir.exists());
assertFalse(fileObjectDir.getChild("a_not_exists_file.txt").exists());
// getName
assertEquals("vfs", fileObjectDir.getName().getBaseName());
assertEquals("SimpleFileObject.class", fileObjectFile.getName().getBaseName());
// getRelativeName
assertEquals(
"SimpleFileObject.class",
fileObjectDir.getName().getRelativeName(fileObjectFile.getName()));
assertEquals("", fileObjectDir.getName().getRelativeName(fileObjectDir.getName()));
assertEquals("", fileObjectFile.getName().getRelativeName(fileObjectFile.getName()));
// getType
assertSame(FileType.FOLDER, fileObjectDir.getType());
assertSame(FileType.FILE, fileObjectFile.getType());
// getChild, getParent, and equals, getChildren
assertEquals(fileObjectFile, fileObjectDir.getChild("SimpleFileObject.class"));
assertEquals(fileObjectDir, fileObjectFile.getParent());
assertSame(fileObjectFile, fileObjectDir.getChild("SimpleFileObject.class"));
assertSame(fileObjectDir, fileObjectFile.getParent());
assertTrue(ArrayUtils.contains(fileObjectDir.getChildren(), fileObjectFile));
// getURL
assertEquals(urlDir, fileObjectDir.getURL());
assertEquals(urlFile, fileObjectFile.getURL());
}
private static void delete(FileObject file, boolean recursive) throws FileSystemException {
if (file.exists()) {
if (file.getType() == FileType.FILE) {
file.delete();
} else if (file.getType() == FileType.FOLDER) {
if (file.getChildren().length == 0) {
file.delete();
} else if (recursive) {
file.delete();
}
}
}
}
/**
* Utility function to extract information about files on the user's file systems.
*
* @param context current call Context
* @param paths the list of files for which to return information
* @return list column-oriented table of file information
* @throws FileSystemException
*/
@Internal("file.info")
public static ListVector fileInfo(@Current Context context, StringVector paths)
throws FileSystemException {
DoubleArrayVector.Builder size = new DoubleArrayVector.Builder();
LogicalArrayVector.Builder isdir = new LogicalArrayVector.Builder();
IntArrayVector.Builder mode =
(IntArrayVector.Builder)
new IntArrayVector.Builder()
.setAttribute(Symbols.CLASS, StringVector.valueOf("octmode"));
DoubleArrayVector.Builder mtime = new DoubleArrayVector.Builder();
StringVector.Builder exe = new StringVector.Builder();
for (String path : paths) {
if (StringVector.isNA(path)) {
throw new EvalException("invalid filename argument");
}
FileObject file = context.resolveFile(path);
if (file.exists()) {
if (file.getType() == FileType.FILE) {
size.add((int) file.getContent().getSize());
} else {
size.add(0);
}
isdir.add(file.getType() == FileType.FOLDER);
mode.add(mode(file));
try {
mtime.add(file.getContent().getLastModifiedTime());
} catch (Exception e) {
mtime.add(0);
}
exe.add(file.getName().getBaseName().endsWith(".exe") ? "yes" : "no");
} else {
size.addNA();
isdir.addNA();
mode.addNA();
mtime.addNA();
exe.addNA();
}
}
return ListVector.newNamedBuilder()
.add("size", size)
.add("isdir", isdir)
.add("mode", mode)
.add("mtime", mtime)
.add("ctime", mtime)
.add("atime", mtime)
.add("exe", exe)
.build();
}
public void testJarFileObject() throws IOException {
FileSystemManager fs = new FileSystemManager();
URL urlFile = loader.getResource("org/apache/commons/lang/StringUtils.class");
assertTrue(urlFile.toString().startsWith("jar:"));
FileObject fileObjectFile = fs.resolveFile(urlFile);
assertEquals(JarFileObject.class, fileObjectFile.getClass());
//
URL urlDir = new URL(urlFile.toString().substring(0, urlFile.toString().lastIndexOf('/') + 1));
assertNotNull(urlDir);
FileObject fileObjectDir = fs.resolveFile(urlDir);
assertEquals(JarFileObject.class, fileObjectDir.getClass());
// exists
assertTrue(fileObjectFile.exists());
assertTrue(fileObjectDir.exists());
assertFalse(fileObjectDir.getChild("a_not_exists_file.txt").exists());
// getName
assertEquals("lang", fileObjectDir.getName().getBaseName());
assertEquals("StringUtils.class", fileObjectFile.getName().getBaseName());
// getRelativeName
assertEquals(
"StringUtils.class", fileObjectDir.getName().getRelativeName(fileObjectFile.getName()));
assertEquals("", fileObjectDir.getName().getRelativeName(fileObjectDir.getName()));
assertEquals("", fileObjectFile.getName().getRelativeName(fileObjectFile.getName()));
// getType
assertSame(FileType.FOLDER, fileObjectDir.getType());
assertSame(FileType.FILE, fileObjectFile.getType());
// getChild, getParent, and equals
assertEquals(fileObjectFile, fileObjectDir.getChild("StringUtils.class"));
assertEquals(fileObjectDir, fileObjectFile.getParent());
assertSame(fileObjectFile, fileObjectDir.getChild("StringUtils.class"));
assertSame(fileObjectDir, fileObjectFile.getParent());
assertTrue(ArrayUtils.contains(fileObjectDir.getChildren(), fileObjectFile));
// getURL
assertEquals(urlDir, fileObjectDir.getURL());
assertEquals(urlFile, fileObjectFile.getURL());
}
public void updateParentFileCombo(FileObject selectedItem) {
try {
List<FileObject> parentChain = new ArrayList<FileObject>();
// are we a directory?
try {
if (selectedItem.getType() == FileType.FOLDER && selectedItem.getType().hasChildren()) {
// we have real children....
parentChain.add(selectedItem);
}
} catch (Exception e) {
// we are not a folder
}
FileObject parentFileObject;
parentFileObject = selectedItem.getParent();
while (parentFileObject != null) {
parentChain.add(parentFileObject);
parentFileObject = parentFileObject.getParent();
}
File roots[] = File.listRoots();
if (currentPanel != null) {
for (int i = 0; i < roots.length; i++) {
parentChain.add(currentPanel.resolveFile(roots[i].getAbsolutePath()));
}
}
String items[] = new String[parentChain.size()];
int idx = 0;
for (int i = parentChain.size() - 1; i >= 0; i--) {
items[idx++] = ((FileObject) parentChain.get(i)).getName().getURI();
}
openFileCombo.setItems(items);
openFileCombo.select(items.length - 1);
} catch (Exception e) {
e.printStackTrace();
// then let's not update the GUI
}
}
@Invisible
@Internal
public static String setwd(@Current Context context, String workingDirectoryName)
throws FileSystemException {
FileObject newWorkingDirectory = context.resolveFile(workingDirectoryName);
if (!newWorkingDirectory.exists() || newWorkingDirectory.getType() != FileType.FOLDER) {
throw new EvalException("cannot change working directory");
}
String previous = context.getSession().getWorkingDirectory().getName().getURI();
context.getSession().setWorkingDirectory(newWorkingDirectory);
return previous;
}
/**
* Gets the type or storage mode of an object.
*
* @return unix-style file mode integer
*/
private static int mode(FileObject file) throws FileSystemException {
int access = 0;
if (file.isReadable()) {
access += 4;
}
if (file.isWriteable()) {
access += 2;
}
if (file.getType() == FileType.FOLDER) {
access += 1;
}
// i know this is braindead but i can't be bothered
// to do octal math at the moment
String digit = Integer.toString(access);
String octalString = digit + digit + digit;
return Integer.parseInt(octalString, 8);
}
public FileObject open(
Shell applicationShell,
String[] schemeRestrictions,
String initialScheme,
boolean showFileScheme,
String fileName,
String[] fileFilters,
String[] fileFilterNames,
boolean returnUserAuthenticatedFile,
int fileDialogMode,
boolean showLocation,
boolean showCustomUI) {
this.fileDialogMode = fileDialogMode;
this.fileFilters = fileFilters;
this.fileFilterNames = fileFilterNames;
this.applicationShell = applicationShell;
this.showFileScheme = showFileScheme;
this.initialScheme = initialScheme;
this.schemeRestrictions = schemeRestrictions;
this.showLocation = showLocation;
this.showCustomUI = showCustomUI;
FileObject tmpInitialFile = initialFile;
if (defaultInitialFile != null && rootFile == null) {
try {
rootFile = defaultInitialFile.getFileSystem().getRoot();
initialFile = defaultInitialFile;
} catch (FileSystemException ignored) {
// well we tried
}
}
createDialog(applicationShell);
if (!showLocation) {
comboPanel.setParent(fakeShell);
} else {
comboPanel.setParent(customUIPanel);
}
if (!showCustomUI) {
customUIPanel.setParent(fakeShell);
} else {
customUIPanel.setParent(dialog);
}
// create our file chooser tool bar, contains parent folder combo and various controls
createToolbarPanel(dialog);
// create our vfs browser component
createVfsBrowser(dialog);
populateCustomUIPanel(dialog);
if (fileDialogMode == VFS_DIALOG_SAVEAS) {
createFileNamePanel(dialog, fileName);
} else {
// create file filter panel
createFileFilterPanel(dialog);
}
// create our ok/cancel buttons
createButtonPanel(dialog);
initialFile = tmpInitialFile;
// set the initial file selection
try {
if (initialFile != null || rootFile != null) {
vfsBrowser.selectTreeItemByFileObject(initialFile != null ? initialFile : rootFile, true);
updateParentFileCombo(initialFile != null ? initialFile : rootFile);
setSelectedFile(initialFile != null ? initialFile : rootFile);
openFileCombo.setText(
initialFile != null ? initialFile.getName().getURI() : rootFile.getName().getURI());
}
} catch (FileSystemException e) {
MessageBox box = new MessageBox(dialog.getShell());
box.setText(Messages.getString("VfsFileChooserDialog.error")); // $NON-NLS-1$
box.setMessage(e.getMessage());
box.open();
}
// set the size and show the dialog
int height = 550;
int width = 800;
dialog.setSize(width, height);
Rectangle bounds = dialog.getDisplay().getPrimaryMonitor().getClientArea();
int x = (bounds.width - width) / 2;
int y = (bounds.height - height) / 2;
dialog.setLocation(x, y);
dialog.open();
if (rootFile != null && fileDialogMode == VFS_DIALOG_SAVEAS) {
if (!rootFile.getFileSystem().hasCapability(Capability.WRITE_CONTENT)) {
MessageBox messageDialog = new MessageBox(dialog.getShell(), SWT.OK);
messageDialog.setText(Messages.getString("VfsFileChooserDialog.warning")); // $NON-NLS-1$
messageDialog.setMessage(
Messages.getString("VfsFileChooserDialog.noWriteSupport")); // $NON-NLS-1$
messageDialog.open();
}
}
vfsBrowser.fileSystemTree.forceFocus();
while (!dialog.isDisposed()) {
if (!dialog.getDisplay().readAndDispatch()) dialog.getDisplay().sleep();
}
// we just woke up, we are probably disposed already..
if (!dialog.isDisposed()) {
hideCustomPanelChildren();
dialog.dispose();
}
if (okPressed) {
FileObject returnFile = vfsBrowser.getSelectedFileObject();
if (returnFile != null && fileDialogMode == VFS_DIALOG_SAVEAS) {
try {
if (returnFile.getType().equals(FileType.FILE)) {
returnFile = returnFile.getParent();
}
returnFile = returnFile.resolveFile(enteredFileName);
} catch (FileSystemException e) {
e.printStackTrace();
}
}
// put user/pass on the filename so it comes out in the getUri call.
if (!returnUserAuthenticatedFile) {
// make sure to put the user/pass on the url if it's not there
if (returnFile != null && returnFile.getName() instanceof URLFileName) {
URLFileName urlFileName = (URLFileName) returnFile.getName();
if (urlFileName.getUserName() == null || urlFileName.getPassword() == null) {
// set it
String user = "";
String pass = "";
UserAuthenticator userAuthenticator =
DefaultFileSystemConfigBuilder.getInstance()
.getUserAuthenticator(returnFile.getFileSystem().getFileSystemOptions());
if (userAuthenticator != null) {
UserAuthenticationData data =
userAuthenticator.requestAuthentication(AUTHENTICATOR_TYPES);
user = String.valueOf(data.getData(UserAuthenticationData.USERNAME));
pass = String.valueOf(data.getData(UserAuthenticationData.PASSWORD));
try {
user = URLEncoder.encode(user, "UTF-8");
pass = URLEncoder.encode(pass, "UTF-8");
} catch (UnsupportedEncodingException e) {
// ignore, just use the un encoded values
}
}
// build up the url with user/pass on it
int port = urlFileName.getPort();
String portString = (port < 1) ? "" : (":" + port);
String urlWithUserPass =
urlFileName.getScheme()
+ "://"
+ user
+ ":"
+ pass
+ "@"
+ urlFileName.getHostName()
+ portString
+ urlFileName.getPath();
try {
returnFile = currentPanel.resolveFile(urlWithUserPass);
} catch (FileSystemException e) {
// couldn't resolve with user/pass on url??? interesting
e.printStackTrace();
}
}
}
}
return returnFile;
} else {
return null;
}
}
/**
* Search for files that match the given regex pattern and create a list Then process each of
* these files and update the status of the scan on the poll table
*
* @param entry the poll table entry for the scan
* @param fileURI the file or directory to be scanned
*/
private void scanFileOrDirectory(final PollTableEntry entry, String fileURI) {
FileObject fileObject = null;
if (log.isDebugEnabled()) {
log.debug("Scanning directory or file : " + VFSUtils.maskURLPassword(fileURI));
}
boolean wasError = true;
int retryCount = 0;
int maxRetryCount = entry.getMaxRetryCount();
long reconnectionTimeout = entry.getReconnectTimeout();
while (wasError) {
try {
retryCount++;
fileObject = fsManager.resolveFile(fileURI);
if (fileObject == null) {
log.error("fileObject is null");
throw new FileSystemException("fileObject is null");
}
wasError = false;
} catch (FileSystemException e) {
if (retryCount >= maxRetryCount) {
processFailure(
"Repeatedly failed to resolve the file URI: " + VFSUtils.maskURLPassword(fileURI),
e,
entry);
return;
} else {
log.warn(
"Failed to resolve the file URI: "
+ VFSUtils.maskURLPassword(fileURI)
+ ", in attempt "
+ retryCount
+ ", "
+ e.getMessage()
+ " Retrying in "
+ reconnectionTimeout
+ " milliseconds.");
}
}
if (wasError) {
try {
Thread.sleep(reconnectionTimeout);
} catch (InterruptedException e2) {
log.error("Thread was interrupted while waiting to reconnect.", e2);
}
}
}
try {
if (fileObject.exists() && fileObject.isReadable()) {
entry.setLastPollState(PollTableEntry.NONE);
FileObject[] children = null;
try {
children = fileObject.getChildren();
} catch (FileSystemException ignore) {
}
// if this is a file that would translate to a single message
if (children == null || children.length == 0) {
boolean isFailedRecord = false;
if (entry.getMoveAfterMoveFailure() != null) {
isFailedRecord = isFailedRecord(fileObject, entry);
}
if (fileObject.getType() == FileType.FILE && !isFailedRecord) {
if (!entry.isFileLockingEnabled()
|| (entry.isFileLockingEnabled() && VFSUtils.acquireLock(fsManager, fileObject))) {
try {
processFile(entry, fileObject);
entry.setLastPollState(PollTableEntry.SUCCSESSFUL);
metrics.incrementMessagesReceived();
} catch (AxisFault e) {
logException("Error processing File URI : " + fileObject.getName(), e);
entry.setLastPollState(PollTableEntry.FAILED);
metrics.incrementFaultsReceiving();
}
try {
moveOrDeleteAfterProcessing(entry, fileObject);
} catch (AxisFault axisFault) {
logException(
"File object '" + fileObject.getURL().toString() + "' " + "cloud not be moved",
axisFault);
entry.setLastPollState(PollTableEntry.FAILED);
String timeStamp = VFSUtils.getSystemTime(entry.getFailedRecordTimestampFormat());
addFailedRecord(entry, fileObject, timeStamp);
}
if (entry.isFileLockingEnabled()) {
VFSUtils.releaseLock(fsManager, fileObject);
if (log.isDebugEnabled()) {
log.debug(
"Removed the lock file '"
+ fileObject.toString()
+ ".lock' of the file '"
+ fileObject.toString());
}
}
} else if (log.isDebugEnabled()) {
log.debug("Couldn't get the lock for processing the file : " + fileObject.getName());
} else if (isFailedRecord) {
if (entry.isFileLockingEnabled()) {
VFSUtils.releaseLock(fsManager, fileObject);
}
// schedule a cleanup task if the file is there
if (fsManager.resolveFile(fileObject.getURL().toString()) != null
&& removeTaskState == STATE_STOPPED
&& entry.getMoveAfterMoveFailure() != null) {
workerPool.execute(new FileRemoveTask(entry, fileObject));
}
if (log.isDebugEnabled()) {
log.debug(
"File '"
+ fileObject.getURL()
+ "' has been marked as a failed"
+ " record, it will not process");
}
}
}
} else {
int failCount = 0;
int successCount = 0;
if (log.isDebugEnabled()) {
log.debug("File name pattern : " + entry.getFileNamePattern());
}
for (FileObject child : children) {
boolean isFailedRecord = false;
if (entry.getMoveAfterMoveFailure() != null) {
isFailedRecord = isFailedRecord(child, entry);
}
if (log.isDebugEnabled()) {
log.debug("Matching file : " + child.getName().getBaseName());
}
if ((entry.getFileNamePattern() != null)
&& (child.getName().getBaseName().matches(entry.getFileNamePattern()))
&& (!entry.isFileLockingEnabled()
|| (entry.isFileLockingEnabled() && VFSUtils.acquireLock(fsManager, child)))
&& !isFailedRecord) {
try {
if (log.isDebugEnabled()) {
log.debug("Processing file :" + child);
}
processFile(entry, child);
successCount++;
// tell moveOrDeleteAfterProcessing() file was success
entry.setLastPollState(PollTableEntry.SUCCSESSFUL);
metrics.incrementMessagesReceived();
} catch (Exception e) {
logException("Error processing File URI : " + child.getName(), e);
failCount++;
// tell moveOrDeleteAfterProcessing() file failed
entry.setLastPollState(PollTableEntry.FAILED);
metrics.incrementFaultsReceiving();
}
try {
moveOrDeleteAfterProcessing(entry, child);
} catch (AxisFault axisFault) {
logException(
"File object '" + child.getURL().toString() + "'cloud not be moved", axisFault);
failCount++;
entry.setLastPollState(PollTableEntry.FAILED);
String timeStamp = VFSUtils.getSystemTime(entry.getFailedRecordTimestampFormat());
addFailedRecord(entry, child, timeStamp);
}
// if there is a failure or not we'll try to release the lock
if (entry.isFileLockingEnabled()) {
VFSUtils.releaseLock(fsManager, child);
}
} else if (!(!entry.isFileLockingEnabled()
|| (entry.isFileLockingEnabled()
&& VFSUtils.acquireLock(fsManager, fileObject)))
&& log.isDebugEnabled()) {
log.debug("Couldn't get the lock for processing the file : " + child.getName());
} else if (isFailedRecord) {
if (entry.isFileLockingEnabled()) {
VFSUtils.releaseLock(fsManager, child);
VFSUtils.releaseLock(fsManager, fileObject);
}
if (fsManager.resolveFile(child.getURL().toString()) != null
&& removeTaskState == STATE_STOPPED
&& entry.getMoveAfterMoveFailure() != null) {
workerPool.execute(new FileRemoveTask(entry, child));
}
if (log.isDebugEnabled()) {
log.debug(
"File '"
+ fileObject.getURL()
+ "' has been marked as a failed record, it will not "
+ "process");
}
}
}
if (failCount == 0 && successCount > 0) {
entry.setLastPollState(PollTableEntry.SUCCSESSFUL);
} else if (successCount == 0 && failCount > 0) {
entry.setLastPollState(PollTableEntry.FAILED);
} else {
entry.setLastPollState(PollTableEntry.WITH_ERRORS);
}
}
// processing of this poll table entry is complete
long now = System.currentTimeMillis();
entry.setLastPollTime(now);
entry.setNextPollTime(now + entry.getPollInterval());
} else if (log.isDebugEnabled()) {
log.debug(
"Unable to access or read file or directory : "
+ VFSUtils.maskURLPassword(fileURI)
+ "."
+ " Reason: "
+ (fileObject.exists()
? (fileObject.isReadable() ? "Unknown reason" : "The file can not be read!")
: "The file does not exists!"));
}
onPollCompletion(entry);
} catch (FileSystemException e) {
processFailure(
"Error checking for existence and readability : " + VFSUtils.maskURLPassword(fileURI),
e,
entry);
}
}
