diff options
Diffstat (limited to 'go/operation')
| -rw-r--r-- | go/operation/allocate_volume.go | 32 | ||||
| -rw-r--r-- | go/operation/delete_content.go | 16 | ||||
| -rw-r--r-- | go/operation/lookup_volume_id.go | 38 | ||||
| -rw-r--r-- | go/operation/upload_content.go | 47 |
4 files changed, 133 insertions, 0 deletions
diff --git a/go/operation/allocate_volume.go b/go/operation/allocate_volume.go new file mode 100644 index 000000000..19166eaed --- /dev/null +++ b/go/operation/allocate_volume.go @@ -0,0 +1,32 @@ +package operation + +import ( + "encoding/json" + "errors" + "net/url" + "code.google.com/p/weed-fs/go/storage" + "code.google.com/p/weed-fs/go/topology" + "code.google.com/p/weed-fs/go/util" +) + +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, err := util.Post("http://"+dn.Url()+"/admin/assign_volume", values) + if err != nil { + return err + } + var ret AllocateVolumeResult + if err := json.Unmarshal(jsonBlob, &ret); err != nil { + return err + } + if ret.Error != "" { + return errors.New(ret.Error) + } + return nil +} diff --git a/go/operation/delete_content.go b/go/operation/delete_content.go new file mode 100644 index 000000000..2bdb49651 --- /dev/null +++ b/go/operation/delete_content.go @@ -0,0 +1,16 @@ +package operation + +import ( + "log" + "net/http" +) + +func Delete(url string) error { + req, err := http.NewRequest("DELETE", url, nil) + if err != nil { + log.Println("failing to delete", url) + return err + } + _, err = http.DefaultClient.Do(req) + return err +} diff --git a/go/operation/lookup_volume_id.go b/go/operation/lookup_volume_id.go new file mode 100644 index 000000000..516ba9129 --- /dev/null +++ b/go/operation/lookup_volume_id.go @@ -0,0 +1,38 @@ +package operation + +import ( + "encoding/json" + "errors" + _ "fmt" + "net/url" + "code.google.com/p/weed-fs/go/storage" + "code.google.com/p/weed-fs/go/util" +) + +type Location struct { + Url string "url" + PublicUrl string "publicUrl" +} +type LookupResult struct { + Locations []Location "locations" + Error string "error" +} + +//TODO: Add a caching for vid here +func Lookup(server string, vid storage.VolumeId) (*LookupResult, error) { + values := make(url.Values) + values.Add("volumeId", vid.String()) + jsonBlob, err := util.Post("http://"+server+"/dir/lookup", values) + if err != nil { + return nil, err + } + var ret LookupResult + err = json.Unmarshal(jsonBlob, &ret) + if err != nil { + return nil, err + } + if ret.Error != "" { + return nil, errors.New(ret.Error) + } + return &ret, nil +} diff --git a/go/operation/upload_content.go b/go/operation/upload_content.go new file mode 100644 index 000000000..0bdb697da --- /dev/null +++ b/go/operation/upload_content.go @@ -0,0 +1,47 @@ +package operation + +import ( + "bytes" + "encoding/json" + "errors" + _ "fmt" + "io" + "io/ioutil" + "log" + "mime/multipart" + "net/http" +) + +type UploadResult struct { + Size int + Error string +} + +func Upload(uploadUrl string, filename string, reader io.Reader) (*UploadResult, error) { + body_buf := bytes.NewBufferString("") + body_writer := multipart.NewWriter(body_buf) + file_writer, err := body_writer.CreateFormFile("file", filename) + io.Copy(file_writer, reader) + content_type := body_writer.FormDataContentType() + body_writer.Close() + resp, err := http.Post(uploadUrl, content_type, body_buf) + if err != nil { + log.Println("failing to upload to", uploadUrl) + return nil, err + } + defer resp.Body.Close() + resp_body, err := ioutil.ReadAll(resp.Body) + if err != nil { + return nil, err + } + var ret UploadResult + err = json.Unmarshal(resp_body, &ret) + if err != nil { + log.Println("failing to read upload resonse", uploadUrl, resp_body) + return nil, err + } + if ret.Error != "" { + return nil, errors.New(ret.Error) + } + return &ret, nil +} |
