marc/scotty
1
0
Fork 0
mirror of https://git.sr.ht/~phw/scotty synced 2025-05-09 18:07:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Handle wait group for progress bar centrally

Browse source 
This does not need to be exposed and caller only
needs to wait for the Progress instance.
This commit is contained in:
Philipp Wolfer 2025-05-02 08:36:59 +02:00
parent b3136bde9a
commit 9480c69cbb
No known key found for this signature in database
GPG key ID: 8FDF744D4919943B
2 changed files with 5 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

5
internal/cli/progress.go
View file

@ -1,5 +1,5 @@
/*
Copyright © 2023 Philipp Wolfer <phw@uploadedlobster.com>
Copyright © 2023-2025 Philipp Wolfer <phw@uploadedlobster.com>
This file is part of Scotty.
@ -28,7 +28,8 @@ import (
"go.uploadedlobster.com/scotty/internal/models"
)
func progressBar(wg *sync.WaitGroup, exportProgress chan models.Progress, importProgress chan models.Progress) *mpb.Progress {
func progressBar(exportProgress chan models.Progress, importProgress chan models.Progress) *mpb.Progress {
wg := &sync.WaitGroup{}
p := mpb.New(
mpb.WithWaitGroup(wg),
mpb.WithOutput(color.Output),

7
internal/cli/transfer.go
View file

@ -1,5 +1,5 @@
/*
Copyright © 2023 Philipp Wolfer <phw@uploadedlobster.com>
Copyright © 2023-2025 Philipp Wolfer <phw@uploadedlobster.com>
Scotty is free software: you can redistribute it and/or modify it under the
terms of the GNU General Public License as published by the Free Software
@ -19,7 +19,6 @@ import (
"errors"
"fmt"
"strconv"
"sync"
"time"
"github.com/spf13/cobra"
@ -112,8 +111,7 @@ func (c *TransferCmd[E, I, R]) Transfer(exp backends.ExportProcessor[R], imp bac
// Prepare progress bars
exportProgress := make(chan models.Progress)
importProgress := make(chan models.Progress)
var wg sync.WaitGroup
progress := progressBar(&wg, exportProgress, importProgress)
progress := progressBar(exportProgress, importProgress)
// Export from source
exportChan := make(chan R, 1000)
@ -126,7 +124,6 @@ func (c *TransferCmd[E, I, R]) Transfer(exp backends.ExportProcessor[R], imp bac
if timestamp.After(result.LastTimestamp) {
result.LastTimestamp = timestamp
}
wg.Wait()
progress.Wait()
if result.Error != nil {
printTimestamp("Import failed, last reported timestamp was %v (%s)", result.LastTimestamp)