From f2ebb86bf3aa0fdd508ea659c22d650c52bc5d52 Mon Sep 17 00:00:00 2001 From: Carl Kittelberger Date: Thu, 11 Jul 2019 14:47:30 +0200 Subject: [PATCH] Use Connected event to mark end of file transfer. --- cmd/sendaround/main.go | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/cmd/sendaround/main.go b/cmd/sendaround/main.go index 0229b6d..3133b86 100644 --- a/cmd/sendaround/main.go +++ b/cmd/sendaround/main.go @@ -82,8 +82,11 @@ func main() { pbars[state.CurrentFile.FileName()].Start() } pbars[state.CurrentFile.FileName()].SetCurrent(int64(state.TransmittedLength)) - if state.TransmittedLength == state.CurrentFile.Length() { - pbars[state.CurrentFile.FileName()].Finish() + case sendaround.Connected: + if state.CurrentFile != nil { // Was a file transferred successfully just now? + if state.TransmittedLength == state.CurrentFile.Length() { + pbars[state.CurrentFile.FileName()].Finish() + } } case sendaround.Failed: log.Fatal(state.Error) @@ -161,8 +164,11 @@ func main() { pbars[state.CurrentFile.FileName()].Start() } pbars[state.CurrentFile.FileName()].SetCurrent(int64(state.TransmittedLength)) - if state.TransmittedLength == state.CurrentFile.Length() { - pbars[state.CurrentFile.FileName()].Finish() + case sendaround.Connected: + if state.CurrentFile != nil { // Was a file transferred successfully just now? + if state.TransmittedLength == state.CurrentFile.Length() { + pbars[state.CurrentFile.FileName()].Finish() + } } case sendaround.Failed: log.Fatal("Failed:", state.Error)