feat: add support for in-memory database

haltcase · Jul 25, 2017 · 8587f4b · 8587f4b
1 parent 071aa5f
commit 8587f4b
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/src/index.js b/src/index.js
@@ -16,18 +16,26 @@ class Trilogy {
     }
 
     let obj = this.options = setup(options)
-    obj.connection.filename = resolve(obj.dir, path)
 
-    // ensure the directory exists
-    makeDirPath(dirname(obj.connection.filename))
+    if (path === ':memory:') {
+      obj.connection.filename = path
+    } else {
+      obj.connection.filename = resolve(obj.dir, path)
+
+      // ensure the directory exists
+      makeDirPath(dirname(obj.connection.filename))
+    }
 
     this.isNative = obj.client === 'sqlite3'
     this.verbose = obj.verbose
 
     let config = { client: 'sqlite3', useNullAsDefault: true }
 
     if (this.isNative) {
-      touchFile(obj.connection.filename)
+      if (path !== ':memory:') {
+        touchFile(obj.connection.filename)
+      }
+
       this.knex = knex({ ...config, connection: obj.connection })
     } else {
       this.knex = knex(config)

diff --git a/src/sqljs-handler.js b/src/sqljs-handler.js
@@ -11,6 +11,10 @@ export function readDatabase (instance) {
   let client
   let { filename } = instance.options.connection
 
+  if (filename === ':memory:') {
+    return new SQL.Database()
+  }
+
   try {
     makeDirPath(dirname(filename))
     let file = readFileSync(filename)
@@ -28,9 +32,11 @@ export function readDatabase (instance) {
 }
 
 export function writeDatabase (instance, db) {
+  let { filename } = instance.options.connection
+  if (filename === ':memory:') return
+
   let data = db.export()
   let buffer = Buffer.from(data)
-  let { filename } = instance.options.connection
 
   makeDirPath(dirname(filename))
   writeFileSync(filename, buffer, { mode: parseInt('0777', 8) })