dolthub · timsehn · Jun 3, 2025 · Jun 2, 2025 · Jun 2, 2025 · Jun 2, 2025
diff --git a/go/libraries/doltcore/env/actions/clone.go b/go/libraries/doltcore/env/actions/clone.go
@@ -131,6 +131,7 @@ func clonePrint(eventCh <-chan pull.TableFileEvent) {
 		case pull.DownloadFailed:
 			// Ignore for now and output errors on the main thread
 			for _, tf := range tblFEvt.TableFiles {
+				chunksDownloading -= int64(tf.NumChunks())
 				delete(currStats, tf.FileID())
 			}
 		}
@@ -146,6 +147,10 @@ func clonePrint(eventCh <-chan pull.TableFileEvent) {
 		}
 		p.Display()
 	}
+
+	// Final status: ensure we show all chunks as complete when clone finishes
+	p.Printf("%s of %s chunks complete.\n",
+		strhelp.CommaIfy(chunksC), strhelp.CommaIfy(chunksC))
 	p.Display()
 }
 

diff --git a/go/libraries/utils/iohelp/read_with_stats.go b/go/libraries/utils/iohelp/read_with_stats.go
@@ -29,11 +29,12 @@ type ReadStats struct {
 }
 
 type ReaderWithStats struct {
-	read    uint64
-	size    int64
-	rd      io.Reader
-	start   time.Time
-	closeCh chan struct{}
+	read      uint64
+	size      int64
+	rd        io.Reader
+	start     time.Time
+	closeCh   chan struct{}
+	runningCh chan struct{}
 }
 
 func NewReaderWithStats(rd io.Reader, size int64) *ReaderWithStats {
@@ -45,6 +46,11 @@ func NewReaderWithStats(rd io.Reader, size int64) *ReaderWithStats {
 }
 
 func (rws *ReaderWithStats) Start(updateFunc func(ReadStats)) {
+	if rws.runningCh != nil {
+		panic("cannot start ReaderWithStats more than once.")
+	}
+	rws.runningCh = make(chan struct{})
+	defer close(rws.runningCh)
 	rws.start = time.Now()
 	go func() {
 		timer := time.NewTimer(updateFrequency)
@@ -67,6 +73,10 @@ func (rws *ReaderWithStats) Start(updateFunc func(ReadStats)) {
 }
 
 func (rws *ReaderWithStats) Close() error {
+	// Ensure that we never call |updateFunc| after Close() returns.
+	if rws.runningCh != nil {
+		<-rws.runningCh
+	}
 	close(rws.closeCh)
 
 	if closer, ok := rws.rd.(io.Closer); ok {