seaweedfs/weed-fs/src/cmd/weed/volume.go

259 lines
7.2 KiB
Go
Raw Normal View History

package main
import (
"bytes"
2012-09-12 08:07:23 +00:00
"encoding/json"
2012-08-24 05:46:54 +00:00
"log"
"math/rand"
"mime"
"net/http"
"os"
"pkg/operation"
2012-08-24 05:46:54 +00:00
"pkg/storage"
"strconv"
"strings"
"time"
)
func init() {
2012-08-24 05:46:54 +00:00
cmdVolume.Run = runVolume // break init cycle
IsDebug = cmdVolume.Flag.Bool("debug", false, "enable debug mode")
}
var cmdVolume = &Command{
UsageLine: "volume -port=8080 -dir=/tmp -min=3 -max=5 -publicUrl=server_name:8080 -mserver=localhost:9333",
2012-08-24 05:46:54 +00:00
Short: "start a volume server",
Long: `start a volume server to provide storage spaces
`,
}
var (
vport = cmdVolume.Flag.Int("port", 8080, "http listen port")
volumeFolder = cmdVolume.Flag.String("dir", "/tmp", "data directory to store files")
volumes = cmdVolume.Flag.String("volumes", "", "comma-separated list, or ranges of volume ids")
publicUrl = cmdVolume.Flag.String("publicUrl", "localhost:8080", "public url to serve data read")
masterNode = cmdVolume.Flag.String("mserver", "localhost:9333", "master directory server to store mappings")
vpulse = cmdVolume.Flag.Int("pulseSeconds", 5, "number of seconds between heartbeats")
2012-09-19 23:56:35 +00:00
maxVolumeCount = cmdVolume.Flag.Int("max", 5, "maximum number of volumes")
2012-08-24 05:46:54 +00:00
store *storage.Store
)
func statusHandler(w http.ResponseWriter, r *http.Request) {
2012-08-24 05:46:54 +00:00
writeJson(w, r, store.Status())
}
2012-09-11 00:08:52 +00:00
func assignVolumeHandler(w http.ResponseWriter, r *http.Request) {
err := store.AddVolume(r.FormValue("volume"), r.FormValue("replicationType"))
if err == nil {
writeJson(w, r, map[string]string{"error": ""})
} else {
writeJson(w, r, map[string]string{"error": err.Error()})
}
if *IsDebug {
log.Println("volume =", r.FormValue("volume"), ", replicationType =", r.FormValue("replicationType"), ", error =", err)
}
2012-09-11 00:08:52 +00:00
}
func setVolumeLocationsHandler(w http.ResponseWriter, r *http.Request) {
2012-09-12 08:07:23 +00:00
if *IsDebug {
log.Println("volumeLocationsList =", r.FormValue("volumeLocationsList"))
}
volumeLocationsList := new([]storage.VolumeLocations)
err := json.Unmarshal([]byte(r.FormValue("volumeLocationsList")), volumeLocationsList)
if err == nil {
err = store.SetVolumeLocations(*volumeLocationsList)
}
if err == nil {
writeJson(w, r, map[string]string{"error": ""})
} else {
writeJson(w, r, map[string]string{"error": err.Error()})
}
}
func storeHandler(w http.ResponseWriter, r *http.Request) {
2012-08-24 05:46:54 +00:00
switch r.Method {
case "GET":
GetHandler(w, r)
case "DELETE":
DeleteHandler(w, r)
case "POST":
PostHandler(w, r)
}
}
func GetHandler(w http.ResponseWriter, r *http.Request) {
2012-08-24 05:46:54 +00:00
n := new(storage.Needle)
vid, fid, ext := parseURLPath(r.URL.Path)
volumeId, _ := storage.NewVolumeId(vid)
n.ParsePath(fid)
if *IsDebug {
log.Println("volume", volumeId, "reading", n)
}
if !store.HasVolume(volumeId) {
lookupResult, err := operation.Lookup(*server, volumeId)
if *IsDebug {
log.Println("volume", volumeId, "found on", lookupResult, "error", err)
}
if err == nil {
http.Redirect(w, r, "http://"+lookupResult.Locations[0].PublicUrl+r.URL.Path, http.StatusMovedPermanently)
} else {
w.WriteHeader(http.StatusNotFound)
}
return
}
2012-08-24 05:46:54 +00:00
cookie := n.Cookie
count, e := store.Read(volumeId, n)
if *IsDebug {
log.Println("read bytes", count, "error", e)
}
if e != nil || count <= 0 {
w.WriteHeader(404)
return
2012-08-24 05:46:54 +00:00
}
if n.Cookie != cookie {
log.Println("request with unmaching cookie from ", r.RemoteAddr, "agent", r.UserAgent())
w.WriteHeader(404)
2012-08-24 05:46:54 +00:00
return
}
if ext != "" {
mtype := mime.TypeByExtension(ext)
w.Header().Set("Content-Type", mtype)
if storage.IsCompressable(ext, mtype) {
if strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") {
w.Header().Set("Content-Encoding", "gzip")
} else {
n.Data = storage.UnGzipData(n.Data)
}
}
}
w.Write(n.Data)
}
func PostHandler(w http.ResponseWriter, r *http.Request) {
2012-09-21 08:30:31 +00:00
r.ParseForm()
2012-08-24 05:46:54 +00:00
vid, _, _ := parseURLPath(r.URL.Path)
2012-09-04 03:40:38 +00:00
volumeId, e := storage.NewVolumeId(vid)
2012-08-24 05:46:54 +00:00
if e != nil {
writeJson(w, r, e)
} else {
needle, filename, ne := storage.NewNeedle(r)
2012-08-24 05:46:54 +00:00
if ne != nil {
writeJson(w, r, ne)
} else {
ret := store.Write(volumeId, needle)
2012-09-21 08:30:31 +00:00
if ret > 0 || !store.HasVolume(volumeId) { //send to other replica locations
2012-09-21 00:58:29 +00:00
if r.FormValue("type") != "standard" {
2012-09-21 08:30:31 +00:00
if lookupResult, lookupErr := operation.Lookup(*server, volumeId); lookupErr == nil {
2012-09-21 00:58:29 +00:00
sendFunc := func(background bool) {
postContentFunc := func(location operation.Location) bool {
operation.Upload("http://"+location.PublicUrl+r.URL.Path+"?type=standard", filename, bytes.NewReader(needle.Data))
return true
2012-09-21 00:58:29 +00:00
}
for _, location := range lookupResult.Locations {
if location.PublicUrl != *publicUrl {
if background {
go postContentFunc(location)
} else {
postContentFunc(location)
}
2012-09-21 00:58:29 +00:00
}
}
}
2012-09-21 08:30:31 +00:00
waitTime, err := strconv.Atoi(r.FormValue("wait"))
2012-09-21 00:58:29 +00:00
sendFunc(err == nil && waitTime > 0)
} else {
log.Println("Failed to lookup for", volumeId, lookupErr.Error())
}
}
}
2012-08-24 05:46:54 +00:00
m := make(map[string]uint32)
m["size"] = ret
writeJson(w, r, m)
}
}
}
func DeleteHandler(w http.ResponseWriter, r *http.Request) {
2012-08-24 05:46:54 +00:00
n := new(storage.Needle)
vid, fid, _ := parseURLPath(r.URL.Path)
2012-09-04 03:40:38 +00:00
volumeId, _ := storage.NewVolumeId(vid)
2012-08-24 05:46:54 +00:00
n.ParsePath(fid)
if *IsDebug {
log.Println("deleting", n)
}
cookie := n.Cookie
count, ok := store.Read(volumeId, n)
if ok != nil {
m := make(map[string]uint32)
m["size"] = 0
writeJson(w, r, m)
return
}
if n.Cookie != cookie {
log.Println("delete with unmaching cookie from ", r.RemoteAddr, "agent", r.UserAgent())
return
}
n.Size = 0
store.Delete(volumeId, n)
m := make(map[string]uint32)
m["size"] = uint32(count)
writeJson(w, r, m)
}
func parseURLPath(path string) (vid, fid, ext string) {
2012-08-24 05:46:54 +00:00
sepIndex := strings.LastIndex(path, "/")
commaIndex := strings.LastIndex(path[sepIndex:], ",")
if commaIndex <= 0 {
if "favicon.ico" != path[sepIndex+1:] {
log.Println("unknown file id", path[sepIndex+1:])
}
return
}
dotIndex := strings.LastIndex(path[sepIndex:], ".")
vid = path[sepIndex+1 : commaIndex]
fid = path[commaIndex+1:]
ext = ""
if dotIndex > 0 {
fid = path[commaIndex+1 : dotIndex]
ext = path[dotIndex:]
}
return
}
func runVolume(cmd *Command, args []string) bool {
fileInfo, err := os.Stat(*volumeFolder)
2012-08-24 05:46:54 +00:00
//TODO: now default to 1G, this value should come from server?
if err != nil {
log.Fatalf("No Existing Folder:%s", *volumeFolder)
}
if !fileInfo.IsDir() {
log.Fatalf("Volume Folder should not be a file:%s", *volumeFolder)
}
perm := fileInfo.Mode().Perm()
log.Println("Volume Folder permission:", perm)
store = storage.NewStore(*vport, *publicUrl, *volumeFolder, *maxVolumeCount, *volumes)
2012-08-24 05:46:54 +00:00
defer store.Close()
http.HandleFunc("/", storeHandler)
http.HandleFunc("/status", statusHandler)
2012-09-11 00:08:52 +00:00
http.HandleFunc("/admin/assign_volume", assignVolumeHandler)
2012-09-12 08:07:23 +00:00
http.HandleFunc("/admin/set_volume_locations_list", setVolumeLocationsHandler)
2012-08-24 05:46:54 +00:00
go func() {
for {
2012-09-08 23:25:44 +00:00
store.Join(*masterNode)
2012-09-04 03:40:38 +00:00
time.Sleep(time.Duration(float32(*vpulse*1e3)*(1+rand.Float32())) * time.Millisecond)
2012-08-24 05:46:54 +00:00
}
}()
2012-09-08 23:25:44 +00:00
log.Println("store joined at", *masterNode)
2012-08-24 05:46:54 +00:00
2012-09-03 22:41:24 +00:00
log.Println("Start storage service at http://127.0.0.1:"+strconv.Itoa(*vport), "public url", *publicUrl)
e := http.ListenAndServe(":"+strconv.Itoa(*vport), nil)
2012-08-24 05:46:54 +00:00
if e != nil {
log.Fatalf("Fail to start:", e)
}
return true
}