Browse Source
all switching to "weed command [args]" usage mode
all switching to "weed command [args]" usage mode
git-svn-id: https://weed-fs.googlecode.com/svn/trunk@65 282b0af5-e82d-9cf1-ede4-77906d7719d0pull/2/head
chris.lu@gmail.com
13 years ago
1 changed files with 0 additions and 117 deletions
@ -1,117 +0,0 @@ |
|||||
package main |
|
||||
|
|
||||
import ( |
|
||||
"bytes" |
|
||||
"encoding/json" |
|
||||
"errors" |
|
||||
"flag" |
|
||||
"fmt" |
|
||||
"io" |
|
||||
"io/ioutil" |
|
||||
"mime/multipart" |
|
||||
"net/http" |
|
||||
"net/url" |
|
||||
"os" |
|
||||
"pkg/util" |
|
||||
"strconv" |
|
||||
) |
|
||||
|
|
||||
var ( |
|
||||
server = flag.String("server", "localhost:9333", "weedfs master location") |
|
||||
IsDebug = flag.Bool("debug", false, "verbose debug information") |
|
||||
) |
|
||||
|
|
||||
type AssignResult struct { |
|
||||
Fid string "fid" |
|
||||
Url string "url" |
|
||||
PublicUrl string "publicUrl" |
|
||||
Count int `json:",string"` |
|
||||
Error string "error" |
|
||||
} |
|
||||
|
|
||||
func assign(count int) (AssignResult, error) { |
|
||||
values := make(url.Values) |
|
||||
values.Add("count", strconv.Itoa(count)) |
|
||||
jsonBlob := util.Post("http://"+*server+"/dir/assign", values) |
|
||||
var ret AssignResult |
|
||||
err := json.Unmarshal(jsonBlob, &ret) |
|
||||
if err != nil { |
|
||||
return ret, err |
|
||||
} |
|
||||
if ret.Count <= 0 { |
|
||||
return ret, errors.New(ret.Error) |
|
||||
} |
|
||||
return ret, nil |
|
||||
} |
|
||||
|
|
||||
type UploadResult struct { |
|
||||
Size int |
|
||||
} |
|
||||
|
|
||||
func upload(filename string, uploadUrl string) (int, string) { |
|
||||
body_buf := bytes.NewBufferString("") |
|
||||
body_writer := multipart.NewWriter(body_buf) |
|
||||
file_writer, err := body_writer.CreateFormFile("file", filename) |
|
||||
if err != nil { |
|
||||
panic(err.Error()) |
|
||||
} |
|
||||
fh, err := os.Open(filename) |
|
||||
if err != nil { |
|
||||
panic(err.Error()) |
|
||||
} |
|
||||
io.Copy(file_writer, fh) |
|
||||
content_type := body_writer.FormDataContentType() |
|
||||
body_writer.Close() |
|
||||
resp, err := http.Post(uploadUrl, content_type, body_buf) |
|
||||
if err != nil { |
|
||||
panic(err.Error()) |
|
||||
} |
|
||||
defer resp.Body.Close() |
|
||||
resp_body, err := ioutil.ReadAll(resp.Body) |
|
||||
if err != nil { |
|
||||
panic(err.Error()) |
|
||||
} |
|
||||
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.Size, uploadUrl |
|
||||
} |
|
||||
|
|
||||
type SubmitResult struct { |
|
||||
Fid string "fid" |
|
||||
Size int "size" |
|
||||
} |
|
||||
|
|
||||
func submit(files []string)([]SubmitResult) { |
|
||||
ret, err := assign(len(files)) |
|
||||
if err != nil { |
|
||||
panic(err) |
|
||||
} |
|
||||
results := make([]SubmitResult, len(files)) |
|
||||
for index, file := range files { |
|
||||
fid := ret.Fid |
|
||||
if index > 0 { |
|
||||
fid = fid + "_" + strconv.Itoa(index) |
|
||||
} |
|
||||
uploadUrl := "http://" + ret.PublicUrl + "/" + fid |
|
||||
results[index].Size, _ = upload(file, uploadUrl) |
|
||||
results[index].Fid = fid |
|
||||
} |
|
||||
return results |
|
||||
} |
|
||||
|
|
||||
func main() { |
|
||||
flag.Parse() |
|
||||
if len(flag.Args()) == 0 { |
|
||||
fmt.Fprintln(os.Stderr, "Submit one file or multiple version of the same file.") |
|
||||
fmt.Fprintf(os.Stderr, "Usage: %s -server=<host>:<port> file1 [file2 file3 ...]\n", os.Args[0]) |
|
||||
flag.Usage() |
|
||||
return |
|
||||
} |
|
||||
results := submit(flag.Args()) |
|
||||
bytes, _ := json.Marshal(results) |
|
||||
fmt.Print(string(bytes)) |
|
||||
} |
|
Write
Preview
Loading…
Cancel
Save
Reference in new issue