Fix compacting disable/enable

Enabling and disabling compaction no longer blocks are potentially
causes panics.

db.go

@@ -100,7 +100,6 @@ type DB struct {
 	opts    *Options
 
 	// Mutex for that must be held when modifying the general block layout.
-	// cmtx must be held before acquiring it.
 	mtx    sync.RWMutex
 	blocks []Block
 
@@ -117,8 +116,8 @@ type DB struct {
 	stopc    chan struct{}
 
 	// cmtx is used to control compactions and deletions.
-	cmtx       sync.Mutex
+	cmtx               sync.Mutex
-	compacting bool
+	compactionsEnabled bool
 }
 
 type dbMetrics struct {
@@ -197,13 +196,13 @@ func Open(dir string, l log.Logger, r prometheus.Registerer, opts *Options) (db
 	}
 
 	db = &DB{
-		dir:        dir,
+		dir:                dir,
-		logger:     l,
+		logger:             l,
-		opts:       opts,
+		opts:               opts,
-		compactc:   make(chan struct{}, 1),
+		compactc:           make(chan struct{}, 1),
-		donec:      make(chan struct{}),
+		donec:              make(chan struct{}),
-		stopc:      make(chan struct{}),
+		stopc:              make(chan struct{}),
-		compacting: true,
+		compactionsEnabled: true,
 	}
 	db.metrics = newDBMetrics(db, r)
 
@@ -371,6 +370,10 @@ func (db *DB) compact() (changes bool, err error) {
 	db.cmtx.Lock()
 	defer db.cmtx.Unlock()
 
+	if !db.compactionsEnabled {
+		return false, nil
+	}
+
 	// Check whether we have pending head blocks that are ready to be persisted.
 	// They have the highest priority.
 	for _, h := range db.completedHeads() {
@@ -579,20 +582,20 @@ func (db *DB) Close() error {
 
 // DisableCompactions disables compactions.
 func (db *DB) DisableCompactions() {
-	if db.compacting {
+	db.cmtx.Lock()
-		db.cmtx.Lock()
+	defer db.cmtx.Unlock()
-		db.compacting = false
+
-		db.logger.Log("msg", "compactions disabled")
+	db.compactionsEnabled = false
-	}
+	db.logger.Log("msg", "compactions disabled")
 }
 
 // EnableCompactions enables compactions.
 func (db *DB) EnableCompactions() {
-	if !db.compacting {
+	db.cmtx.Lock()
-		db.cmtx.Unlock()
+	defer db.cmtx.Unlock()
-		db.compacting = true
+
-		db.logger.Log("msg", "compactions enabled")
+	db.compactionsEnabled = true
-	}
+	db.logger.Log("msg", "compactions enabled")
 }
 
 // Snapshot writes the current data to the directory.