50 lines
938 B
Go
50 lines
938 B
Go
package suture
|
|
|
|
import "fmt"
|
|
|
|
type Incrementor struct {
|
|
current int
|
|
next chan int
|
|
stop chan bool
|
|
}
|
|
|
|
func (i *Incrementor) Stop() {
|
|
fmt.Println("Stopping the service")
|
|
i.stop <- true
|
|
}
|
|
|
|
func (i *Incrementor) Serve() {
|
|
for {
|
|
select {
|
|
case i.next <- i.current:
|
|
i.current++
|
|
case <-i.stop:
|
|
// We sync here just to guarantee the output of "Stopping the service",
|
|
// so this passes the test reliably.
|
|
// Most services would simply "return" here.
|
|
i.stop <- true
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func ExampleNew_simple() {
|
|
supervisor := NewSimple("Supervisor")
|
|
service := &Incrementor{0, make(chan int), make(chan bool)}
|
|
supervisor.Add(service)
|
|
|
|
go supervisor.ServeBackground()
|
|
|
|
fmt.Println("Got:", <-service.next)
|
|
fmt.Println("Got:", <-service.next)
|
|
supervisor.Stop()
|
|
|
|
// We sync here just to guarantee the output of "Stopping the service"
|
|
<-service.stop
|
|
|
|
// Output:
|
|
// Got: 0
|
|
// Got: 1
|
|
// Stopping the service
|
|
}
|