mirror of
https://github.com/kemko/nomad.git
synced 2026-01-10 12:25:42 +03:00
0edda116ad
Adding '-verbose' will print out the allocation information for the deployment. This also changes the job run command so that it now blocks until deployment is complete and adds timestamps to the output so that it's more in line with the output of node drain. This uses glint to print in place in running in a tty. Because glint doesn't yet support cmd/powershell, Windows workflows use a different library to print in place, which results in slightly different formatting: 1) different margins, and 2) no spinner indicating deployment in progress.
89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
package pb
|
|
|
|
import (
|
|
"math/rand"
|
|
"sync"
|
|
"text/template"
|
|
|
|
"github.com/fatih/color"
|
|
)
|
|
|
|
// ProgressBarTemplate that template string
|
|
type ProgressBarTemplate string
|
|
|
|
// New creates new bar from template
|
|
func (pbt ProgressBarTemplate) New(total int) *ProgressBar {
|
|
return New(total).SetTemplate(pbt)
|
|
}
|
|
|
|
// Start64 create and start new bar with given int64 total value
|
|
func (pbt ProgressBarTemplate) Start64(total int64) *ProgressBar {
|
|
return New64(total).SetTemplate(pbt).Start()
|
|
}
|
|
|
|
// Start create and start new bar with given int total value
|
|
func (pbt ProgressBarTemplate) Start(total int) *ProgressBar {
|
|
return pbt.Start64(int64(total))
|
|
}
|
|
|
|
var templateCacheMu sync.Mutex
|
|
var templateCache = make(map[string]*template.Template)
|
|
|
|
var defaultTemplateFuncs = template.FuncMap{
|
|
// colors
|
|
"black": color.New(color.FgBlack).SprintFunc(),
|
|
"red": color.New(color.FgRed).SprintFunc(),
|
|
"green": color.New(color.FgGreen).SprintFunc(),
|
|
"yellow": color.New(color.FgYellow).SprintFunc(),
|
|
"blue": color.New(color.FgBlue).SprintFunc(),
|
|
"magenta": color.New(color.FgMagenta).SprintFunc(),
|
|
"cyan": color.New(color.FgCyan).SprintFunc(),
|
|
"white": color.New(color.FgWhite).SprintFunc(),
|
|
"resetcolor": color.New(color.Reset).SprintFunc(),
|
|
"rndcolor": rndcolor,
|
|
"rnd": rnd,
|
|
}
|
|
|
|
func getTemplate(tmpl string) (t *template.Template, err error) {
|
|
templateCacheMu.Lock()
|
|
defer templateCacheMu.Unlock()
|
|
t = templateCache[tmpl]
|
|
if t != nil {
|
|
// found in cache
|
|
return
|
|
}
|
|
t = template.New("")
|
|
fillTemplateFuncs(t)
|
|
_, err = t.Parse(tmpl)
|
|
if err != nil {
|
|
t = nil
|
|
return
|
|
}
|
|
templateCache[tmpl] = t
|
|
return
|
|
}
|
|
|
|
func fillTemplateFuncs(t *template.Template) {
|
|
t.Funcs(defaultTemplateFuncs)
|
|
emf := make(template.FuncMap)
|
|
elementsM.Lock()
|
|
for k, v := range elements {
|
|
emf[k] = v
|
|
}
|
|
elementsM.Unlock()
|
|
t.Funcs(emf)
|
|
return
|
|
}
|
|
|
|
func rndcolor(s string) string {
|
|
c := rand.Intn(int(color.FgWhite-color.FgBlack)) + int(color.FgBlack)
|
|
return color.New(color.Attribute(c)).Sprint(s)
|
|
}
|
|
|
|
func rnd(args ...string) string {
|
|
if len(args) == 0 {
|
|
return ""
|
|
}
|
|
return args[rand.Intn(len(args))]
|
|
}
|