Merge commit '42b2e5ca519766e37ce6941ba4faecc9691cc403' into upstream

[debian/openrocket] / core / src / net / sf / openrocket / startup / ConcurrentComponentPresetDatabaseLoader.java

diff --git a/core/src/net/sf/openrocket/startup/ConcurrentComponentPresetDatabaseLoader.java b/core/src/net/sf/openrocket/startup/ConcurrentComponentPresetDatabaseLoader.java
new file mode 100644 (file)
index 0000000..79838ec
--- /dev/null
+++ b/core/src/net/sf/openrocket/startup/ConcurrentComponentPresetDatabaseLoader.java
@@ -0,0 +1,140 @@
+package net.sf.openrocket.startup;
+
+import java.io.InputStream;
+import java.util.Collection;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ThreadFactory;
+import java.util.concurrent.TimeUnit;
+
+import net.sf.openrocket.database.ComponentPresetDatabase;
+import net.sf.openrocket.file.iterator.DirectoryIterator;
+import net.sf.openrocket.file.iterator.FileIterator;
+import net.sf.openrocket.gui.util.SimpleFileFilter;
+import net.sf.openrocket.logging.LogHelper;
+import net.sf.openrocket.preset.ComponentPreset;
+import net.sf.openrocket.preset.xml.OpenRocketComponentLoader;
+import net.sf.openrocket.util.Pair;
+
+public class ConcurrentComponentPresetDatabaseLoader {
+
+       private static final LogHelper log = Application.getLogger();
+       private static final String SYSTEM_PRESET_DIR = "datafiles/presets";
+
+       private final CountDownLatch latch = new CountDownLatch(1);
+
+       private final ComponentPresetDatabase componentPresetDao;
+
+       private final ExecutorService writerPool;
+
+       private final ExecutorService loaderPool;
+
+       private final Thread workGenerator;
+
+       private FileIterator iterator;
+
+       private long startTime;
+       private long fileCount = 0;
+       private long presetCount = 0;
+
+       ConcurrentComponentPresetDatabaseLoader( ComponentPresetDatabase componentPresetDao ) {
+               this.componentPresetDao = componentPresetDao;
+
+               writerPool = Executors.newSingleThreadExecutor(new ThreadFactory() {
+                       @Override
+                       public Thread newThread(Runnable r) {
+                               Thread t = new Thread(r,"PresetWriterThread");
+                               return t;
+                       }
+               });
+
+               loaderPool = Executors.newFixedThreadPool(3, new ThreadFactory() {
+                       int threadCount = 0;
+                       @Override
+                       public Thread newThread(Runnable r) {
+                               Thread t = new Thread(r,"PresetLoaderPool-" + threadCount++);
+                               t.setPriority(Thread.MIN_PRIORITY);
+                               return t;
+                       }
+
+               });
+
+               workGenerator = new Thread( new WorkGenerator(),"PresetGeneratorThread");
+       }
+
+       public void load() {
+               startTime = System.currentTimeMillis();
+               workGenerator.start();
+       }
+
+       public void await() throws InterruptedException {
+               latch.await();
+               loaderPool.shutdown();
+               loaderPool.awaitTermination(90, TimeUnit.SECONDS);
+               writerPool.shutdown();
+               writerPool.awaitTermination(90, TimeUnit.SECONDS);
+               if ( iterator != null ) {
+                       iterator.close();
+               }
+               long end = System.currentTimeMillis();
+               log.debug("Time to load presets: " + (end-startTime) + "ms " + presetCount + " loaded from " + fileCount + " files");
+       }
+
+
+       private class WorkGenerator implements Runnable {
+               @Override
+               public void run() {
+                       // Start loading
+                       log.info("Loading component presets from " + SYSTEM_PRESET_DIR);
+
+                       iterator = DirectoryIterator.findDirectory(SYSTEM_PRESET_DIR,
+                                       new SimpleFileFilter("", false, "orc"));
+
+                       if (iterator != null) {
+                               while( iterator.hasNext() ) {
+                                       Pair<String,InputStream> f = iterator.next();
+                                       FileLoader loader = new FileLoader( f.getV(), f.getU() );
+                                       loaderPool.execute(loader);
+                                       fileCount ++;
+                               }
+                       }
+                       latch.countDown();
+               }
+       }
+
+       private class FileLoader implements Runnable {
+               private final InputStream is;
+               private final String fileName;
+
+               public FileLoader(InputStream is, String fileName) {
+                       super();
+                       this.is = is;
+                       this.fileName = fileName;
+               }
+
+               @Override
+               public void run() {
+                       OpenRocketComponentLoader loader = new OpenRocketComponentLoader();
+                       Collection<ComponentPreset> presets = loader.load(is, fileName);
+                       PresetWriter writer = new PresetWriter(presets);
+                       writerPool.execute(writer);
+               }               
+       }
+
+       private class PresetWriter implements Runnable {
+               private final Collection<ComponentPreset> presets;
+
+               public PresetWriter(Collection<ComponentPreset> presets) {
+                       super();
+                       this.presets = presets;
+               }
+
+               @Override
+               public void run() {
+                       presetCount += presets.size();
+                       componentPresetDao.addAll(presets);
+               }
+
+       }
+}