aboutsummaryrefslogtreecommitdiff
path: root/widgets/spinner.go
diff options
context:
space:
mode:
Diffstat (limited to 'widgets/spinner.go')
-rw-r--r--widgets/spinner.go27
1 files changed, 15 insertions, 12 deletions
diff --git a/widgets/spinner.go b/widgets/spinner.go
index bafc712..0ab3e13 100644
--- a/widgets/spinner.go
+++ b/widgets/spinner.go
@@ -1,6 +1,7 @@
package widgets
import (
+ "sync/atomic"
"time"
"github.com/gdamore/tcell"
@@ -22,14 +23,14 @@ var (
)
type Spinner struct {
- frame int
+ frame int64 // access via atomic
onInvalidate func(d ui.Drawable)
- stop chan interface{}
+ stop chan struct{}
}
func NewSpinner() *Spinner {
spinner := Spinner{
- stop: make(chan interface{}),
+ stop: make(chan struct{}),
frame: -1,
}
return &spinner
@@ -40,17 +41,17 @@ func (s *Spinner) Start() {
return
}
- s.frame = 0
+ atomic.StoreInt64(&s.frame, 0)
+
go func() {
for {
select {
case <-s.stop:
+ atomic.StoreInt64(&s.frame, -1)
+ s.stop <- struct{}{}
return
case <-time.After(200 * time.Millisecond):
- s.frame++
- if s.frame >= len(frames) {
- s.frame = 0
- }
+ atomic.AddInt64(&s.frame, 1)
s.Invalidate()
}
}
@@ -62,13 +63,13 @@ func (s *Spinner) Stop() {
return
}
- s.stop <- nil
- s.frame = -1
+ s.stop <- struct{}{}
+ <-s.stop
s.Invalidate()
}
func (s *Spinner) IsRunning() bool {
- return s.frame != -1
+ return atomic.LoadInt64(&s.frame) != -1
}
func (s *Spinner) Draw(ctx *ui.Context) {
@@ -76,9 +77,11 @@ func (s *Spinner) Draw(ctx *ui.Context) {
s.Start()
}
+ cur := int(atomic.LoadInt64(&s.frame) % int64(len(frames)))
+
ctx.Fill(0, 0, ctx.Width(), ctx.Height(), ' ', tcell.StyleDefault)
col := ctx.Width()/2 - len(frames[0])/2 + 1
- ctx.Printf(col, 0, tcell.StyleDefault, "%s", frames[s.frame])
+ ctx.Printf(col, 0, tcell.StyleDefault, "%s", frames[cur])
}
func (s *Spinner) OnInvalidate(onInvalidate func(d ui.Drawable)) {