diff --git a/go/images/orientation.go b/go/images/orientation.go index 7c70f9ed7..a7c126fc1 100644 --- a/go/images/orientation.go +++ b/go/images/orientation.go @@ -5,17 +5,12 @@ import ( "github.com/rwcarlsen/goexif/exif" "image" "image/draw" - "image/gif" "image/jpeg" - "image/png" "log" ) //many code is copied from http://camlistore.org/pkg/images/images.go -func FixJpgOrientation(ext string, data []byte) (oriented []byte) { - if ext != ".jpg" { - return data - } +func FixJpgOrientation(data []byte) (oriented []byte) { ex, err := exif.Decode(bytes.NewReader(data)) if err != nil { return data @@ -53,14 +48,7 @@ func FixJpgOrientation(ext string, data []byte) (oriented []byte) { if srcImage, _, err := image.Decode(bytes.NewReader(data)); err == nil { dstImage := flip(rotate(srcImage, angle), flipMode) 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) - } + jpeg.Encode(&buf, dstImage, nil) return buf.Bytes() } diff --git a/go/weed/server.go b/go/weed/server.go index 8d8504285..eea9f9037 100644 --- a/go/weed/server.go +++ b/go/weed/server.go @@ -63,6 +63,7 @@ var ( volumeDataFolders = cmdServer.Flag.String("dir", os.TempDir(), "directories to store data files. dir[,dir]...") volumeMaxDataVolumeCounts = cmdServer.Flag.String("volume.max", "7", "maximum numbers of volumes, count[,count]...") volumePulse = cmdServer.Flag.Int("pulseSeconds", 5, "number of seconds between heartbeats") + volumeFixJpgOrientation = cmdServer.Flag.Bool("volume.fix.jpg.orientation", false, "Adjust jpg orientation when uploading.") isStartingFiler = cmdServer.Flag.Bool("filer", false, "whether to start filer") serverWhiteList []string @@ -206,6 +207,7 @@ func runServer(cmd *Command, args []string) bool { r := http.NewServeMux() volumeServer := weed_server.NewVolumeServer(r, *serverIp, *volumePort, *serverPublicIp, folders, maxCounts, *serverIp+":"+strconv.Itoa(*masterPort), *volumePulse, *serverDataCenter, *serverRack, serverWhiteList, + *volumeFixJpgOrientation, ) glog.V(0).Infoln("Start Weed volume server", util.VERSION, "at", *serverIp+":"+strconv.Itoa(*volumePort)) diff --git a/go/weed/volume.go b/go/weed/volume.go index c4177a98a..142002ec1 100644 --- a/go/weed/volume.go +++ b/go/weed/volume.go @@ -37,6 +37,7 @@ var ( dataCenter = cmdVolume.Flag.String("dataCenter", "", "current volume server's data center name") rack = cmdVolume.Flag.String("rack", "", "current volume server's rack name") volumeWhiteListOption = cmdVolume.Flag.String("whiteList", "", "comma separated Ip addresses having write permission. No limit if empty.") + fixJpgOrientation = cmdVolume.Flag.Bool("fix.jpg.orientation", false, "Adjust jpg orientation when uploading.") volumeWhiteList []string ) @@ -80,6 +81,7 @@ func runVolume(cmd *Command, args []string) bool { volumeServer := weed_server.NewVolumeServer(r, *ip, *vport, *publicIp, folders, maxCounts, *masterNode, *vpulse, *dataCenter, *rack, volumeWhiteList, + *fixJpgOrientation, ) listeningAddress := *ip + ":" + strconv.Itoa(*vport) diff --git a/go/weed/weed_server/volume_server.go b/go/weed/weed_server/volume_server.go index 4f64eb2c5..b118e8c35 100644 --- a/go/weed/weed_server/volume_server.go +++ b/go/weed/weed_server/volume_server.go @@ -16,19 +16,22 @@ type VolumeServer struct { rack string whiteList []string store *storage.Store + + FixJpgOrientation bool } func NewVolumeServer(r *http.ServeMux, ip string, port int, publicIp string, folders []string, maxCounts []int, masterNode string, pulseSeconds int, dataCenter string, rack string, - whiteList []string) *VolumeServer { + whiteList []string, fixJpgOrientation bool) *VolumeServer { publicUrl := publicIp + ":" + strconv.Itoa(port) vs := &VolumeServer{ - masterNode: masterNode, - pulseSeconds: pulseSeconds, - dataCenter: dataCenter, - rack: rack, - whiteList: whiteList, + masterNode: masterNode, + pulseSeconds: pulseSeconds, + dataCenter: dataCenter, + rack: rack, + whiteList: whiteList, + FixJpgOrientation: fixJpgOrientation, } vs.store = storage.NewStore(port, ip, publicUrl, folders, maxCounts) diff --git a/go/weed/weed_server/volume_server_handlers.go b/go/weed/weed_server/volume_server_handlers.go index 5b99233a2..65a3d14f8 100644 --- a/go/weed/weed_server/volume_server_handlers.go +++ b/go/weed/weed_server/volume_server_handlers.go @@ -156,6 +156,10 @@ func (vs *VolumeServer) PostHandler(w http.ResponseWriter, r *http.Request) { return } + if vs.FixJpgOrientation && strings.HasSuffix(string(needle.Name), ".jpg") { + needle.Data = images.FixJpgOrientation(needle.Data) + } + ret := operation.UploadResult{} size, errorStatus := topology.ReplicatedWrite(vs.masterNode, vs.store, volumeId, needle, r) if errorStatus == "" {