ca.ualberta.physics.cssdp.file.remote.command.RecursiveLs.java Source code

Introduction

Here is the source code for ca.ualberta.physics.cssdp.file.remote.command.RecursiveLs.java

Source

/* ============================================================
 * RecursiveLs.java
 * ============================================================
 * Copyright 2013 University of Alberta
 *
 * 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 ca.ualberta.physics.cssdp.file.remote.command;

import java.util.ArrayList;
import java.util.List;

import org.joda.time.LocalDateTime;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import ca.ualberta.physics.cssdp.domain.file.Host.Protocol;
import ca.ualberta.physics.cssdp.domain.file.RemoteFile;
import ca.ualberta.physics.cssdp.file.remote.protocol.RemoteConnection;
import ca.ualberta.physics.cssdp.util.UrlParser;

import com.google.common.base.Strings;
import com.google.common.base.Throwables;

public class RecursiveLs extends RemoteServerCommand<List<RemoteFile>> {

    private static final long serialVersionUID = 1L;

    private static final Logger logger = LoggerFactory.getLogger(RecursiveLs.class);

    private final String rootDir;
    private final int maxDepth;

    private final List<RemoteFile> fileList = new ArrayList<RemoteFile>();

    public RecursiveLs(String hostname, String rootDir, Integer maxDepth) {
        super(hostname);
        if (Strings.isNullOrEmpty(rootDir)) {
            throw new IllegalArgumentException("Must supply a root dir to start listing at");
        }

        if (rootDir.length() > 1)
            this.rootDir = rootDir.endsWith("/") ? rootDir.substring(0, rootDir.length() - 1) : rootDir;
        else
            this.rootDir = rootDir;

        if (maxDepth != null && maxDepth > 250) {
            error("Max Depth must be <= 250");
            this.maxDepth = 0;
        } else {
            this.maxDepth = maxDepth == null ? 0 : maxDepth;
        }
    }

    @Override
    public void execute(RemoteConnection connection) {

        if (!hasError()) {
            try {
                connection.connect();
                String rootUrl = connection.getHostEntry().getProtocol().name() + "://"
                        + (connection.getHostEntry().getProtocol().equals(Protocol.file) ? ""
                                : connection.getHostEntry().getHostname())
                        + rootDir;

                RemoteFile root = new RemoteFile(rootUrl, 0, new LocalDateTime(), true);

                fileList.addAll(visit(connection, root));

            } catch (Exception e) {
                logger.error("Recursive LS failed ", e);
                error(Throwables.getRootCause(e).getMessage());
            } finally {
                connection.disconnect();
                setDone(true);
            }

        }

    }

    private int currentDepth = 0;

    private List<RemoteFile> visit(RemoteConnection connection, RemoteFile remoteFile) {
        logger.info("RemoteFileUrl: " + remoteFile.getUrl());
        currentDepth++;
        String url = remoteFile.getUrl();
        String path = UrlParser.getPath(url);
        List<RemoteFile> contents = connection.ls(path);
        List<RemoteFile> files = new ArrayList<RemoteFile>();
        for (RemoteFile file : contents) {
            if (file.isDir()) {
                if (currentDepth <= maxDepth) {
                    files.addAll(visit(connection, file));
                    currentDepth--;
                }
            } else {
                files.add(file);
            }
        }
        return files;
    }

    @Override
    public List<RemoteFile> getResult() {
        return fileList;
    }

    @Override
    public String _pk() {
        return getHostname() + rootDir + maxDepth;
    }
}