Commit 40d8360a5719990d0a30119e5a76997ce0faf66a - golang-github-vbauerster-mpb

Syncable interface Vladimir Bauer 7 years ago

2 changed file(s) with 18 addition(s) and 6 deletion(s). Raw diff Collapse all Expand all

-2

bar.go less more

405	405	columns := make([]chan int, 0, len(s.pDecorators)+len(s.aDecorators))
406	406	var pCount int
407	407	for _, d := range s.pDecorators {
408		if ok, ch := d.SyncWidth(); ok {
	408	if ok, ch := d.Syncable(); ok {
409	409	columns = append(columns, ch)
410	410	pCount++
411	411	}
412	412	}
413	413	var aCount int
414	414	for _, d := range s.aDecorators {
415		if ok, ch := d.SyncWidth(); ok {
	415	if ok, ch := d.Syncable(); ok {
416	416	columns = append(columns, ch)
417	417	aCount++
418	418	}

+16

-4

decor/decorator.go less more

46	46	}
47	47
48	48	// Decorator interface.
	49	// A decorator must implement this interface, in order to be used with mpb library.
49	50	type Decorator interface {
50	51	Decor(*Statistics) string
51		SyncWidth() (bool, chan int)
	52	Syncable
	53	}
	54
	55	// Syncable interface.
	56	// All decorators implement this interface implicitly.
	57	// Its Syncable method exposes width sync channel, if sync is enabled.
	58	type Syncable interface {
	59	Syncable() (bool, chan int)
52	60	}
53	61
54	62	// OnCompleteMessenger interface.

57	65	OnCompleteMessage(string)
58	66	}
59	67
	68	// AmountReceiver interface.
	69	// If decorator needs to receive increment amount,
	70	// so this is the right interface to implement.
60	71	type AmountReceiver interface {
61	72	NextAmount(int, ...time.Duration)
62	73	}
63	74
64	75	// ShutdownListener interface.
65		// If decorator implements this interface, its Shutdown method
66		// will be called once on bar shutdown event.
	76	// If decorator needs to be notified once upon bar shutdown event,
	77	// so this is the right interface to implement.
67	78	type ShutdownListener interface {
68	79	Shutdown()
69	80	}

114	125	}
115	126	}
116	127
117		func (wc *WC) SyncWidth() (bool, chan int) {
	128	func (wc *WC) Syncable() (bool, chan int) {
118	129	return (wc.C & DSyncWidth) != 0, wc.wsync
119	130	}
120	131

131	142	return decorator
132	143	}
133	144
	145	// completeMsg for internal usage.
134	146	type completeMsg struct {
135	147	msg string
136	148	}