util: Add subscribed signal primitive

Add a little sync primitive to our utility library. This should
hopefully make some of the future code easier to deal with.
This commit is contained in:
James Shubin
2019-03-15 18:09:55 -04:00
parent 6628fc02f2
commit 398706246e
2 changed files with 149 additions and 0 deletions

View File

@@ -20,6 +20,8 @@
package util
import (
"fmt"
"sync"
"testing"
"time"
)
@@ -61,3 +63,84 @@ func TestEasyAck3(t *testing.T) {
t.Errorf("the second Ack did not arrive in time")
}
}
func ExampleSubscribeSync() {
fmt.Println("hello")
x := &SubscribedSignal{}
wg := &sync.WaitGroup{}
ready := &sync.WaitGroup{}
// unit1
wg.Add(1)
ready.Add(1)
go func() {
defer wg.Done()
ch, ack := x.Subscribe()
ready.Done()
select {
case <-ch:
fmt.Println("got signal")
}
time.Sleep(1 * time.Second) // wait a bit for fun
fmt.Println("(1) sending ack...")
ack() // must call ack
fmt.Println("done sending ack")
}()
// unit2
wg.Add(1)
ready.Add(1)
go func() {
defer wg.Done()
ch, ack := x.Subscribe()
ready.Done()
select {
case <-ch:
fmt.Println("got signal")
}
time.Sleep(2 * time.Second) // wait a bit for fun
fmt.Println("(2) sending ack...")
ack() // must call ack
fmt.Println("done sending ack")
}()
// unit3
wg.Add(1)
ready.Add(1)
go func() {
defer wg.Done()
ch, ack := x.Subscribe()
ready.Done()
select {
case <-ch:
fmt.Println("got signal")
}
time.Sleep(3 * time.Second) // wait a bit for fun
fmt.Println("(3) sending ack...")
ack() // must call ack
fmt.Println("done sending ack")
}()
ready.Wait() // wait for all subscribes
fmt.Println("sending signal...")
x.Send() // trigger!
fmt.Println("done sending signal")
wg.Wait() // wait for everyone to exit
fmt.Println("exiting...")
// Output: hello
// sending signal...
// got signal
// got signal
// got signal
// (1) sending ack...
// (2) sending ack...
// (3) sending ack...
// done sending signal
// done sending ack
// done sending ack
// done sending ack
// exiting...
}