From 471089b6e9914a1c1e0b591c69e9737af91f4177 Mon Sep 17 00:00:00 2001
From: Liang Ding <d@b3log.org>
Date: Sat, 24 Dec 2022 10:51:02 +0800
Subject: [PATCH] =?UTF-8?q?:art:=20=E5=9D=97=E6=A0=91=E5=B9=B6=E5=8F=91?=
 =?UTF-8?q?=E8=AE=BF=E9=97=AE=E5=BC=82=E5=B8=B8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 kernel/treenode/blocktree.go | 5 +++++
 1 file changed, 5 insertions(+)

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)
 }