HeartbeatRoutine: return context.CancelFunc instead of chan struct{}

[serprex]

context.WithCancel is a bit hairy in its implementation since internally they use a lazily allocated chan struct{} themselves, but seems idiomatic to use it

HeartbeatRoutine could keep creating its own chan struct{} & instead return func() { close(shutdown) } to keep the same API