Exemple #1
0
func publishEvents(client publisher.Client, events []common.MapStr) {

	// Sends event to beat (outputs).
	// Wait/Repeat until all events are published
	for {
		ok := client.PublishEvents(events, publisher.Confirm)
		if ok {
			break
		}
	}
}
Exemple #2
0
func (ab *AmqpBeat) publishStream(stream <-chan []*AmqpEvent, client publisher.Client, wg *sync.WaitGroup,
	isDryRun bool, dryRunsLeft int) {

	pubCount := int64(0)
	for evList := range stream {

		evCount := len(evList)
		payloads := make([]common.MapStr, evCount)
		for i, ev := range evList {
			payloads[i] = ev.body
		}
		logp.Debug("flow", "Publishing %d events", len(evList))
		success := client.PublishEvents(payloads, publisher.Sync)
		pubCount += int64(evCount)
		ab.metrics <- &metric{"published.count", pubCount}

		for _, ev := range evList {
			if success {
				if isDryRun {
					dryRunsLeft--
					if dryRunsLeft >= 0 {
						logp.Info("Suppressing Ack due to DryRun. %d dry runs left.", dryRunsLeft)
					}
					ev.acknowledger.Nack(ev.deliveryTag, false, true)
					if dryRunsLeft == 0 {
						// Stop and let the remaining messages get nack'ed
						ab.Stop()
					}
				} else {
					logp.Debug("flow", "Acked event")
					ev.acknowledger.Ack(ev.deliveryTag, false)
				}
			} else {
				logp.Err("Failed to publish event: %v", ev.body)
				ev.acknowledger.Nack(ev.deliveryTag, false, true)
			}
		}
	}

	wg.Done()
}