mirror of
https://github.com/ollama/ollama.git
synced 2025-05-11 18:36:41 +02:00
progress
This commit is contained in:
parent
9f04e5a8ea
commit
c4a3ccd7ac
3 changed files with 285 additions and 0 deletions
118
progress/bar.go
Normal file
118
progress/bar.go
Normal file
|
@ -0,0 +1,118 @@
|
|||
package progress
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/jmorganca/ollama/format"
|
||||
"golang.org/x/term"
|
||||
)
|
||||
|
||||
type Bar struct {
|
||||
message string
|
||||
messageWidth int
|
||||
|
||||
maxValue int64
|
||||
initialValue int64
|
||||
currentValue int64
|
||||
|
||||
started time.Time
|
||||
stopped time.Time
|
||||
}
|
||||
|
||||
func NewBar(message string, maxValue, initialValue int64) *Bar {
|
||||
return &Bar{
|
||||
message: message,
|
||||
messageWidth: -1,
|
||||
maxValue: maxValue,
|
||||
initialValue: initialValue,
|
||||
currentValue: initialValue,
|
||||
started: time.Now(),
|
||||
}
|
||||
}
|
||||
|
||||
func (b *Bar) String() string {
|
||||
termWidth, _, err := term.GetSize(int(os.Stderr.Fd()))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
var pre, mid, suf strings.Builder
|
||||
|
||||
if b.message != "" {
|
||||
message := strings.TrimSpace(b.message)
|
||||
if b.messageWidth > 0 && len(message) > b.messageWidth {
|
||||
message = message[:b.messageWidth]
|
||||
}
|
||||
|
||||
fmt.Fprintf(&pre, "%s", message)
|
||||
if b.messageWidth-pre.Len() >= 0 {
|
||||
pre.WriteString(strings.Repeat(" ", b.messageWidth-pre.Len()))
|
||||
}
|
||||
|
||||
pre.WriteString(" ")
|
||||
}
|
||||
|
||||
fmt.Fprintf(&pre, "%.1f%% ", b.percent())
|
||||
|
||||
fmt.Fprintf(&suf, "(%s/%s, %s/s, %s)",
|
||||
format.HumanBytes(b.currentValue),
|
||||
format.HumanBytes(b.maxValue),
|
||||
format.HumanBytes(int64(b.rate())),
|
||||
b.elapsed())
|
||||
|
||||
mid.WriteString("[")
|
||||
|
||||
// pad 3 for last = or > and "] "
|
||||
f := termWidth - pre.Len() - mid.Len() - suf.Len() - 3
|
||||
n := int(float64(f) * b.percent() / 100)
|
||||
if n > 0 {
|
||||
mid.WriteString(strings.Repeat("=", n))
|
||||
}
|
||||
|
||||
if b.currentValue >= b.maxValue {
|
||||
mid.WriteString("=")
|
||||
} else {
|
||||
mid.WriteString(">")
|
||||
}
|
||||
|
||||
if f-n > 0 {
|
||||
mid.WriteString(strings.Repeat(" ", f-n))
|
||||
}
|
||||
|
||||
mid.WriteString("] ")
|
||||
|
||||
return pre.String() + mid.String() + suf.String()
|
||||
}
|
||||
|
||||
func (b *Bar) Set(value int64) {
|
||||
if value >= b.maxValue {
|
||||
value = b.maxValue
|
||||
b.stopped = time.Now()
|
||||
}
|
||||
|
||||
b.currentValue = value
|
||||
}
|
||||
|
||||
func (b *Bar) percent() float64 {
|
||||
if b.maxValue > 0 {
|
||||
return float64(b.currentValue) / float64(b.maxValue) * 100
|
||||
}
|
||||
|
||||
return 0
|
||||
}
|
||||
|
||||
func (b *Bar) rate() float64 {
|
||||
return (float64(b.currentValue) - float64(b.initialValue)) / b.elapsed().Seconds()
|
||||
}
|
||||
|
||||
func (b *Bar) elapsed() time.Duration {
|
||||
stopped := b.stopped
|
||||
if stopped.IsZero() {
|
||||
stopped = time.Now()
|
||||
}
|
||||
|
||||
return stopped.Sub(b.started).Round(time.Second)
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue