github/siyuan
4
0
Fork 0
mirror of https://github.com/siyuan-note/siyuan.git synced 2025-05-03 00:31:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

🎨 改进清理未引用资源 https://github.com/siyuan-note/siyuan/issues/7216

Browse Source
This commit is contained in:
Liang Ding 2023-01-31 20:24:44 +08:00
parent bc7b29bd0e
commit bd153b89c5
No known key found for this signature in database
GPG Key ID: 136F30F901A2231D
3 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
kernel/model/assets.go
View File

@ -419,7 +419,7 @@ func RemoveUnusedAssets() (ret []string) {
}
}
sql.DeleteAssetsByHashes(hashes)
sql.BatchRemoveAssetsQueue(hashes)
for _, unusedAsset := range unusedAssets {
if unusedAsset = filepath.Join(util.DataDir, unusedAsset); gulu.File.IsExist(unusedAsset) {
@ -458,7 +458,7 @@ func RemoveUnusedAsset(p string) (ret string) {
}
hash, _ := util.GetEtag(absPath)
sql.DeleteAssetsByHashes([]string{hash})
sql.BatchRemoveAssetsQueue([]string{hash})
}
if err = os.RemoveAll(absPath); nil != err {

10
kernel/sql/asset.go
View File

@ -92,14 +92,10 @@ func docTitleImgAsset(root *ast.Node) *Asset {
return nil
}
func DeleteAssetsByHashes(hashes []string) {
func deleteAssetsByHashes(tx *sql.Tx, hashes []string) (err error) {
sqlStmt := "DELETE FROM assets WHERE hash IN ('" + strings.Join(hashes, "','") + "') OR hash = ''"
tx, err := beginTx()
if nil != err {
return
}
execStmtTx(tx, sqlStmt)
commitTx(tx)
err = execStmtTx(tx, sqlStmt)
return
}
func QueryAssetByHash(hash string) (ret *Asset) {

15
kernel/sql/queue.go
View File

@ -41,7 +41,7 @@ var (
type dbQueueOperation struct {
inQueueTime time.Time
action string // upsert/delete/delete_id/rename/delete_box/delete_box_refs/insert_refs/index/delete_ids/update_block_content
action string // upsert/delete/delete_id/rename/delete_box/delete_box_refs/insert_refs/index/delete_ids/update_block_content/delete_assets
indexPath string // index
upsertTree *parse.Tree // upsert/insert_refs
@ -52,6 +52,7 @@ type dbQueueOperation struct {
renameTree *parse.Tree // rename
renameTreeOldHPath string // rename
block *Block // update_block_content
removeAssetHashes []string // delete_assets
}
func FlushTxJob() {
@ -169,6 +170,8 @@ func execOp(op *dbQueueOperation, tx *sql.Tx, context map[string]interface{}) (e
err = upsertRefs(tx, op.upsertTree)
case "update_block_content":
err = updateBlockContent(tx, op.block)
case "delete_assets":
err = deleteAssetsByHashes(tx, op.removeAssetHashes)
default:
msg := fmt.Sprintf("unknown operation [%s]", op.action)
logging.LogErrorf(msg)
@ -177,6 +180,14 @@ func execOp(op *dbQueueOperation, tx *sql.Tx, context map[string]interface{}) (e
return
}
func BatchRemoveAssetsQueue(hashes []string) {
dbQueueLock.Lock()
defer dbQueueLock.Unlock()
newOp := &dbQueueOperation{removeAssetHashes: hashes, inQueueTime: time.Now(), action: "delete_assets"}
operationQueue = append(operationQueue, newOp)
}
func UpdateBlockContentQueue(block *Block) {
dbQueueLock.Lock()
defer dbQueueLock.Unlock()
@ -336,4 +347,4 @@ func mergeUpsertTrees() (ops []*dbQueueOperation) {
ops = operationQueue
operationQueue = nil
return
}
}