Andreas/siyuan
1
0
Fork 0
mirror of https://github.com/siyuan-note/siyuan.git synced 2025-12-16 22:50:13 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

🎨 嵌入块查询结果中显示块引用计数 https://github.com/siyuan-note/siyuan/issues/7191

Browse source
This commit is contained in:
Liang Ding 2023-01-31 19:40:55 +08:00
parent ef436cf12e
commit 35baf86b6e
No known key found for this signature in database
GPG key ID: 136F30F901A2231D
2 changed files with 35 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

22
kernel/sql/block_ref_query.go
View file

@ -79,6 +79,28 @@ func queryRefTexts() (ret []string) {
return
}
func QueryRefCount(defIDs []string) (ret map[string]int) {
ret = map[string]int{}
ids := strings.Join(defIDs, "','")
ids = "('" + ids + "')"
rows, err := query("SELECT def_block_id, COUNT(*) AS ref_cnt FROM refs WHERE def_block_id IN " + ids + " GROUP BY def_block_id")
if nil != err {
logging.LogErrorf("sql query failed: %s", err)
return
}
defer rows.Close()
for rows.Next() {
var id string
var cnt int
if err = rows.Scan(&id, &cnt); nil != err {
logging.LogErrorf("query scan field failed: %s", err)
return
}
ret[id] = cnt
}
return
}
func QueryRootChildrenRefCount(defRootID string) (ret map[string]int) {
ret = map[string]int{}
rows, err := query("SELECT def_block_id, COUNT(*) AS ref_cnt FROM refs WHERE def_block_root_id = ? GROUP BY def_block_id", defRootID)