wood/webp_server_go
1
0
Fork 0
mirror of https://github.com/woodchen-ink/webp_server_go.git synced 2025-07-18 13:42:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

better file type

Browse Source
This commit is contained in:
Benny~ 2020-02-25 22:20:26 +08:00
parent 8bdffcc246
commit f790b24a0b
No known key found for this signature in database
GPG Key ID: 6CD0DBDA5235D481
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
webp-server.go
View File

@ -10,6 +10,7 @@ import (
"image/png"
"io/ioutil"
"log"
"net/http"
"os"
"path"
"strconv"
@ -29,13 +30,22 @@ type Config struct {
var configPath string
func GetFileContentType(buffer []byte) (string, error) {
// Use the net/http package's handy DectectContentType function. Always returns a valid
// content-type by returning "application/octet-stream" if no others seemed to match.
contentType := http.DetectContentType(buffer)
return contentType, nil
}
func webpEncoder(p1, p2 string, quality float32) {
var buf bytes.Buffer
var img image.Image
data, _ := ioutil.ReadFile(p1)
if strings.Contains(p1, "jpg") || strings.Contains(p1, "jpeg") {
contentType, _ := GetFileContentType(data[:512])
if strings.Contains(contentType, "jpeg") {
img, _ = jpeg.Decode(bytes.NewReader(data))
} else if strings.Contains(p1, "png") {
} else if strings.Contains(contentType, "png") {
img, _ = png.Decode(bytes.NewReader(data))
}