replication related work

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

@@ -1,7 +1,7 @@
 package main
 
 import (
-  "encoding/json"
+	"encoding/json"
 	"log"
 	"math/rand"
 	"mime"
@@ -40,6 +40,9 @@ func statusHandler(w http.ResponseWriter, r *http.Request) {
 	writeJson(w, r, store.Status())
 }
 func assignVolumeHandler(w http.ResponseWriter, r *http.Request) {
+	if *IsDebug {
+		log.Println("volume =", r.FormValue("volume"), ", replicationType =", r.FormValue("replicationType"))
+	}
 	err := store.AddVolume(r.FormValue("volume"), r.FormValue("replicationType"))
 	if err == nil {
 		writeJson(w, r, map[string]string{"error": ""})
@@ -48,14 +51,19 @@ func assignVolumeHandler(w http.ResponseWriter, r *http.Request) {
 	}
 }
 func setVolumeLocationsHandler(w http.ResponseWriter, r *http.Request) {
-  volumeLocationsList := new([]storage.VolumeLocations)
-  json.Unmarshal([]byte(r.FormValue("volumeLocations")), volumeLocationsList)
-	err := store.SetVolumeLocations(*volumeLocationsList)
-  if err == nil {
-    writeJson(w, r, map[string]string{"error": ""})
-  } else {
-    writeJson(w, r, map[string]string{"error": err.Error()})
-  }
+	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) {
 	switch r.Method {
@@ -173,7 +181,7 @@ func runVolume(cmd *Command, args []string) bool {
 	http.HandleFunc("/", storeHandler)
 	http.HandleFunc("/status", statusHandler)
 	http.HandleFunc("/admin/assign_volume", assignVolumeHandler)
-	http.HandleFunc("/admin/set_volume_locations", setVolumeLocationsHandler)
+	http.HandleFunc("/admin/set_volume_locations_list", setVolumeLocationsHandler)
 
 	go func() {
 		for {

weed-fs/src/pkg/admin/storage.go

@@ -0,0 +1,31 @@
+package admin
+
+import (
+  "encoding/json"
+  "errors"
+  "strconv"
+  "net/url"
+  "pkg/util"
+  "pkg/storage"
+  "pkg/topology"
+)
+
+type AllocateVolumeResult struct {
+  error string
+}
+
+func AllocateVolume(dn *topology.DataNode, vid storage.VolumeId, repType storage.ReplicationType) error{
+  values := make(url.Values)
+  values.Add("volume", vid.String())
+  values.Add("replicationType", repType.String())
+  jsonBlob := util.Post("http://"+dn.Ip+":"+strconv.Itoa(dn.Port)+"/admin/assign_volume", values)
+  var ret AllocateVolumeResult
+  err := json.Unmarshal(jsonBlob, &ret)
+  if err != nil {
+    return err
+  }
+  if ret.error != "" {
+    return errors.New(ret.error)
+  }
+  return nil
+}

weed-fs/src/pkg/storage/volume_info.go

@@ -33,6 +33,21 @@ func NewReplicationType(t string) ReplicationType {
   }
   return Copy00
 }
+func (r *ReplicationType) String() string {
+  switch *r {
+  case Copy00:
+    return "00"
+  case Copy01:
+    return "01"
+  case Copy10:
+    return "10"
+  case Copy11:
+    return "11"
+  case Copy20:
+    return "20"
+  }
+  return "00"
+}
 
 func GetReplicationLevelIndex(v *VolumeInfo) int {
 	switch v.RepType {

weed-fs/src/pkg/topology/data_node.go

@@ -8,9 +8,9 @@ import (
 type DataNode struct {
 	NodeImpl
 	volumes   map[storage.VolumeId]*storage.VolumeInfo
-	ip        string
-	port      int
-	publicUrl string
+	Ip        string
+	Port      int
+	PublicUrl string
 	lastSeen  int64 // unix time in seconds
 }
 

weed-fs/src/pkg/topology/volume_location.go

@@ -12,7 +12,7 @@ func NewDataNodeLocationList() *DataNodeLocationList {
 
 func (dnll *DataNodeLocationList) Add(loc *DataNode) {
 	for _, dnl := range dnll.list {
-		if loc.ip == dnl.ip && loc.port == dnl.port {
+		if loc.Ip == dnl.Ip && loc.Port == dnl.Port {
 			break
 		}
 	}