Add support for room inMemory databases

anandgaurav10 · anandgaurav10 · commit bd32b4851608 · 2018-02-12T20:02:46.000+05:30
diff --git a/app/src/main/java/com/sample/MainActivity.java b/app/src/main/java/com/sample/MainActivity.java
@@ -138,6 +138,7 @@ protected void onCreate(Bundle savedInstanceState) {
         }
 
         Utils.setCustomDatabaseFiles(getApplicationContext());
+        Utils.setInMemoryRoomDatabases(userDBHelper.getInMemoryDatabase());
     }
 
     public void showDebugDbAddress(View view) {
diff --git a/app/src/main/java/com/sample/database/room/UserDBHelper.java b/app/src/main/java/com/sample/database/room/UserDBHelper.java
@@ -1,5 +1,6 @@
 package com.sample.database.room;
 
+import android.arch.persistence.db.SupportSQLiteDatabase;
 import android.arch.persistence.room.Room;
 import android.content.Context;
 
@@ -15,7 +16,7 @@ public class UserDBHelper {
     private final AppDatabase inMemoryAppDatabase;
 
     public UserDBHelper(Context context) {
-        appDatabase = Room.databaseBuilder(context, AppDatabase.class, "User-Database")
+        appDatabase = Room.databaseBuilder(context, AppDatabase.class, "User.db")
                 .allowMainThreadQueries()
                 .build();
         inMemoryAppDatabase = Room.inMemoryDatabaseBuilder(context, AppDatabase.class)
@@ -39,4 +40,7 @@ public int countInMemory() {
         return inMemoryAppDatabase.userDao().loadAll().size();
     }
 
+    public SupportSQLiteDatabase getInMemoryDatabase() {
+        return inMemoryAppDatabase.getOpenHelper().getWritableDatabase();
+    }
 }
diff --git a/app/src/main/java/com/sample/utils/Utils.java b/app/src/main/java/com/sample/utils/Utils.java
@@ -19,6 +19,7 @@
 
 package com.sample.utils;
 
+import android.arch.persistence.db.SupportSQLiteDatabase;
 import android.content.Context;
 import android.util.Pair;
 import android.widget.Toast;
@@ -71,4 +72,20 @@ public static void setCustomDatabaseFiles(Context context) {
         }
     }
 
+    public static void setInMemoryRoomDatabases(SupportSQLiteDatabase... database) {
+        if (BuildConfig.DEBUG) {
+            try {
+                Class<?> debugDB = Class.forName("com.amitshekhar.DebugDB");
+                Class[] argTypes = new Class[]{HashMap.class};
+                HashMap<String, SupportSQLiteDatabase> inMemoryDatabases = new HashMap<>();
+                // set your inMemory database
+                inMemoryDatabases.put("InMemoryOne.db", database[0]);
+                Method setRoomInMemoryDatabase = debugDB.getMethod("setInMemoryRoomDatabases", argTypes);
+                setRoomInMemoryDatabase.invoke(null, inMemoryDatabases);
+            } catch (Exception ignore) {
+
+            }
+        }
+    }
+
 }
diff --git a/debug-db/src/main/java/com/amitshekhar/DebugDB.java b/debug-db/src/main/java/com/amitshekhar/DebugDB.java
@@ -19,6 +19,7 @@
 
 package com.amitshekhar;
 
+import android.arch.persistence.db.SupportSQLiteDatabase;
 import android.content.Context;
 import android.util.Log;
 import android.util.Pair;
@@ -79,6 +80,12 @@ public static void setCustomDatabaseFiles(HashMap<String, Pair<File, String>> cu
         }
     }
 
+    public static void setInMemoryRoomDatabases(HashMap<String, SupportSQLiteDatabase> databases) {
+        if (clientServer != null) {
+            clientServer.setInMemoryRoomDatabases(databases);
+        }
+    }
+
     public static boolean isServerRunning() {
         return clientServer != null && clientServer.isRunning();
     }
diff --git a/debug-db/src/main/java/com/amitshekhar/server/ClientServer.java b/debug-db/src/main/java/com/amitshekhar/server/ClientServer.java
@@ -24,6 +24,7 @@
  */
 
 
+import android.arch.persistence.db.SupportSQLiteDatabase;
 import android.content.Context;
 import android.util.Log;
 import android.util.Pair;
