celogeek/piwigo-cli
1
0
Fork 0
mirror of https://github.com/celogeek/piwigo-cli.git synced 2025-05-25 02:02:37 +02:00
Code Issues Projects Releases Wiki Activity

defer lock and handle errors

Browse Source
This commit is contained in:
Celogeek 2024-03-03 12:05:44 +01:00
parent a9254b7977
commit 40a79e8e6b
Signed by: celogeek
SSH Key Fingerprint: SHA256:njNJLzoLQdbV9PC6ehcruRb0QnEgxABoCYZ+0+aUIYc
1 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
internal/piwigo/piwigotools/file_to_upload_stat.go
View File

@ -22,7 +22,7 @@ type FileToUploadStat struct {
func NewFileToUploadStat() *FileToUploadStat {
bar := progressbar.DefaultBytes(1, "...")
progressbar.OptionOnCompletion(func() { os.Stderr.WriteString("\n") })(bar)
progressbar.OptionOnCompletion(func() { _, _ = os.Stderr.WriteString("\n") })(bar)
return &FileToUploadStat{
Progress: bar,
}
@ -43,63 +43,70 @@ func (s *FileToUploadStat) Refresh() {
func (s *FileToUploadStat) Check() {
s.mu.Lock()
defer s.mu.Unlock()
s.Checked++
s.Refresh()
s.mu.Unlock()
}
func (s *FileToUploadStat) AddBytes(filesize int64) {
s.mu.Lock()
defer s.mu.Unlock()
s.TotalBytes += filesize
s.Progress.ChangeMax64(s.TotalBytes + 1)
s.Refresh()
s.mu.Unlock()
}
func (s *FileToUploadStat) Add() {
s.mu.Lock()
defer s.mu.Unlock()
s.Total++
s.Refresh()
s.mu.Unlock()
}
func (s *FileToUploadStat) Commit(fileread int64) {
s.mu.Lock()
defer s.mu.Unlock()
s.UploadedBytes += fileread
s.Progress.Set64(s.UploadedBytes)
s.mu.Unlock()
_ = s.Progress.Set64(s.UploadedBytes)
}
func (s *FileToUploadStat) Done() {
s.mu.Lock()
defer s.mu.Unlock()
s.Uploaded++
s.Refresh()
s.mu.Unlock()
}
func (s *FileToUploadStat) Close() {
s.Progress.Close()
_ = s.Progress.Close()
}
func (s *FileToUploadStat) Fail() {
s.mu.Lock()
defer s.mu.Unlock()
s.Failed++
s.Refresh()
s.mu.Unlock()
}
func (s *FileToUploadStat) Skip() {
s.mu.Lock()
defer s.mu.Unlock()
s.Skipped++
s.Refresh()
s.mu.Unlock()
}
func (s *FileToUploadStat) Error(origin string, filename string, err error) error {
func (s *FileToUploadStat) Error(origin string, filename string, err error) {
s.mu.Lock()
s.Progress.Clear()
defer s.mu.Unlock()
_ = s.Progress.Clear()
fmt.Printf("[%s] %s: %s\n", origin, filename, err)
s.Progress.RenderBlank()
s.mu.Unlock()
return err
_ = s.Progress.RenderBlank()
}