From 200df850ccc2c747f75acf9d06b00230d2f72c98 Mon Sep 17 00:00:00 2001
From: Lucas Molas <schomatis@gmail.com>
Date: Mon, 22 Oct 2018 09:17:18 -0300
Subject: [PATCH] badger: add truncate flag

License: MIT
Signed-off-by: Lucas Molas <schomatis@gmail.com>
---
 docs/datastores.md                  |  6 +++++-
 plugin/plugins/badgerds/badgerds.go | 13 +++++++++++++
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/docs/datastores.md b/docs/datastores.md
index 669ffa615..bec26c649 100644
--- a/docs/datastores.md
+++ b/docs/datastores.md
@@ -38,11 +38,15 @@ Uses a leveldb database to store key value pairs.
 ## badgerds
 Uses [badger](https://github.com/dgraph-io/badger) as a key value store.
 
+* `syncWrites`: Synchronize every write to disk.
+* `truncate`: Truncate the DB if a corrupted sector is found (otherwise Badger won't start). This option is always set to `true` in Windows if `syncWrites` is set.
+
 ```json
 {
 	"type": "badgerds",
 	"path": "<location of badger inside repo",
-	"syncWrites": true|false
+	"syncWrites": true|false,
+	"truncate": true|false,
 }
 ```
 
diff --git a/plugin/plugins/badgerds/badgerds.go b/plugin/plugins/badgerds/badgerds.go
index bbc5509c9..1b8c83af8 100644
--- a/plugin/plugins/badgerds/badgerds.go
+++ b/plugin/plugins/badgerds/badgerds.go
@@ -41,6 +41,7 @@ func (*badgerdsPlugin) DatastoreTypeName() string {
 type datastoreConfig struct {
 	path       string
 	syncWrites bool
+	truncate   bool
 
 	vlogFileSize int64
 }
@@ -68,6 +69,17 @@ func (*badgerdsPlugin) DatastoreConfigParser() fsrepo.ConfigFromMap {
 			}
 		}
 
+		truncate, ok := params["truncate"]
+		if !ok {
+			c.truncate = true
+		} else {
+			if truncate, ok := truncate.(bool); ok {
+				c.truncate = truncate
+			} else {
+				return nil, fmt.Errorf("'truncate' field was not a boolean")
+			}
+		}
+
 		vls, ok := params["vlogFileSize"]
 		if !ok {
 			// default to 1GiB
@@ -108,6 +120,7 @@ func (c *datastoreConfig) Create(path string) (repo.Datastore, error) {
 
 	defopts := badgerds.DefaultOptions
 	defopts.SyncWrites = c.syncWrites
+	defopts.Truncate = c.truncate
 	defopts.ValueLogFileSize = c.vlogFileSize
 
 	return badgerds.NewDatastore(p, &defopts)