diff src/main/java/app/bbs/NetworkJungleBulletinBoard.java @ 135:2e8034524259

Added NetworkJournal and SingletonMessageFromAlice class
author Nobuyasu Oshiro <dimolto@cr.ie.u-ryukyu.ac.jp>
date Sun, 12 Jan 2014 06:18:37 +0900
parents 00fcb468de27
children c1d75b031b15
line wrap: on
line diff
--- a/src/main/java/app/bbs/NetworkJungleBulletinBoard.java	Sat Jan 11 09:33:45 2014 +0900
+++ b/src/main/java/app/bbs/NetworkJungleBulletinBoard.java	Sun Jan 12 06:18:37 2014 +0900
@@ -1,17 +1,24 @@
 package app.bbs;
 
+import java.io.File;
+import java.io.IOException;
 import java.nio.ByteBuffer;
 import java.util.Date;
 import java.util.concurrent.atomic.AtomicInteger;
 
 import alice.jungle.core.NetworkDefaultJungle;
 import alice.jungle.persistent.AliceJournal;
+import alice.jungle.persistent.NetworkJournal;
+import alice.jungle.persistent.PersistentJournal;
+import alice.jungle.transaction.JungleUpdater;
 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.Jungle;
 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTree;
 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTreeEditor;
 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.bbs.BoardMessage;
 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.core.Children;
 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.core.Node;
+import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.persistent.ChangeList;
+import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.persistent.ChangeListReader;
 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultNodePath;
 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultTreeEditor;
 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.trasnformer.EditableNode;
@@ -25,16 +32,85 @@
 public class NetworkJungleBulletinBoard implements NetworkBulletinBoard
 {
 	private final Jungle jungle;
-		
+	private final NetworkJournal journal;
+	private final String LOG_DIR;
+	private Boolean persistentFlag;
+	
+	private NetworkJungleBulletinBoard(String _uuid, NetworkJournal _journal) 
+	{
+		journal = _journal;
+		jungle = new NetworkDefaultJungle(journal, _uuid,new DefaultTreeEditor(new DefaultTraverser()));
+		BulletinBoardJungleManager.setJungle(jungle);
+		persistentFlag = false;
+		LOG_DIR = "./log";
+	}
+	
 	public NetworkJungleBulletinBoard(String _uuid)
 	{
-		jungle = new NetworkDefaultJungle(new AliceJournal(), _uuid,new DefaultTreeEditor(new DefaultTraverser()));	
+		this(_uuid, new AliceJournal());
 		jungle.createNewTree("boards");
-		BulletinBoardJungleManager.setJungle(jungle);
+	}
+	
+	public static NetworkBulletinBoard NewPersistentJungle(String _uuid)
+	{
+		NetworkJungleBulletinBoard board = new NetworkJungleBulletinBoard(_uuid, new PersistentJournal());
+		board.persistentFlag = true;
+		return board;
 	}
 	
 	public void init() {
-		
+		if(!persistentFlag) {
+			return;
+		}
+		checkAndCreateLogDirectory();
+		try {
+			commitLogRecover();
+		} catch (IOException e) {
+			e.printStackTrace();
+		}
+	}
+	
+	public void checkAndCreateLogDirectory() {
+		File logFile = new File(LOG_DIR);
+		if(!logFile.exists()) {
+			logFile.mkdir();
+			return;
+		}
+		if (logFile.isFile()) {
+			logFile.delete();
+			logFile.mkdir();
+		}
+	}
+	
+	public void commitLogRecover() throws IOException {
+		File[] logFiles = new File(LOG_DIR).listFiles(); 
+		for(File logFile : logFiles) {
+			commitLogRecover(logFile);
+			logFile.delete();
+		}
+		if(jungle.getTreeByName("boards") == null) {
+			jungle.createNewTree("boards");
+		}
+	}
+
+	private void commitLogRecover(File logFile) throws IOException {
+		journal.setInputFile(logFile);
+		ChangeListReader reader = journal.getReader();
+		if (reader == null) return;
+		for (ChangeList chList : reader) {
+			String treeName = chList.getTreeName();
+			JungleTree tree = jungle.getTreeByName(treeName);
+			if(tree == null) {
+				tree = jungle.createNewTree(treeName);
+			}
+			JungleTreeEditor editor = tree.getLocalTreeEditor();
+			Either<Error, JungleTreeEditor> either = JungleUpdater.edit(editor, chList);
+			editor = either.b();
+			if(either.isA()) {
+				throw new IOException("Failed commit log recovery");
+			}
+			editor.success();
+		}
 	}
 
 	public Iterable<String> getBoards()