shezmu/examples/daemons-kafka/daemons/number_printer.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!")
		}
	}
}