fumiama/dupimage
1
0
Fork 0
mirror of https://github.com/fumiama/dupimage.git synced 2026-06-24 21:54:39 +08:00
Code Activity

shrink panic

Browse Source
This commit is contained in:
源文雨
2022-07-25 11:46:57 +08:00
parent d660ca02c1
commit ff7448b64a
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
main.go
View File

@@ -85,15 +85,20 @@ func main() {
if !img.IsDir() && isextinlist(n) { if !img.IsDir() && isextinlist(n) {
f, err := os.Open(n) f, err := os.Open(n)
if err != nil { if err != nil {
panic(err) fmt.Println("ERROR:", err)
continue
} }
im, _, err := image.Decode(f) im, _, err := image.Decode(f)
if err != nil { if err != nil {
panic(err) fmt.Println("ERROR:", err)
_ = f.Close()
continue
} }
dh, err := goimagehash.DifferenceHash(im) dh, err := goimagehash.DifferenceHash(im)
if err != nil { if err != nil {
panic(err) fmt.Println("ERROR:", err)
_ = f.Close()
continue
} }
mu.Lock() mu.Lock()
chklst = append(chklst, imagecheck{ chklst = append(chklst, imagecheck{
@@ -117,7 +122,8 @@ func main() {
for j := len(chklst) - 1; j > i; j-- { for j := len(chklst) - 1; j > i; j-- {
dis, err := chklst[i].dh.Distance(chklst[j].dh) dis, err := chklst[i].dh.Distance(chklst[j].dh)
if err != nil { if err != nil {
panic(err) fmt.Println("ERROR:", err)
continue
} }
if uint(dis) < throttle { if uint(dis) < throttle {
mu.Lock() mu.Lock()
@@ -147,12 +153,13 @@ func main() {
newdir := strconv.Itoa(j) newdir := strconv.Itoa(j)
err = os.MkdirAll(newdir, 0755) err = os.MkdirAll(newdir, 0755)
if err != nil { if err != nil {
panic(err) fmt.Println("ERROR:", err)
continue
} }
for _, i := range lst { for _, i := range lst {
err = os.Rename(i.name, newdir+"/"+i.name) err = os.Rename(i.name, newdir+"/"+i.name)
if err != nil { if err != nil {
panic(err) fmt.Println("ERROR:", err)
} }
} }
} }