From 8161d86dfd52a8506f634ee3989b861531c3aae7 Mon Sep 17 00:00:00 2001 From: wanyaoqi Date: Fri, 18 Dec 2020 18:25:28 +0800 Subject: [PATCH] fix(host): add print migrate progress info --- .../storageman/remotefile/remotefile.go | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/pkg/hostman/storageman/remotefile/remotefile.go b/pkg/hostman/storageman/remotefile/remotefile.go index 7b7c67b488..787ee44f08 100644 --- a/pkg/hostman/storageman/remotefile/remotefile.go +++ b/pkg/hostman/storageman/remotefile/remotefile.go @@ -235,8 +235,25 @@ func (r *SRemoteFile) downloadInternal(getData bool, preChksum string) bool { defer zlibRC.Close() reader = zlibRC } - + var finishChan = make(chan struct{}) + go func() { + defer recover() + for { + select { + case <-time.After(10 * time.Second): + info, err := fi.Stat() + if err != nil { + log.Errorf("failed stat file %s", r.tmpPath) + return + } + log.Infof("written file %s size %dM", r.tmpPath, info.Size()/1024/1024) + case <-finishChan: + return + } + } + }() _, err = io.Copy(fi, reader) + close(finishChan) if err != nil { log.Errorln(err) return false