diff --git a/kernel/treenode/blocktree.go b/kernel/treenode/blocktree.go index 4b56896f3..bae71b186 100644 --- a/kernel/treenode/blocktree.go +++ b/kernel/treenode/blocktree.go @@ -59,6 +59,9 @@ func GetBlockTreeByPath(path string) *BlockTree { } func CountTrees() (ret int) { + blockTreesLock.Lock() + defer blockTreesLock.Unlock() + roots := map[string]bool{} for _, b := range blockTrees { roots[b.RootID] = true @@ -68,6 +71,8 @@ func CountTrees() (ret int) { } func CountBlocks() (ret int) { + blockTreesLock.Lock() + defer blockTreesLock.Unlock() return len(blockTrees) }