Merge pull request #38 from yannh/dump-all-dbs

Add missing SELECT statement when dumping all dbs

acceptance-tests/acceptance.bats

@@ -21,6 +21,11 @@
   [ "$status" -eq 0 ]
 }
 
+@test "Dumping / restoring all databases" {
+  run tests/multiple-dbs.sh
+  [ "$status" -eq 0 ]
+}
+
 # https://github.com/yannh/redis-dump-go/issues/11
 # https://github.com/yannh/redis-dump-go/issues/18
 @test "Pass when importing a ZSET with 1M entries" {
@@ -31,4 +36,4 @@
 @test "Pass when importing 1M key/values" {
   run tests/1mkeys.sh
   [ "$status" -eq 0 ]
-}
+}

acceptance-tests/tests/multiple-dbs.sh

@@ -0,0 +1,27 @@
+#!/bin/sh -e
+
+NDBS=3
+
+echo "-> Filling Redis with Mock Data..."
+redis-cli -h redis FLUSHALL
+for DB in `seq 1 $NDBS`; do
+  redis-cli -h redis -n $DB SET thisdb $DB
+done
+
+echo "-> Dumping DB..."
+time /redis-dump-go -host redis -n 250 -output commands >backup
+
+echo "-> Flushing DB and restoring dump..."
+redis-cli -h redis FLUSHALL
+redis-cli -h redis -n $DB --pipe <backup
+
+NEWNDBS=`redis-cli -h redis info keyspace | grep keys | wc -l`
+
+echo "-> Expecting right amount of DBS..."
+if [ $NDBS -ne $NEWNDBS ]; then
+  echo "ERROR - only $NEWNDBS found, expected $NDBS"
+  exit 1
+else
+  echo "OK - $NEWNDBS dbs"
+  exit 0
+fi

pkg/redisdump/redisdump.go

@@ -355,6 +355,8 @@ func dumpDB(client radix.Client, db *uint8, filter string, nWorkers int, withTTL
 		keyGenerator = scanKeysLegacy
 	}
 
+	logger.Print(serializer([]string{"SELECT", fmt.Sprint(*db)}))
+
 	errors := make(chan error)
 	nErrors := 0
 	go func() {