func (db *serviceClient) CellEvents(logger lager.Logger) <-chan models.CellEvent {
logger = logger.Session("cell-events")
disappearanceWatcher, disappeared := locket.NewDisappearanceWatcher(logger, db.consulClient, CellSchemaRoot(), db.clock)
process := ifrit.Invoke(disappearanceWatcher)
events := make(chan models.CellEvent)
go func() {
for {
select {
case keys, ok := <-disappeared:
if !ok {
process.Signal(os.Interrupt)
return
}
cellIDs := make([]string, len(keys))
for i, key := range keys {
cellIDs[i] = path.Base(key)
}
logger.Info("cell-disappeared", lager.Data{"cell_ids": cellIDs})
events <- models.NewCellDisappearedEvent(cellIDs)
}
}
}()
return events
}
var (
consulClient consuladapter.Client
watcherRunner ifrit.Runner
watcherProcess ifrit.Process
disappearChan <-chan []string
logger *lagertest.TestLogger
)
BeforeEach(func() {
consulClient = consulRunner.NewClient()
logger = lagertest.NewTestLogger("test")
clock := clock.NewClock()
watcherRunner, disappearChan = locket.NewDisappearanceWatcher(logger, consulClient, "under", clock)
watcherProcess = ifrit.Invoke(watcherRunner)
})
AfterEach(func() {
ginkgomon.Kill(watcherProcess)
})
var addThenRemovePresence = func(presenceKey string) {
presenceRunner := locket.NewPresence(logger, consulClient, presenceKey, []byte("value"), clock.NewClock(), retryInterval, 10*time.Second)
presenceProcess := ifrit.Invoke(presenceRunner)
Eventually(func() int {
sessions, _, err := consulClient.Session().List(nil)
Expect(err).NotTo(HaveOccurred())
return len(sessions)