func Initialize() {
http.DefaultTransport = &http.Transport{Proxy: http.ProxyFromEnvironment}
autowiredEmitter = nil
origin := os.Getenv("DROPSONDE_ORIGIN")
if len(origin) == 0 {
log.Println("Failed to auto-initialize dropsonde: DROPSONDE_ORIGIN environment variable not set")
return
}
destination = os.Getenv("DROPSONDE_DESTINATION")
if len(destination) == 0 {
log.Println("DROPSONDE_DESTINATION not set. Using " + defaultDestination)
destination = defaultDestination
}
udpEmitter, err := emitter.NewUdpEmitter(destination)
if err != nil {
log.Printf("Failed to auto-initialize dropsonde: %v\n", err)
return
}
hbEmitter, err := emitter.NewHeartbeatEmitter(udpEmitter, origin)
if err != nil {
log.Printf("Failed to auto-initialize dropsonde: %v\n", err)
return
}
autowiredEmitter = emitter.NewEventEmitter(hbEmitter, origin)
http.DefaultTransport = InstrumentedRoundTripper(http.DefaultTransport)
}
package emitter_test
import (
"github.com/cloudfoundry-incubator/dropsonde/emitter"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"net"
)
var _ = Describe("UdpEmitter", func() {
var testData = []byte("hello")
Describe("Close()", func() {
It("closes the UDP connection", func() {
udpEmitter, _ := emitter.NewUdpEmitter("localhost:42420")
udpEmitter.Close()
err := udpEmitter.Emit(testData)
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(Equal("use of closed network connection"))
})
})
Describe("Emit()", func() {
var udpEmitter emitter.ByteEmitter
Context("when the agent is listening", func() {
var agentListener net.PacketConn