database backup

ByteStorage · Jan 23, 2024 · 15d6df8 · 15d6df8
1 parent d9c6d82
commit 15d6df8
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/db/engine/db.go b/db/engine/db.go
@@ -8,6 +8,7 @@ import (
 	"github.com/ByteStorage/FlyDB/config"
 	data2 "github.com/ByteStorage/FlyDB/db/data"
 	"github.com/ByteStorage/FlyDB/db/index"
+	"github.com/ByteStorage/FlyDB/lib/backup"
 	"github.com/ByteStorage/FlyDB/lib/const"
 	"go.uber.org/zap"
 	"io"
@@ -544,6 +545,15 @@ func (db *DB) loadIndexFromDataFiles() error {
 	return nil
 }
 
+// Backup the database to the specified directory
+func (db *DB) Backup(dir string) error {
+	db.lock.RLock()
+	defer db.lock.RUnlock()
+
+	// Create a backup directory
+	return backup.CopyDir(db.options.DirPath, dir)
+}
+
 // Clean the DB data directory after the test is complete
 func (db *DB) Clean() {
 	if db != nil {