func New(host string, config *Config, logger *gosteno.Logger) *Loggregator {
cfcomponent.Logger = logger
keepAliveInterval := 30 * time.Second
listener, incomingLogChan := agentlistener.NewAgentListener(fmt.Sprintf("%s:%d", host, config.IncomingPort), logger)
unmarshaller, messageChan := unmarshaller.NewLogMessageUnmarshaller(config.SharedSecret, incomingLogChan)
blacklist := blacklist.New(config.BlackListIps)
sinkManager, appStoreInputChan := sinkmanager.NewSinkManager(config.MaxRetainedLogMessages, config.SkipCertVerify, blacklist, logger)
workerPool := workerpool.NewWorkerPool(config.EtcdMaxConcurrentRequests)
storeAdapter := etcdstoreadapter.NewETCDStoreAdapter(config.EtcdUrls, workerPool)
appStoreCache := cache.NewAppServiceCache()
appStoreWatcher, newAppServiceChan, deletedAppServiceChan := store.NewAppServiceStoreWatcher(storeAdapter, appStoreCache)
appStore := store.NewAppServiceStore(storeAdapter, appStoreWatcher)
return &Loggregator{
Logger: logger,
listener: listener,
unmarshaller: unmarshaller,
sinkManager: sinkManager,
messageChan: messageChan,
appStoreInputChan: appStoreInputChan,
appStore: appStore,
messageRouter: sinkserver.NewMessageRouter(sinkManager, logger),
websocketServer: websocketserver.New(fmt.Sprintf("%s:%d", host, config.OutgoingPort), sinkManager, keepAliveInterval, config.WSMessageBufferSize, logger),
newAppServiceChan: newAppServiceChan,
deletedAppServiceChan: deletedAppServiceChan,
appStoreWatcher: appStoreWatcher,
storeAdapter: storeAdapter,
}
}
func init() {
dataReadChannel = make(chan *logmessage.Message)
logger := loggertesthelper.Logger()
cfcomponent.Logger = logger
newAppServiceChan := make(chan domain.AppService)
deletedAppServiceChan := make(chan domain.AppService)
emptyBlacklist := blacklist.New(nil)
sinkManager, _ = sinkmanager.NewSinkManager(1024, false, emptyBlacklist, logger)
go sinkManager.Start(newAppServiceChan, deletedAppServiceChan)
TestMessageRouter = sinkserver.NewMessageRouter(sinkManager, logger)
go TestMessageRouter.Start(dataReadChannel)
apiEndpoint := "localhost:" + SERVER_PORT
TestWebsocketServer = websocketserver.New(apiEndpoint, sinkManager, 10*time.Second, 100, loggertesthelper.Logger())
go TestWebsocketServer.Start()
timeoutApiEndpoint := "localhost:" + FAST_TIMEOUT_SERVER_PORT
FastTimeoutTestWebsocketServer := websocketserver.New(timeoutApiEndpoint, sinkManager, 10*time.Millisecond, 100, loggertesthelper.Logger())
go FastTimeoutTestWebsocketServer.Start()
blackListDataReadChannel = make(chan *logmessage.Message)
localhostBlacklist := blacklist.New([]iprange.IPRange{iprange.IPRange{Start: "127.0.0.0", End: "127.0.0.2"}})
blacklistSinkManager, _ := sinkmanager.NewSinkManager(1024, false, localhostBlacklist, logger)
go blacklistSinkManager.Start(newAppServiceChan, deletedAppServiceChan)
blacklistTestMessageRouter := sinkserver.NewMessageRouter(blacklistSinkManager, logger)
go blacklistTestMessageRouter.Start(blackListDataReadChannel)
blacklistApiEndpoint := "localhost:" + BLACKLIST_SERVER_PORT
blackListTestWebsocketServer = websocketserver.New(blacklistApiEndpoint, blacklistSinkManager, 10*time.Second, 100, loggertesthelper.Logger())
go blackListTestWebsocketServer.Start()
time.Sleep(5 * time.Millisecond)
}
)
var _ = Describe("WebsocketServer", func() {
var server *websocketserver.WebsocketServer
var sinkManager, _ = sinkmanager.NewSinkManager(1024, false, blacklist.New(nil), loggertesthelper.Logger())
var appId = "my-app"
var wsReceivedChan = make(chan []byte)
var connectionDropped <-chan struct{}
var apiEndpoint = "127.0.0.1:9091"
BeforeEach(func() {
logger := loggertesthelper.Logger()
cfcomponent.Logger = logger
server = websocketserver.New(apiEndpoint, sinkManager, 100*time.Millisecond, 100, logger)
go server.Start()
serverUrl := fmt.Sprintf("ws://%s/tail/?app=%s", apiEndpoint, appId)
Eventually(func() error { _, _, err := websocket.DefaultDialer.Dial(serverUrl, http.Header{}); return err }).ShouldNot(HaveOccurred())
})
AfterEach(func() {
server.Stop()
})
Describe("failed connections", func() {
It("should fail without an appId", func() {
_, connectionDropped = AddWSSink(wsReceivedChan, fmt.Sprintf("ws://%s/tail/?", apiEndpoint))
Expect(connectionDropped).To(BeClosed())
})