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

🐛 清理资源文件后再次上传相同文件异常 Fix https://github.com/siyuan-note/siyuan/issues/6260

Browse source
This commit is contained in:
Liang Ding 2022-10-18 22:36:20 +08:00
parent 4e7c898273
commit 4e2d8b7798
No known key found for this signature in database
GPG key ID: 136F30F901A2231D
2 changed files with 25 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
kernel/model/assets.go
View file

@ -405,15 +405,21 @@ func RemoveUnusedAssets() (ret []string) {
return
}
var hashes []string
for _, p := range unusedAssets {
historyPath := filepath.Join(historyDir, p)
if p = filepath.Join(util.DataDir, p); gulu.File.IsExist(p) {
if err = gulu.File.Copy(p, historyPath); nil != err {
return
}
hash, _ := util.GetEtag(p)
hashes = append(hashes, hash)
}
}
sql.DeleteAssetsByHashes(hashes)
for _, unusedAsset := range unusedAssets {
if unusedAsset = filepath.Join(util.DataDir, unusedAsset); gulu.File.IsExist(unusedAsset) {
if err := os.RemoveAll(unusedAsset); nil != err {
@ -444,8 +450,13 @@ func RemoveUnusedAsset(p string) (ret string) {
newP := strings.TrimPrefix(p, util.DataDir)
historyPath := filepath.Join(historyDir, newP)
if err = gulu.File.Copy(p, historyPath); nil != err {
return
if gulu.File.IsExist(p) {
if err = gulu.File.Copy(p, historyPath); nil != err {
return
}
hash, _ := util.GetEtag(p)
sql.DeleteAssetsByHashes([]string{hash})
}
if err = os.RemoveAll(p); nil != err {

15
kernel/sql/aseet.go
View file

@ -17,9 +17,7 @@
package sql
import (
"crypto/sha256"
"database/sql"
"fmt"
"path/filepath"
"strings"
@ -75,7 +73,7 @@ func docTitleImgAsset(root *ast.Node) *Asset {
absPath := filepath.Join(util.DataDir, p)
if hash, err = util.GetEtag(absPath); nil != err {
logging.LogErrorf("read asset [%s] data failed: %s", absPath, err)
hash = fmt.Sprintf("%x", sha256.Sum256([]byte(gulu.Rand.String(7))))
return nil
}
name, _ := util.LastID(p)
asset := &Asset{
@ -94,6 +92,17 @@ func docTitleImgAsset(root *ast.Node) *Asset {
return nil
}
func DeleteAssetsByHashes(hashes []string) {
sqlStmt := "DELETE FROM assets WHERE hash IN ('" + strings.Join(hashes, "','") + "') OR hash = ''"
tx, err := BeginTx()
if nil != err {
return
}
execStmtTx(tx, sqlStmt)
CommitTx(tx)
}
func QueryAssetByHash(hash string) (ret *Asset) {
sqlStmt := "SELECT * FROM assets WHERE hash = ?"
row := queryRow(sqlStmt, hash)