se.kth.karamel.common.util.IoUtils.java Source code

Introduction

Here is the source code for se.kth.karamel.common.util.IoUtils.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package se.kth.karamel.common.util;

import com.google.common.base.Charsets;
import com.google.common.io.Files;
import com.google.common.io.Resources;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.Collection;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;

/**
 *
 * @author kamal
 */
public class IoUtils {

    static class Worker implements Runnable {

        ConcurrentHashMap<String, String> map;
        String url;

        public Worker(ConcurrentHashMap<String, String> map, String url) {
            this.map = map;
            this.url = url;
        }

        @Override
        public void run() {
            try {
                String content = readContent(url);
                map.put(url, content);
            } catch (IOException ex) {
            }
        }
    }

    public static List<String> readLines(String url) throws IOException {
        if (Settings.CB_CLASSPATH_MODE) {
            return readLinesFromClasspath(url);
        } else if (Settings.USE_CLONED_REPO_FILES) {
            return readLinesFromPath(url);
        } else {
            return readLinesFromWeb(url);
        }
    }

    public static String readContent(String path) throws IOException {
        if (Settings.CB_CLASSPATH_MODE) {
            return readContentFromClasspath(path);
        } else if (Settings.USE_CLONED_REPO_FILES) {
            return readContentFromPath(path);
        } else {
            return readContentFromWeb(path);
        }
    }

    public static Map<String, String> readContentParallel(Set<String> paths, ExecutorService tp) {
        ConcurrentHashMap<String, String> map = new ConcurrentHashMap<>();
        Set<Worker> workers = new HashSet<>();
        Collection<Future<?>> futures = new LinkedList<>();

        for (String path : paths) {
            Worker worker = new Worker(map, path);
            workers.add(worker);
            futures.add(tp.submit(worker));
        }
        for (Future<?> future : futures) {
            try {
                future.get();
            } catch (InterruptedException ex) {
            } catch (ExecutionException ex) {
            }
        }
        return map;
    }

    public static String readContentFromClasspath(String path) throws IOException {
        URL url = Resources.getResource(path);
        if (url == null) {
            throw new IOException("No config.props file found in cookbook");
        }
        return Resources.toString(url, Charsets.UTF_8);
    }

    public static String readContentFromPath(String path) throws IOException {
        return Files.toString(new File(path), Charsets.UTF_8);
    }

    public static List<String> readLinesFromClasspath(String url) throws IOException {
        return Resources.readLines(Resources.getResource(url), Charsets.UTF_8);
    }

    public static List<String> readLinesFromPath(String url) throws IOException {
        return Files.readLines(new File(url), Charsets.UTF_8);
    }

    public static List<String> readLinesFromWeb(String url) throws IOException {
        URL fileUrl = new URL(url);
        return Resources.readLines(fileUrl, Charsets.UTF_8);
    }

    public static String readContentFromWeb(String url) throws IOException {
        URL fileUrl = new URL(url);
        return Resources.toString(fileUrl, Charsets.UTF_8);
    }
}