heimoshuiyu/msw-open-music
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

fix postgres insert on conflict not returning id

Browse Source
This commit is contained in:
黑墨水鱼
2022-11-04 21:27:46 +08:00
parent 73da4f8dc5
commit e4032069a5
1 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
pkg/database/method.go
View File

@@ -168,6 +168,7 @@ func (database *Database) Walk(root string, pattern []string, tagIDs []int64, us
insertFileStmt := tx.Stmt(database.stmt.insertFile)
putTagOnFileStmt := tx.Stmt(database.stmt.putTagOnFile)
findFolderStmt := tx.Stmt(database.stmt.findFolder)
findFileStmt := tx.Stmt(database.stmt.findFile)
err = filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
if err != nil {
@@ -205,11 +206,25 @@ func (database *Database) Walk(root string, pattern []string, tagIDs []int64, us
}
}
}
result, err := insertFileStmt.Query(folderID, filename, filename, info.Size())
// try find file id
var fileID int64
result, err := findFileStmt.Query(folderID, filename)
if err != nil {
return err
}
for result.Next() {
err = result.Scan(&fileID)
if err != nil {
return err
}
}
// insert new file
result, err = insertFileStmt.Query(folderID, filename, filename, info.Size())
if err != nil {
return err
}
var fileID int64
for result.Next() {
err = result.Scan(&fileID)
if err != nil {