diff options
Diffstat (limited to 'src/de/fhswf/in/inf/java1/aufgabe10/ParallelWordCount.java')
| -rw-r--r-- | src/de/fhswf/in/inf/java1/aufgabe10/ParallelWordCount.java | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/src/de/fhswf/in/inf/java1/aufgabe10/ParallelWordCount.java b/src/de/fhswf/in/inf/java1/aufgabe10/ParallelWordCount.java new file mode 100644 index 0000000..5793f02 --- /dev/null +++ b/src/de/fhswf/in/inf/java1/aufgabe10/ParallelWordCount.java @@ -0,0 +1,135 @@ +/** + * + */ +package de.fhswf.in.inf.java1.aufgabe10; + +import java.io.BufferedReader; +import java.io.File; +import java.io.FileReader; +import java.io.IOException; +import java.util.Collections; +import java.util.Set; +import java.util.concurrent.ConcurrentSkipListMap; +import java.util.concurrent.ExecutorService; +import java.util.concurrent.Executors; +import java.util.concurrent.TimeUnit; + +/** + * A class, that counts words in files. + * + * @author $Author: $ + * @version $Revision: $, $Date: $ UTC + */ +public class ParallelWordCount +{ + + private static final int TIMEOUT = 30; + + private ConcurrentSkipListMap<String, Integer> wordMap; + + /** + * Empty constructor. + * + */ + public ParallelWordCount() + { + } + + /** + * Reads a file and counts the words. + * + * @param file + * The file to be read. + */ + public final void readFile(File file) + { + + try (BufferedReader f = new BufferedReader(new FileReader(file))) + { + if (wordMap == null) + { + wordMap = new ConcurrentSkipListMap<>(); + } + + wordMap.clear(); + + String line = null; + + ExecutorService executer = Executors.newFixedThreadPool(Runtime + .getRuntime().availableProcessors()); + + while ((line = f.readLine()) != null) + { + Runnable worker = new LineWordCount(wordMap, line); + executer.execute(worker); + } + + executer.shutdown(); + boolean success = false; + try + { + success = executer.awaitTermination(TIMEOUT, TimeUnit.SECONDS); + } + catch (InterruptedException e) + { + e.printStackTrace(); + } + if (!success) + { + wordMap = null; + throw new IllegalStateException( + "Waiting for threads has timed out."); + } + + } + catch (IOException e) + { + e.printStackTrace(System.err); + } + } + + /** + * Returns all words from the file. + * + * @return An unmodifiable Set of the words from the file. + */ + public final Set<String> getWords() + { + if (wordMap == null) + { + throw new IllegalStateException("First a file must be read."); + } + return Collections.unmodifiableSet(wordMap.keySet()); + } + + /** + * Returns the count of a word in a file. + * + * @param word + * The file to be counted. + * @return Returns the count of findings in the file. + */ + public final int getCount(String word) + { + if (wordMap == null) + { + throw new IllegalStateException("First a file must be read."); + } + if (word == null) + { + throw new IllegalArgumentException("Word must be a valid referece."); + } + if (word.isEmpty()) + { + throw new IllegalArgumentException("The word can't be empty."); + } + + Integer tmp = wordMap.get(word); + if (tmp == null) + { + return 0; + } + return tmp; + } + +} |
