用java线程实现批量修改文件名

 

/**
 * this program TODO
 * @version 1.1 2012-9-1
 * @ausor widjan wu
 */
package file;

import java.io.File;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Scanner;
import java.util.Set;
import java.util.concurrent.*;

public class ChangeFileName
{

    public static void main(String[] args)
    {
        Scanner in = new Scanner(System.in);
        System.out.print("Enter base directory :");
        String directory = in.nextLine();
        System.out.print("Enter key words:");
        String keywords = in.nextLine();

        ExecutorService pool = Executors.newCachedThreadPool();
        ChangeName change = new ChangeName(new File(directory), keywords, pool);

        Future<Integer> result = pool.submit(change);

        try {
            System.out.println(result.get() + "files were changed");
        } catch (ExecutionException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {

        }

        pool.shutdown();

        int largestPoolSize = ((ThreadPoolExecutor) pool).getLargestPoolSize();
        System.out.println("largest pool size :" + largestPoolSize);
    }

}

class ChangeName implements Callable<Integer>
{
    public ChangeName(File directory, String keywords, ExecutorService pool) {
        this.directory = directory;
        this.pool = pool;
        this.keywords = keywords;
    }

    public Integer call()
    {
        count = 0;

        try
        {
            File[] files = directory.listFiles();
            ArrayList<Future<Integer>> results = new ArrayList<Future<Integer>>();

            HashMap failsMap = new HashMap();


            for (File file : files) {
                if (file.isDirectory()) {
                    ChangeName change = new ChangeName(file, keywords, pool);
                    Future<Integer> result = pool.submit(change);
                } else {
                    count++;
                    String path = file.getPath();
                    int index1 = path.lastIndexOf("\\");
                    path = path.substring(0, index1+1);

                    String oldName = file.getName();

                    String fileType = oldName.substring(oldName.lastIndexOf("."));
                    String newFName = path + keywords + count + fileType;
                    boolean success = file.renameTo(new File(newFName));
                    if (!success)
                    {
                        count--;
                        failsMap.put(file, newFName);
                    }
                }
            }

            if(!failsMap.isEmpty())
            {
                Set <File> c = failsMap.keySet();
                String name = "";

                for(File file: c)
                {
                    name = (String) failsMap.get(file);
                    file.renameTo(new File(name));
                }
            }

            for(Future<Integer> result:results)
            {
                try
                {
                    count +=result.get();
                }catch(ExecutionException e)
                {
                    e.printStackTrace();
                }
            }

        }catch(InterruptedException e)
        {

        }

        return count;
    }

    private File directory;
    private String keywords;
    private ExecutorService pool;
    private int count;
}