Fix potential deadlock in levelsController compaction

dgraph-io · Jul 20, 2022 · 596a5ca · 596a5ca
1 parent 7d159dd
commit 596a5ca
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/levels.go b/levels.go
@@ -1095,11 +1095,15 @@ func (s *levelsController) addSplits(cd *compactDef) {
 
 func (cd *compactDef) lockLevels() {
 	cd.thisLevel.RLock()
-	cd.nextLevel.RLock()
+	if cd.thisLevel.level != cd.nextLevel.level {
+		cd.nextLevel.RLock()
+	}
 }
 
 func (cd *compactDef) unlockLevels() {
-	cd.nextLevel.RUnlock()
+	if cd.thisLevel.level != cd.nextLevel.level {
+		cd.nextLevel.RUnlock()
+	}
 	cd.thisLevel.RUnlock()
 }