0ct0pu5/siyuan
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

🎨 Improve workspace startup stability https://github.com/siyuan-note/siyuan/issues/11483

Browse source
This commit is contained in:
Daniel 2024-05-20 23:40:15 +08:00
parent 47f884f2bb
commit dfc8d39a68
No known key found for this signature in database
GPG key ID: 86211BA83DF03017
1 changed files with 16 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
kernel/treenode/blocktree.go
View file

@ -451,6 +451,7 @@ func InitBlockTree(force bool) {
return
}
loadErr := atomic.Bool{}
size := atomic.Int64{}
waitGroup := &sync.WaitGroup{}
p, _ := ants.NewPoolWithFunc(4, func(arg interface{}) {
@ -462,14 +463,15 @@ func InitBlockTree(force bool) {
f, err := os.OpenFile(p, os.O_RDONLY, 0644)
if nil != err {
logging.LogErrorf("open block tree failed: %s", err)
os.Exit(logging.ExitCodeFileSysErr)
loadErr.Store(true)
return
}
defer f.Close()
info, err := f.Stat()
if nil != err {
logging.LogErrorf("stat block tree failed: %s", err)
os.Exit(logging.ExitCodeFileSysErr)
loadErr.Store(true)
return
}
size.Add(info.Size())
@ -477,16 +479,7 @@ func InitBlockTree(force bool) {
sliceData := map[string]*BlockTree{}
if err = msgpack.NewDecoder(f).Decode(&sliceData); nil != err {
logging.LogErrorf("unmarshal block tree failed: %s", err)
if err = os.RemoveAll(util.BlockTreePath); nil != err {
logging.LogErrorf("removed corrupted block tree failed: %s", err)
}
os.Exit(logging.ExitCodeFileSysErr)
return
}
if err = f.Close(); nil != err {
logging.LogErrorf("close block tree failed: %s", err)
os.Exit(logging.ExitCodeFileSysErr)
loadErr.Store(true)
return
}
@ -505,6 +498,17 @@ func InitBlockTree(force bool) {
waitGroup.Wait()
p.Release()
if loadErr.Load() {
logging.LogInfof("cause block tree load error, remove block tree file")
if removeErr := os.RemoveAll(util.BlockTreePath); nil != removeErr {
logging.LogErrorf("remove block tree file failed: %s", removeErr)
os.Exit(logging.ExitCodeFileSysErr)
return
}
blockTrees = &sync.Map{}
return
}
elapsed := time.Since(start).Seconds()
logging.LogInfof("read block tree [%s] to [%s], elapsed [%.2fs]", humanize.BytesCustomCeil(uint64(size.Load()), 2), util.BlockTreePath, elapsed)
return