Esempio n. 1
0
func main() {
	log.Printf("Agent starting, use CTL-C to quit\n")

	defer log.Printf("Agent stopped\n")

	var debug bool
	flag.BoolVar(&debug, "debug", false, "Enable debug logging.")
	flag.Parse()

	if debug {
		emitter.DefaultEmitter = emitter.NewLoggingEmitter()
	}

	stopChan := make(chan struct{})

	go func() {
		err := agent.Run(stopChan)
		if err != nil {
			log.Fatalf("failed to run agent: %v", err)
		}
	}()

	killChan := make(chan os.Signal, 2)
	signal.Notify(killChan, syscall.SIGINT, syscall.SIGTERM)

	for {
		select {
		case <-RegisterGoRoutineDumpSignalChannel():
			DumpGoRoutine()
		case <-killChan:
			close(stopChan)
			return
		}
	}
}
		BeforeEach(func() {
			agent.UdpListeningPort.Set(0)
			agent.TcpListeningPort.Set(0)

			dataChan = make(chan []byte, 10)
			fakeEmitter = &FakeEmitter{dataChan}
			emitter.DefaultEmitter = fakeEmitter

			stopChan = make(chan struct{})
			errChan = make(chan error)
		})

		It("checks that there is a default emitter", func(done Done) {
			defer close(done)
			emitter.DefaultEmitter = nil
			err := agent.Run(nil)
			Expect(err).To(HaveOccurred())
			Expect(err.Error()).To(Equal("Could not start agent. No default emitter provided."))
		})

		It("listens for UDP packets and emits them", func(done Done) {
			defer close(done)

			go func() {
				defer close(errChan)
				err := agent.Run(stopChan)
				errChan <- err
			}()

			Eventually(agent.UdpListeningPort.Get).ShouldNot(BeZero())