X-Git-Url: https://git.gag.com/?a=blobdiff_plain;f=core%2Fsrc%2Fnet%2Fsf%2Fopenrocket%2Fstartup%2FConcurrentComponentPresetDatabaseLoader.java;fp=core%2Fsrc%2Fnet%2Fsf%2Fopenrocket%2Fstartup%2FConcurrentComponentPresetDatabaseLoader.java;h=79838ece20c021febf11b407b5f12fbceae2184d;hb=9349577cdfdff682b2aabd6daa24fdc3a7449b58;hp=0000000000000000000000000000000000000000;hpb=30ba0a882f0c061176ba14dbf86d3d6fad096c02;p=debian%2Fopenrocket

diff --git a/core/src/net/sf/openrocket/startup/ConcurrentComponentPresetDatabaseLoader.java b/core/src/net/sf/openrocket/startup/ConcurrentComponentPresetDatabaseLoader.java
new file mode 100644
index 00000000..79838ece
--- /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);
+		}
+
+	}
+}