@@ -40,13 +41,10 @@ public class ClientServer implements Runnable {
     private static final String TAG = "ClientServer";
 
     private final int mPort;
-
+    private final RequestHandler mRequestHandler;
     private boolean mIsRunning;
-
     private ServerSocket mServerSocket;
 
-    private final RequestHandler mRequestHandler;
-
     public ClientServer(Context context, int port) {
         mRequestHandler = new RequestHandler(context);
         mPort = port;
@@ -91,6 +89,10 @@ public void setCustomDatabaseFiles(HashMap<String, Pair<File, String>> customDat
         mRequestHandler.setCustomDatabaseFiles(customDatabaseFiles);
     }
 
+    public void setInMemoryRoomDatabases(HashMap<String, SupportSQLiteDatabase> databases) {
+        mRequestHandler.setInMemoryRoomDatabases(databases);
+    }
+
     public boolean isRunning() {
         return mIsRunning;
     }
diff --git a/debug-db/src/main/java/com/amitshekhar/server/RequestHandler.java b/debug-db/src/main/java/com/amitshekhar/server/RequestHandler.java
@@ -19,6 +19,7 @@
 
 package com.amitshekhar.server;
 
+import android.arch.persistence.db.SupportSQLiteDatabase;
 import android.content.Context;
 import android.content.res.AssetManager;
 import android.net.Uri;
@@ -30,6 +31,7 @@
 import com.amitshekhar.model.TableDataResponse;
 import com.amitshekhar.model.UpdateRowResponse;
 import com.amitshekhar.sqlite.DebugSQLiteDB;
+import com.amitshekhar.sqlite.InMemoryDebugSQLiteDB;
 import com.amitshekhar.sqlite.SQLiteDB;
 import com.amitshekhar.utils.Constants;
 import com.amitshekhar.utils.DatabaseFileProvider;
@@ -66,6 +68,7 @@ public class RequestHandler {
     private HashMap<String, Pair<File, String>> mDatabaseFiles;
     private HashMap<String, Pair<File, String>> mCustomDatabaseFiles;
     private String mSelectedDatabase = null;
+    private HashMap<String, SupportSQLiteDatabase> mRoomInMemoryDatabases = new HashMap<>();
 
     public RequestHandler(Context context) {
         mContext = context;
@@ -162,19 +165,25 @@ public void setCustomDatabaseFiles(HashMap<String, Pair<File, String>> customDat
         mCustomDatabaseFiles = customDatabaseFiles;
     }
 
+    public void setInMemoryRoomDatabases(HashMap<String, SupportSQLiteDatabase> databases) {
+        mRoomInMemoryDatabases = databases;
+    }
+
     private void writeServerError(PrintStream output) {
         output.println("HTTP/1.0 500 Internal Server Error");
         output.flush();
     }
 
     private void openDatabase(String database) {
         closeDatabase();
-        File databaseFile = mDatabaseFiles.get(database).first;
-        String password = mDatabaseFiles.get(database).second;
-
-        SQLiteDatabase.loadLibs(mContext);
-
-        sqLiteDB = new DebugSQLiteDB(SQLiteDatabase.openOrCreateDatabase(databaseFile.getAbsolutePath(), password, null));
+        if (mRoomInMemoryDatabases.containsKey(database)) {
+            sqLiteDB = new InMemoryDebugSQLiteDB(mRoomInMemoryDatabases.get(database));
+        } else {
+            File databaseFile = mDatabaseFiles.get(database).first;
+            String password = mDatabaseFiles.get(database).second;
+            SQLiteDatabase.loadLibs(mContext);
+            sqLiteDB = new DebugSQLiteDB(SQLiteDatabase.openOrCreateDatabase(databaseFile.getAbsolutePath(), password, null));
+        }
         isDbOpened = true;
     }
 
@@ -198,6 +207,12 @@ private String getDBListResponse() {
                 response.rows.add(dbEntry);
             }
         }
+        if (mRoomInMemoryDatabases != null) {
+            for (HashMap.Entry<String, SupportSQLiteDatabase> entry : mRoomInMemoryDatabases.entrySet()) {
+                String[] dbEntry = {entry.getKey(), "false"};
+                response.rows.add(dbEntry);
+            }
+        }
         response.rows.add(new String[]{Constants.APP_SHARED_PREFERENCES, "false"});
         response.isSuccessful = true;
         return mGson.toJson(response);
diff --git a/debug-db/src/main/java/com/amitshekhar/sqlite/InMemoryDebugSQLiteDB.java b/debug-db/src/main/java/com/amitshekhar/sqlite/InMemoryDebugSQLiteDB.java
@@ -0,0 +1,59 @@
+package com.amitshekhar.sqlite;
+
+import android.arch.persistence.db.SupportSQLiteDatabase;
+import android.content.ContentValues;
+import android.database.Cursor;
+import android.database.SQLException;
+
+/**
+ * Created by anandgaurav on 12/02/18.
+ */
+
+public class InMemoryDebugSQLiteDB implements SQLiteDB {
+
+    private final SupportSQLiteDatabase database;
+
+    public InMemoryDebugSQLiteDB(SupportSQLiteDatabase database) {
+        this.database = database;
+    }
+
+    @Override
+    public int delete(String table, String whereClause, String[] whereArgs) {
+        return database.delete(table, whereClause, whereArgs);
+    }
+
+    @Override
+    public boolean isOpen() {
+        return database.isOpen();
+    }
+
+    @Override
+    public void close() {
+        // no ops
+    }
+
+    @Override
+    public Cursor rawQuery(String sql, String[] selectionArgs) {
+        return database.query(sql, selectionArgs);
+    }
+
+    @Override
+    public void execSQL(String sql) throws SQLException {
+        database.execSQL(sql);
+    }
+
+    @Override
+    public long insert(String table, String nullColumnHack, ContentValues values) {
+        return database.insert(table, 0, values);
+    }
+
+    @Override
+    public int update(String table, ContentValues values, String whereClause, String[] whereArgs) {
+        return database.update(table, 0, values, whereClause, whereArgs);
+    }
+
+    @Override
+    public int getVersion() {
+        return database.getVersion();
+    }
+}
diff --git a/debug-db/src/main/java/com/amitshekhar/utils/Utils.java b/debug-db/src/main/java/com/amitshekhar/utils/Utils.java
@@ -84,7 +84,7 @@ public static byte[] loadContent(String fileName, AssetManager assetManager) thr
     }
 
     public static byte[] getDatabase(String selectedDatabase, HashMap<String, Pair<File, String>> databaseFiles) {
-        if (TextUtils.isEmpty(selectedDatabase)) {
+        if (TextUtils.isEmpty(selectedDatabase) || !databaseFiles.containsKey(selectedDatabase)) {
             return null;
         }
 

Original file line number	Diff line number	Diff line change
`@@ -138,6 +138,7 @@ protected void onCreate(Bundle savedInstanceState) {`
`138`	`138`	`}`
`139`	`139`
`140`	`140`	`Utils.setCustomDatabaseFiles(getApplicationContext());`
	`141`	`+ Utils.setInMemoryRoomDatabases(userDBHelper.getInMemoryDatabase());`
`141`	`142`	`}`
`142`	`143`
`143`	`144`	`public void showDebugDbAddress(View view) {`
Original file line number	Diff line number	Diff line change
`@@ -19,6 +19,7 @@`
`19`	`19`
`20`	`20`	`package com.amitshekhar;`
`21`	`21`
	`22`	`+import android.arch.persistence.db.SupportSQLiteDatabase;`
`22`	`23`	`import android.content.Context;`
`23`	`24`	`import android.util.Log;`
`24`	`25`	`import android.util.Pair;`
`@@ -79,6 +80,12 @@ public static void setCustomDatabaseFiles(HashMap<String, Pair<File, String>> cu`
`79`	`80`	`}`
`80`	`81`	`}`
`81`	`82`
	`83`	`+ public static void setInMemoryRoomDatabases(HashMap<String, SupportSQLiteDatabase> databases) {`
	`84`	`+ if (clientServer != null) {`
	`85`	`+ clientServer.setInMemoryRoomDatabases(databases);`
	`86`	`+ }`
	`87`	`+ }`
	`88`	`+`
`82`	`89`	`public static boolean isServerRunning() {`
`83`	`90`	`return clientServer != null && clientServer.isRunning();`
`84`	`91`	`}`
Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,7 @@ public static byte[] loadContent(String fileName, AssetManager assetManager) thr`
`84`	`84`	`}`
`85`	`85`
`86`	`86`	`public static byte[] getDatabase(String selectedDatabase, HashMap<String, Pair<File, String>> databaseFiles) {`
`87`		`- if (TextUtils.isEmpty(selectedDatabase)) {`
	`87`	`+ if (TextUtils.isEmpty(selectedDatabase) \|\| !databaseFiles.containsKey(selectedDatabase)) {`
`88`	`88`	`return null;`
`89`	`89`	`}`
`90`	`90`