mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
1. refactor, move image resizing to its own package
2. make code compile
This commit is contained in:
parent
5367d96512
commit
f7d6909b6f
|
@ -11,9 +11,9 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
type ApiRequest struct {
|
type ApiRequest struct {
|
||||||
Command string //"listFiles", "listDirectories"
|
Command string //"listFiles", "listDirectories"
|
||||||
DirectoryId DirectoryId
|
Directory string
|
||||||
FileName string
|
FileName string
|
||||||
}
|
}
|
||||||
|
|
||||||
type ListFilesResult struct {
|
type ListFilesResult struct {
|
||||||
|
@ -21,9 +21,9 @@ type ListFilesResult struct {
|
||||||
Error string `json:"error,omitempty"`
|
Error string `json:"error,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func ListFiles(server string, directoryId DirectoryId, fileName string) (*ListFilesResult, error) {
|
func ListFiles(server string, directory string, fileName string) (*ListFilesResult, error) {
|
||||||
var ret ListFilesResult
|
var ret ListFilesResult
|
||||||
if err := call(server, ApiRequest{Command: "listFiles", DirectoryId: directoryId, FileName: fileName}, &ret); err == nil {
|
if err := call(server, ApiRequest{Command: "listFiles", Directory: directory, FileName: fileName}, &ret); err == nil {
|
||||||
if ret.Error != "" {
|
if ret.Error != "" {
|
||||||
return nil, errors.New(ret.Error)
|
return nil, errors.New(ret.Error)
|
||||||
}
|
}
|
||||||
|
@ -38,9 +38,9 @@ type ListDirectoriesResult struct {
|
||||||
Error string `json:"error,omitempty"`
|
Error string `json:"error,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func ListDirectories(server string, directoryId DirectoryId) (*ListDirectoriesResult, error) {
|
func ListDirectories(server string, directory string) (*ListDirectoriesResult, error) {
|
||||||
var ret ListDirectoriesResult
|
var ret ListDirectoriesResult
|
||||||
if err := call(server, ApiRequest{Command: "listDirectories", DirectoryId: directoryId}, &ret); err == nil {
|
if err := call(server, ApiRequest{Command: "listDirectories", Directory: directory}, &ret); err == nil {
|
||||||
if ret.Error != "" {
|
if ret.Error != "" {
|
||||||
return nil, errors.New(ret.Error)
|
return nil, errors.New(ret.Error)
|
||||||
}
|
}
|
||||||
|
|
39
go/images/images.go
Normal file
39
go/images/images.go
Normal file
|
@ -0,0 +1,39 @@
|
||||||
|
package images
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"github.com/disintegration/imaging"
|
||||||
|
"image"
|
||||||
|
"image/gif"
|
||||||
|
"image/jpeg"
|
||||||
|
"image/png"
|
||||||
|
)
|
||||||
|
|
||||||
|
func FixJpgOrientation() {
|
||||||
|
}
|
||||||
|
|
||||||
|
func Resized(ext string, data []byte, width, height int) (resized []byte) {
|
||||||
|
if width == 0 && height == 0 {
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
if srcImage, _, err := image.Decode(bytes.NewReader(data)); err == nil {
|
||||||
|
bounds := srcImage.Bounds()
|
||||||
|
var dstImage *image.NRGBA
|
||||||
|
if width == height && bounds.Dx() != bounds.Dy() {
|
||||||
|
dstImage = imaging.Thumbnail(srcImage, width, height, imaging.Lanczos)
|
||||||
|
} else {
|
||||||
|
dstImage = imaging.Resize(srcImage, width, height, imaging.Lanczos)
|
||||||
|
}
|
||||||
|
var buf bytes.Buffer
|
||||||
|
switch ext {
|
||||||
|
case ".png":
|
||||||
|
png.Encode(&buf, dstImage)
|
||||||
|
case ".jpg":
|
||||||
|
jpeg.Encode(&buf, dstImage, nil)
|
||||||
|
case ".gif":
|
||||||
|
gif.Encode(&buf, dstImage, nil)
|
||||||
|
}
|
||||||
|
return buf.Bytes()
|
||||||
|
}
|
||||||
|
return data
|
||||||
|
}
|
|
@ -59,16 +59,16 @@ func (File) ReadAll(intr fs.Intr) ([]byte, fuse.Error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
type Dir struct {
|
type Dir struct {
|
||||||
DirectoryId filer.DirectoryId
|
Path string
|
||||||
Name string
|
Id uint64
|
||||||
}
|
}
|
||||||
|
|
||||||
func (dir Dir) Attr() fuse.Attr {
|
func (dir Dir) Attr() fuse.Attr {
|
||||||
return fuse.Attr{Inode: 1, Mode: os.ModeDir | 0555}
|
return fuse.Attr{Inode: dir.Id, Mode: os.ModeDir | 0555}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (dir Dir) Lookup(name string, intr fs.Intr) (fs.Node, fuse.Error) {
|
func (dir Dir) Lookup(name string, intr fs.Intr) (fs.Node, fuse.Error) {
|
||||||
files_result, e := filer.ListFiles(*mountOptions.filer, dir.DirectoryId, name)
|
files_result, e := filer.ListFiles(*mountOptions.filer, dir.Path, name)
|
||||||
if e != nil {
|
if e != nil {
|
||||||
return nil, fuse.ENOENT
|
return nil, fuse.ENOENT
|
||||||
}
|
}
|
||||||
|
@ -86,13 +86,13 @@ func (WFS) Root() (fs.Node, fuse.Error) {
|
||||||
|
|
||||||
func (dir *Dir) ReadDir(intr fs.Intr) ([]fuse.Dirent, fuse.Error) {
|
func (dir *Dir) ReadDir(intr fs.Intr) ([]fuse.Dirent, fuse.Error) {
|
||||||
ret := make([]fuse.Dirent, 0)
|
ret := make([]fuse.Dirent, 0)
|
||||||
if dirs, e := filer.ListDirectories(*mountOptions.filer, dir.DirectoryId); e == nil {
|
if dirs, e := filer.ListDirectories(*mountOptions.filer, dir.Path); e == nil {
|
||||||
for _, d := range dirs.Directories {
|
for _, d := range dirs.Directories {
|
||||||
dirId := uint64(d.Id)
|
dirId := uint64(d.Id)
|
||||||
ret = append(ret, fuse.Dirent{Inode: dirId, Name: d.Name, Type: fuse.DT_Dir})
|
ret = append(ret, fuse.Dirent{Inode: dirId, Name: d.Name, Type: fuse.DT_Dir})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if files, e := filer.ListFiles(*mountOptions.filer, dir.DirectoryId, ""); e == nil {
|
if files, e := filer.ListFiles(*mountOptions.filer, dir.Path, ""); e == nil {
|
||||||
for _, f := range files.Files {
|
for _, f := range files.Files {
|
||||||
if fileId, e := storage.ParseFileId(string(f.Id)); e == nil {
|
if fileId, e := storage.ParseFileId(string(f.Id)); e == nil {
|
||||||
fileInode := uint64(fileId.VolumeId)<<48 + fileId.Key
|
fileInode := uint64(fileId.VolumeId)<<48 + fileId.Key
|
||||||
|
|
|
@ -26,7 +26,6 @@ func NewFilerServer(r *http.ServeMux, port int, master string, dir string, colle
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
r.HandleFunc("/__api__", fs.filerApiHandler)
|
|
||||||
r.HandleFunc("/", fs.filerHandler)
|
r.HandleFunc("/", fs.filerHandler)
|
||||||
|
|
||||||
return fs, nil
|
return fs, nil
|
||||||
|
|
|
@ -1,20 +0,0 @@
|
||||||
package weed_server
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net/http"
|
|
||||||
)
|
|
||||||
|
|
||||||
func (fs *FilerServer) filerApiHandler(w http.ResponseWriter, r *http.Request) {
|
|
||||||
switch r.Method {
|
|
||||||
case "GET":
|
|
||||||
fs.GetOrHeadHandler(w, r, true)
|
|
||||||
case "HEAD":
|
|
||||||
fs.GetOrHeadHandler(w, r, false)
|
|
||||||
case "DELETE":
|
|
||||||
fs.DeleteHandler(w, r)
|
|
||||||
case "PUT":
|
|
||||||
fs.PostHandler(w, r)
|
|
||||||
case "POST":
|
|
||||||
fs.PostHandler(w, r)
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,17 +1,12 @@
|
||||||
package weed_server
|
package weed_server
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"code.google.com/p/weed-fs/go/glog"
|
"code.google.com/p/weed-fs/go/glog"
|
||||||
|
"code.google.com/p/weed-fs/go/images"
|
||||||
"code.google.com/p/weed-fs/go/operation"
|
"code.google.com/p/weed-fs/go/operation"
|
||||||
"code.google.com/p/weed-fs/go/stats"
|
"code.google.com/p/weed-fs/go/stats"
|
||||||
"code.google.com/p/weed-fs/go/storage"
|
"code.google.com/p/weed-fs/go/storage"
|
||||||
"code.google.com/p/weed-fs/go/topology"
|
"code.google.com/p/weed-fs/go/topology"
|
||||||
"github.com/disintegration/imaging"
|
|
||||||
"image"
|
|
||||||
"image/gif"
|
|
||||||
"image/jpeg"
|
|
||||||
"image/png"
|
|
||||||
"mime"
|
"mime"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
@ -132,27 +127,7 @@ func (vs *VolumeServer) GetOrHeadHandler(w http.ResponseWriter, r *http.Request,
|
||||||
if r.FormValue("height") != "" {
|
if r.FormValue("height") != "" {
|
||||||
height, _ = strconv.Atoi(r.FormValue("height"))
|
height, _ = strconv.Atoi(r.FormValue("height"))
|
||||||
}
|
}
|
||||||
if width != 0 || height != 0 {
|
n.Data = images.Resized(ext, n.Data, width, height)
|
||||||
if srcImage, _, err := image.Decode(bytes.NewReader(n.Data)); err == nil {
|
|
||||||
bounds := srcImage.Bounds()
|
|
||||||
var dstImage *image.NRGBA
|
|
||||||
if width == height && bounds.Dx() != bounds.Dy() {
|
|
||||||
dstImage = imaging.Thumbnail(srcImage, width, height, imaging.Lanczos)
|
|
||||||
} else {
|
|
||||||
dstImage = imaging.Resize(srcImage, width, height, imaging.Lanczos)
|
|
||||||
}
|
|
||||||
var buf bytes.Buffer
|
|
||||||
switch ext {
|
|
||||||
case ".png":
|
|
||||||
png.Encode(&buf, dstImage)
|
|
||||||
case ".jpg":
|
|
||||||
jpeg.Encode(&buf, dstImage, nil)
|
|
||||||
case ".gif":
|
|
||||||
gif.Encode(&buf, dstImage, nil)
|
|
||||||
}
|
|
||||||
n.Data = buf.Bytes()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
w.Header().Set("Content-Length", strconv.Itoa(len(n.Data)))
|
w.Header().Set("Content-Length", strconv.Itoa(len(n.Data)))
|
||||||
if isGetMethod {
|
if isGetMethod {
|
||||||
|
|
Loading…
Reference in a new issue