cybuzuma/itd
1
0
Fork 0
forked from Elara6331/itd
Code Issues Pull Requests Projects Releases Wiki Activity

Switch from custom socket API to rpcx

Browse Source
This commit is contained in:
Elara
2022-04-22 17:12:30 -07:00
parent d318c584da
commit 0cdf8a4bed
24 changed files with 1569 additions and 1311 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
cmd/itctl/fs.go
View File

@@ -76,7 +76,7 @@ func fsRead(c *cli.Context) error {
}
}
progress, err := client.ReadFile(path, c.Args().Get(0))
progress, err := client.Download(path, c.Args().Get(0))
if err != nil {
return err
}
@@ -91,12 +91,8 @@ func fsRead(c *cli.Context) error {
bar.SetTotal(int64(event.Total))
// Set amount of bytes sent in progress bar
bar.SetCurrent(int64(event.Sent))
// If transfer finished, break
if event.Done {
bar.Finish()
break
}
}
bar.Finish()
if c.Args().Get(1) == "-" {
io.Copy(os.Stdout, tmpFile)
@@ -148,7 +144,7 @@ func fsWrite(c *cli.Context) error {
defer os.Remove(path)
}
progress, err := client.WriteFile(path, c.Args().Get(1))
progress, err := client.Upload(c.Args().Get(1), path)
if err != nil {
return err
}
@@ -163,11 +159,6 @@ func fsWrite(c *cli.Context) error {
bar.SetTotal(int64(event.Total))
// Set amount of bytes sent in progress bar
bar.SetCurrent(int64(event.Sent))
// If transfer finished, break
if event.Done {
bar.Finish()
break
}
}
return nil