= Updated database to allow for multiple tables

src/core/DatabaseManager.java

@@ -5,16 +5,33 @@ import java.util.Vector;
 import utils.GlobalLog;
 import utils.LogFilter;
 
+// This wrangles all of the differentiated storage systems. Internally, these may be tables,
+// different databases, etc -- the goal is that it's abstracted away.
+//
+// TODO: Implement intermediate structure on account of duplicate patterns showing up.
 public class DatabaseManager
 {
 	// Singleton accessor
 	public static DatabaseManager instance = null; 
 
+	// Database names and configs
+	private final String globalTableName = "kitty_globals";     // Table name
+	private final String globalKeyColumnName = "GlobalKey";     // Column name for the key 
+	private final String globalValueColumnName = "GlobalValue"; // Column name for the value
+	
+	private final String characterTableName = "kitty_characters";     // Table name
+	private final String characterKeyColumnName = "CharacterKey";     // Column name for the key 
+	private final String characterValueColumnName = "CharacterValue"; // Column name for the value
+	
+	
 	// Private internal variables
-	private Vector<DatabaseTrackedObject> trackedObjects;
-	private DatabaseDriver driver;
+	private Vector<DatabaseTrackedObject> globalDataTrackedObjects;
+	private Vector<DatabaseTrackedObject> characterDataTrackedObjects;
+	private DatabaseDriverKeyValue globalDataDriver;
+	private DatabaseDriverKeyValue characterDataDriver;
 	private Date lastUpkeep;
 	
+	// Constructor to enforce singleton.
 	public DatabaseManager()
 	{
 		GlobalLog.Log(LogFilter.Database, "Creating database manager");
@@ -29,75 +46,157 @@ public class DatabaseManager
 			return;
 		}
 		
+		// Initialize variables
 		lastUpkeep = new Date();
-		trackedObjects = new Vector<DatabaseTrackedObject>();
-		driver = new DatabaseDriver();
+		globalDataTrackedObjects = new Vector<DatabaseTrackedObject>();
+		characterDataTrackedObjects = new Vector<DatabaseTrackedObject>();
 		
-		if(driver.Connect() == false)
+		// Initialize data sets
+		globalDataDriver = new DatabaseDriverKeyValue(globalTableName, globalKeyColumnName, globalValueColumnName);
+		characterDataDriver = new DatabaseDriverKeyValue(characterTableName, characterKeyColumnName, characterValueColumnName);
+		
+		// Connect data sets
+		if(globalDataDriver.Connect() == false)
 		{
-			GlobalLog.Error("A database failed to connect. Currently, without the DB, this bot can not run.");
+			GlobalLog.Error("Global database failed to connect. Without this DB, this bot can not run.");
+			System.exit(1);
+		}
+		
+		if(characterDataDriver.Connect() == false)
+		{
+			GlobalLog.Error("Character database failed to connect. Without this DB, this bot can not run.");
 			System.exit(1);
 		}
 	}
 	
 	// Thumbs through registered objects and syncs them with the database. 
 	// Consider moving this operation to a separate thread.
-	public int Upkeep()
+	public int upkeep()
 	{
-		synchronized(trackedObjects)
+		int total = 0;
+		
+		total += globalDataUpkeep();
+		total += characterDataUpkeep();
+		
+		lastUpkeep = new Date();
+		return total;
+	}
+	
+	// Global data management and updating, returns how many items were updated.
+	private int globalDataUpkeep()
+	{
+		synchronized(globalDataTrackedObjects)
 		{
 			int numUpdated = 0;
 			
-			for(int i = 0 ; i < trackedObjects.size(); ++i)
+			for(int i = 0; i < globalDataTrackedObjects.size(); ++i)
 			{
-				DatabaseTrackedObject dto = trackedObjects.get(i);
+				DatabaseTrackedObject dto = globalDataTrackedObjects.get(i);
 				
 				if(dto.IsDirty())
 				{
-					SetRemoteValue(dto.identifier, dto.Serialize());
+					globalSetRemoteValue(dto.identifier, dto.Serialize());
 					dto.Resolve();
 					++numUpdated;
 				}
 			}
 			
-			lastUpkeep = new Date();
+			return numUpdated;
+		}
+	}
+	
+	// Character data management and updating, returns how many items were updated.
+	private int characterDataUpkeep()
+	{
+		synchronized(characterDataTrackedObjects)
+		{
+			int numUpdated = 0;
+			
+			for(int i = 0; i < characterDataTrackedObjects.size(); ++i)
+			{
+				DatabaseTrackedObject dto = characterDataTrackedObjects.get(i);
+				
+				if(dto.IsDirty())
+				{
+					characterSetRemoteValue(dto.identifier, dto.Serialize());
+					dto.Resolve();
+					++numUpdated;
+				}
+			}
 			
 			return numUpdated;
 		}
 	}
 	
-	public void Register(DatabaseTrackedObject tracked)
+	
+	  /////////////////
+	 // Global Data //
+	/////////////////
+	public void globalRegister(DatabaseTrackedObject tracked)
 	{
-		synchronized(trackedObjects)
+		synchronized(globalDataTrackedObjects)
 		{
-			trackedObjects.add(tracked);
-			tracked.DeSerialzie(GetRemoteValue(tracked.identifier));
+			globalDataTrackedObjects.add(tracked);
+			tracked.DeSerialzie(globalGetRemoteValue(tracked.identifier));
 		}
 	}
 	
-	public String GetRemoteValue(String key)
+	private String globalGetRemoteValue(String key)
 	{
-		synchronized(driver)
+		synchronized(globalDataDriver)
 		{
-			return driver.CreateGetKey(key);
+			return globalDataDriver.CreateGetKey(key);
 		}
 	}
 	
-	public void SetRemoteValue(String key, String value)
+	private void globalSetRemoteValue(String key, String value)
 	{
-		synchronized(driver)
+		synchronized(globalDataDriver)
 		{
-			driver.CreateSetKey(key, value);
+			globalDataDriver.CreateSetKey(key, value);
 		}
 	}
 	
-	public Date GetLastUpkeep()
+	
+	  ////////////////////
+	 // Character Data //
+	////////////////////
+	public void characterRegister(DatabaseTrackedObject tracked)
+	{
+		synchronized(characterDataTrackedObjects)
+		{
+			characterDataTrackedObjects.add(tracked);
+			tracked.DeSerialzie(characterGetRemoteValue(tracked.identifier));
+		}
+	}
+	
+	private String characterGetRemoteValue(String key)
+	{
+		synchronized(characterDataDriver)
+		{
+			return characterDataDriver.CreateGetKey(key);
+		}
+	}
+	
+	private void characterSetRemoteValue(String key, String value)
+	{
+		synchronized(characterDataDriver)
+		{
+			characterDataDriver.CreateSetKey(key, value);
+		}
+	}
+	
+	
+	  ///////////////////////
+	 // Utility functions //
+	///////////////////////
+	public Date getLastUpkeep()
 	{
 		return lastUpkeep;
 	}
 	
-	public int GetTrackedObjectsSize()
+	public int getTrackedObjectsSize()
 	{
-		return trackedObjects.size();
+		return globalDataTrackedObjects.size();
 	}
 }