fumiama/imago
1
0
Fork 0
mirror of https://github.com/fumiama/imago.git synced 2026-06-11 20:20:29 +08:00
Code Activity

修正uid与save add

Browse Source
This commit is contained in:
fumiama
2021-08-10 18:52:37 +08:00
parent 1856c27944
commit e5efa191af
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -14,8 +14,8 @@ Get hamming distance between two dhash strings
Scan all images like 编码后哈希.webp Scan all images like 编码后哈希.webp
### func Pick(exclude []string) string ### func Pick(exclude []string) string
Pick a random image Pick a random image
### func Saveimgbytes(b []byte, imgdir string, uid string, force bool, samediff int) (string, string) ### func Saveimgbytes(b []byte, imgdir string, force bool, samediff int) (string, string)
### func Saveimg(r io.Reader, imgdir string, uid string, samediff int) (string, string) ### func Saveimg(r io.Reader, imgdir string, samediff int) (string, string)
Save image into imgdir with name like 编码后哈希.webp Save image into imgdir with name like 编码后哈希.webp
Return value: status, dhash Return value: status, dhash
### func Addimage(name string) ### func Addimage(name string)

7
storage.go
View File

@@ -70,7 +70,7 @@ func Addimage(name string) {
} }
// Saveimgbytes Save image into imgdir with name like 编码后哈希.webp Return value: status, dhash // Saveimgbytes Save image into imgdir with name like 编码后哈希.webp Return value: status, dhash
func Saveimgbytes(b []byte, imgdir string, uid string, force bool, samediff int) (string, string) { func Saveimgbytes(b []byte, imgdir string, force bool, samediff int) (string, string) {
r := bytes.NewReader(b) r := bytes.NewReader(b)
img, _, err := image.Decode(r) img, _, err := image.Decode(r)
iswebp := false iswebp := false
@@ -125,14 +125,15 @@ func Saveimgbytes(b []byte, imgdir string, uid string, force bool, samediff int)
log.Debugf("[saveimg] save %d bytes.\n", c) log.Debugf("[saveimg] save %d bytes.\n", c)
} }
log.Debugf("[saveimg] new %s.\n", dh) log.Debugf("[saveimg] new %s.\n", dh)
Addimage(dh)
return "\"stat\":\"success\", \"img\": \"" + url.QueryEscape(dh) + "\"", dh return "\"stat\":\"success\", \"img\": \"" + url.QueryEscape(dh) + "\"", dh
} }
// Saveimg Save image into imgdir with name like 编码后哈希.webp Return value: status, dhash // Saveimg Save image into imgdir with name like 编码后哈希.webp Return value: status, dhash
func Saveimg(r io.Reader, imgdir string, uid string, samediff int) (string, string) { func Saveimg(r io.Reader, imgdir string, samediff int) (string, string) {
imgbuff := make([]byte, 1024*1024) // 1m imgbuff := make([]byte, 1024*1024) // 1m
r.Read(imgbuff) r.Read(imgbuff)
return Saveimgbytes(imgbuff, imgdir, uid, false, samediff) return Saveimgbytes(imgbuff, imgdir, false, samediff)
} }
// Scanimgs Scan all images like 编码后哈希.webp // Scanimgs Scan all images like 编码后哈希.webp