upgraded to v1.10

Author: alash3al

README.md

@@ -19,7 +19,7 @@ Features
 =========
 - Core data structure: `KV`, `List`, `Hashmap` with advanced implementations.
 - Advanced Publish/Subscribe using webhook and websocket!
-- Pluggable Storage Engine (`badgerdb`, `boltdb`, `leveldb`, `null`, `sqlite`)
+- Pluggable Storage Engine (`badgerdb`, `boltdb`, `leveldb`, `null`)
 - Very compatible with any `redis client` including `redis-cli`
 - Standalone with no external dependencies
 - Helpers commands for `Time`, `Encode <hex|md5|sha1|sha256|sha512> <payload>`, `RANDINT`, `RANDSTR`
@@ -192,6 +192,8 @@ Supported Commands
 - `GC`, runs the Garbage Collector.
 - `ECHO [<arg1> <arg2> ...]`
 - `INFO`
+- `FLUSHDB`, clear the database
+- `FLUSHALL`, clear all databases
 
 TODO
 =====
@@ -204,6 +206,4 @@ TODO
 - [x] Adding BoltDB engine
 - [x] Adding LevelDB engine
 - [x] Adding Null engine
-- [x] Adding SQLite engine
-- [ ] Adding TiKV engine
 - [ ] Adding RAM engine

commands_doc.go.wip

@@ -166,3 +166,16 @@ func infoCommand(c Context) {
 func echoCommand(c Context) {
 	c.WriteString(strings.Join(c.args, " "))
 }
+
+// flushdbCommand - FLUSHDB
+func flushdbCommand(ctx Context) {
+	c := ctx.Context().(map[string]interface{})
+	flushDB(c["db"].(string))
+	ctx.WriteString("OK")
+}
+
+// flushallCommand - FLUSHALL
+func flushallCommand(ctx Context) {
+	flushall()
+	ctx.WriteString("OK")
+}

commands_utils.go

@@ -5,17 +5,15 @@ package main
 
 import (
 	"errors"
+	"os"
 	"path/filepath"
 	"strings"
 
-	"github.com/alash3al/redix/kvstore/sqlite"
-
-	"github.com/alash3al/redix/kvstore/null"
-
 	"github.com/alash3al/redix/kvstore"
 	"github.com/alash3al/redix/kvstore/badgerdb"
 	"github.com/alash3al/redix/kvstore/boltdb"
 	"github.com/alash3al/redix/kvstore/leveldb"
+	"github.com/alash3al/redix/kvstore/null"
 )
 
 // selectDB - load/fetches the requested db
@@ -48,11 +46,24 @@ func openDB(engine, dbpath string) (kvstore.DB, error) {
 		return leveldb.OpenLevelDB(dbpath)
 	case "null":
 		return null.OpenNull()
-	case "sqlite":
-		return sqlite.OpenSQLite(dbpath)
 	}
 }
 
+// flushDB clear the specified database
+func flushDB(n string) {
+	dbpath := filepath.Join(*flagStorageDir, n)
+	os.RemoveAll(dbpath)
+	databases.Delete(n)
+
+	selectDB(n)
+}
+
+// flushall clear all databases
+func flushall() {
+	os.RemoveAll(*flagStorageDir)
+	os.MkdirAll(*flagStorageDir, 0755)
+}
+
 // returns a unique string
 func getUniqueString() string {
 	return snowflakeGenerator.Generate().String()

helpers_db.go

@@ -55,6 +55,11 @@ func OpenBadger(path string) (*BadgerDB, error) {
 	return db, nil
 }
 
+// Close ...
+func (db *BadgerDB) Close() {
+	db.Close()
+}
+
 // Size - returns the size of the database (LSM + ValueLog) in bytes
 func (db *BadgerDB) Size() int64 {
 	lsm, vlog := db.badger.Size()

kvstore/badgerdb/badgerdb.go

@@ -45,6 +45,11 @@ func OpenBolt(path string) (*BoltDB, error) {
 	return db, nil
 }
 
+// Close ...
+func (db *BoltDB) Close() {
+	db.Close()
+}
+
 // Size - returns the size of the database (LSM + ValueLog) in bytes
 func (db *BoltDB) Size() int64 {
 	var size int64

kvstore/boltdb/boltdb.go

@@ -17,6 +17,7 @@ type DB interface {
 	Scan(ScannerOpt ScannerOptions) error
 	Size() int64
 	GC() error
+	Close()
 }
 
 // ScannerOptions - represents the options for a scanner

kvstore/db.go

@@ -53,6 +53,11 @@ func (ldb *LevelDB) Size() int64 {
 	return size
 }
 
+// Close ...
+func (ldb *LevelDB) Close() {
+	ldb.Close()
+}
+
 // GC - runs the garbage collector
 func (ldb *LevelDB) GC() error {
 	return ldb.db.CompactRange(util.Range{})

kvstore/leveldb/leveldb.go

@@ -22,6 +22,9 @@ func (ldb *Null) Size() int64 {
 	return 0
 }
 
+// Close ...
+func (ldb *Null) Close() {}
+
 // GC - runs the garbage collector
 func (ldb *Null) GC() error {
 	return nil