com.ccoe.build.core.utils.FileUtils.java Source code

Introduction

Here is the source code for com.ccoe.build.core.utils.FileUtils.java

Source

/*
Copyright [2013-2014] eBay Software Foundation
    
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.ccoe.build.core.utils;

import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.FilenameFilter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Map;

public class FileUtils {
    public static final String DONE_EXT = ".done";
    public static final String FAIL_EXT = ".fail";
    public static final String XML_EXT = ".xml";
    private static final long DELETE_RETRY_SLEEP_MILLIS = 10;
    private static final boolean ON_WINDOWS = Os.isFamily("windows");

    public static String readFile(File file) {
        BufferedReader br = null;
        DataInputStream in = null;
        StringBuffer sBuffer = new StringBuffer();

        try {
            in = new DataInputStream(new FileInputStream(file));
            br = new BufferedReader(new InputStreamReader(in));

            String sCurrentLine = null;

            while ((sCurrentLine = br.readLine()) != null) {
                sBuffer.append(sCurrentLine);
                sBuffer.append("\n");
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                in.close();
            } catch (IOException ex) {
                ex.printStackTrace();
            }
        }
        return sBuffer.toString().trim();
    }

    public static void modifyPropertyFile(File file, Map<String, String> map) {
        List<String> postContent = new ArrayList<String>();
        try {
            List<String> originalContent = org.apache.commons.io.FileUtils.readLines(file);
            HashSet<String> keySet = new HashSet<String>();

            for (String line : originalContent) {
                if (line.trim().startsWith("#")) {
                    postContent.add(line);
                    continue;
                }

                String key = line.split("=")[0];
                keySet.add(key);

                if (!map.containsKey(key)) {
                    postContent.add(line);
                } else {
                    postContent.add(key + "=" + map.get(key));
                }
            }

            for (String key : map.keySet()) {
                if (!keySet.contains(key)) {
                    postContent.add(key + "=" + map.get(key));
                }
            }

            System.out.println("keySet : " + keySet);

            org.apache.commons.io.FileUtils.writeLines(file, postContent);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void diskClean(File targetFolder, int retensionDays) {
        File[] doneFiles = loadDoneFiles(targetFolder);
        List<File> filesToDelete = new ArrayList<File>();
        if (doneFiles == null) {
            System.out.println("No .done files found in folder: " + targetFolder);
            return;
        }
        for (File file : doneFiles) {
            long diff = System.currentTimeMillis() - file.lastModified();
            long interval = retensionDays * 24 * 60 * 60 * 1000;

            if (diff > interval) {
                filesToDelete.add(file);
            }
        }

        System.out.println("[INFO] Cleaning up " + filesToDelete.size() + " DONE files older than " + retensionDays
                + " days in target folder: " + targetFolder);
        for (File file : filesToDelete) {
            tryHardToDelete(file);
        }
    }

    public static File[] loadDoneFiles(File targetFolder) {
        return loadFiles(targetFolder, DONE_EXT);
    }

    public static File[] loadFiles(final File targetFolder, final String ext) {
        return targetFolder.listFiles(new FilenameFilter() {
            public boolean accept(File dir, String name) {
                return name.toLowerCase().endsWith(ext);
            }
        });

    }

    public static void renameFile(File file, String ext) {
        if (!file.exists()) {
            return;
        }
        File dest = new File(file.getParent(), file.getName() + ext);
        boolean success = file.renameTo(dest);
        if (success) {
            System.out.println("[INFO] Rename Session LOG " + dest);
        } else {
            System.out.println("[WARNING] Failed rename session LOG to " + dest);
        }
    }

    public static void renameDoneFile(File file) {
        renameFile(file, DONE_EXT);
    }

    public static void renameFailedFile(File file) {
        renameFile(file, FAIL_EXT);
    }

    public static void writeToFile(File targetFile, String body) {

        FileWriter fileWriter = null;
        try {
            fileWriter = new FileWriter(targetFile);
            fileWriter.write(body);
        } catch (IOException ex) {
            ex.printStackTrace();
        } finally {
            try {
                fileWriter.close();
            } catch (IOException ex) {
                ex.printStackTrace();
            }
        }
    }

    public static boolean tryHardToDelete(File f) {
        if (!f.delete()) {
            if (ON_WINDOWS) {
                System.gc();
            }
            try {
                Thread.sleep(DELETE_RETRY_SLEEP_MILLIS);
            } catch (InterruptedException ex) {
                // Ignore Exception
            }
            return f.delete();
        }
        return true;
    }
}