refactoring content uploading

This commit is contained in:
Chris Lu
2012-09-20 17:58:29 -07:00
parent 7d8e9f829c
commit 5e97179d06
6 changed files with 74 additions and 47 deletions

View File

@@ -0,0 +1,40 @@
package operation
import (
"bytes"
"encoding/json"
"mime/multipart"
"net/http"
_ "fmt"
"io"
"io/ioutil"
)
type UploadResult struct {
Size int
}
func Upload(server string, fid 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("http://"+server+"/"+fid, content_type, body_buf)
if err != nil {
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 {
panic(err.Error())
}
//fmt.Println("Uploaded " + strconv.Itoa(ret.Size) + " Bytes to " + uploadUrl)
return &ret, nil
}

View File

@@ -135,6 +135,9 @@ func (s *Store) Read(i VolumeId, n *Needle) (int, error) {
}
return 0, errors.New("Not Found")
}
func (s *Store) GetVolume(i VolumeId) *Volume {
return s.volumes[i]
}
func (s *Store) HasVolume(i VolumeId) bool {
_, ok := s.volumes[i]