forked from TrueCloudLab/restic
69 lines
1.9 KiB
Go
69 lines
1.9 KiB
Go
package restore
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/restic/restic/internal/ui"
|
|
)
|
|
|
|
type jsonPrinter struct {
|
|
terminal term
|
|
}
|
|
|
|
func NewJSONProgress(terminal term) ProgressPrinter {
|
|
return &jsonPrinter{
|
|
terminal: terminal,
|
|
}
|
|
}
|
|
|
|
func (t *jsonPrinter) print(status interface{}) {
|
|
t.terminal.Print(ui.ToJSONString(status))
|
|
}
|
|
|
|
func (t *jsonPrinter) Update(filesFinished, filesTotal, allBytesWritten, allBytesTotal uint64, duration time.Duration) {
|
|
status := statusUpdate{
|
|
MessageType: "status",
|
|
SecondsElapsed: uint64(duration / time.Second),
|
|
TotalFiles: filesTotal,
|
|
FilesRestored: filesFinished,
|
|
TotalBytes: allBytesTotal,
|
|
BytesRestored: allBytesWritten,
|
|
}
|
|
|
|
if allBytesTotal > 0 {
|
|
status.PercentDone = float64(allBytesWritten) / float64(allBytesTotal)
|
|
}
|
|
|
|
t.print(status)
|
|
}
|
|
|
|
func (t *jsonPrinter) Finish(filesFinished, filesTotal, allBytesWritten, allBytesTotal uint64, duration time.Duration) {
|
|
status := summaryOutput{
|
|
MessageType: "summary",
|
|
SecondsElapsed: uint64(duration / time.Second),
|
|
TotalFiles: filesTotal,
|
|
FilesRestored: filesFinished,
|
|
TotalBytes: allBytesTotal,
|
|
BytesRestored: allBytesWritten,
|
|
}
|
|
t.print(status)
|
|
}
|
|
|
|
type statusUpdate struct {
|
|
MessageType string `json:"message_type"` // "status"
|
|
SecondsElapsed uint64 `json:"seconds_elapsed,omitempty"`
|
|
PercentDone float64 `json:"percent_done"`
|
|
TotalFiles uint64 `json:"total_files,omitempty"`
|
|
FilesRestored uint64 `json:"files_restored,omitempty"`
|
|
TotalBytes uint64 `json:"total_bytes,omitempty"`
|
|
BytesRestored uint64 `json:"bytes_restored,omitempty"`
|
|
}
|
|
|
|
type summaryOutput struct {
|
|
MessageType string `json:"message_type"` // "summary"
|
|
SecondsElapsed uint64 `json:"seconds_elapsed,omitempty"`
|
|
TotalFiles uint64 `json:"total_files,omitempty"`
|
|
FilesRestored uint64 `json:"files_restored,omitempty"`
|
|
TotalBytes uint64 `json:"total_bytes,omitempty"`
|
|
BytesRestored uint64 `json:"bytes_restored,omitempty"`
|
|
}
|