1
0
Fork 0
shezmu/examples/daemons-kafka/daemons/number_printer.go

46 lines
1.0 KiB
Go

package daemons
import (
"math/rand"
"time"
"github.com/localhots/shezmu"
)
// NumberPrinter is a daemon that prints numbers once in a while.
type NumberPrinter struct {
shezmu.BaseDaemon
}
// Startup sets up panic handler and starts enqueuing number printing jobs.
func (n *NumberPrinter) Startup() {
n.HandlePanics(func(err error) {
n.Logf("Oh, crap! There was a panic, take a look: %s", err.Error())
})
n.LimitRate(3, time.Second)
n.SystemProcess("Random Number Generator", n.generateNumbers)
}
func (n *NumberPrinter) generateNumbers() {
for n.Continue() {
if rand.Intn(10) == 0 {
panic("Number generator refuses to work right now!")
}
// Generate a random number between 1000 and 9999 and print it
num := 1000 + rand.Intn(9000)
n.Process(n.makeActor(num))
}
}
func (n *NumberPrinter) makeActor(num int) shezmu.Actor {
return func() {
n.Log("Number printer says:", num)
// Making it crash sometimes
if num%10 == 0 {
panic("Nooooo! Random number generator returned a multiple of ten!")
}
}
}