com.ormanli.duplicatefinder.util.FileUtil.java Source code

Introduction

Here is the source code for com.ormanli.duplicatefinder.util.FileUtil.java

Source

/*******************************************************************************
 * Copyright (c) 2013 Serdar Ormanl.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v2.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * 
 * Contributors:
 *     Serdar Ormanl - initial API and implementation
 ******************************************************************************/
package com.ormanli.duplicatefinder.util;

import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ConcurrentLinkedQueue;

import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import com.google.common.base.Predicate;
import com.google.common.collect.Lists;
import com.google.common.collect.Queues;
import com.google.common.io.Files;

public class FileUtil {

    private static final Logger logger = LogManager.getLogger(FileUtil.class);
    private volatile ConcurrentLinkedQueue<List<File>> queue;

    public FileUtil(String path) {
        List<File> fileList = getFileList(path);
        queue = Queues.newConcurrentLinkedQueue(Lists.partition(fileList,
                (int) Math.ceil(fileList.size() / (double) Runtime.getRuntime().availableProcessors())));
    }

    private static Predicate<File> isEven = new Predicate<File>() {
        @Override
        public boolean apply(File file) {
            return file.isFile();
        }
    };

    private static List<File> getFileList(String directory) {
        try {
            return Files.fileTreeTraverser().breadthFirstTraversal(new File(directory)).filter(isEven).toList();
        } catch (Exception e) {
            logger.error(StringUtils.EMPTY, e);
        }

        return new ArrayList<File>();
    }

    public List<File> getEntryList() throws Exception {
        logger.info(queue);
        return queue.poll();
    }

    public String getFileHash(File file) throws Exception {
        return DigestUtils.sha256Hex(Files.toByteArray(file));
    }
}