// SetupHandler is shared between signalfx and here to setup listening for collectd connections.
// Will do shared basic setup like configuring request counters
func SetupHandler(ctx context.Context, name string, sink dpsink.Sink) (*web.Handler, stats.Keeper) {
metricTracking := web.RequestCounter{}
counter := &dpsink.Counter{}
collectdDecoder := JSONDecoder{
SendTo: dpsink.FromChain(sink, dpsink.NextWrap(counter)),
}
h := web.NewHandler(ctx, &collectdDecoder).Add(web.NextHTTP(metricTracking.ServeHTTP))
st := stats.ToKeeperMany(protocol.ListenerDims(name, "collectd"), &metricTracking, counter, &collectdDecoder)
return h, st
}
func setupChain(ctx context.Context, sink dpsink.Sink, name string, chainType string, getReader func(dpsink.Sink) ErrorReader) (*web.Handler, stats.Keeper) {
counter := &dpsink.Counter{}
finalSink := dpsink.FromChain(sink, dpsink.NextWrap(counter))
errReader := getReader(finalSink)
errorTracker := ErrorTrackerHandler{
reader: errReader,
}
metricTracking := web.RequestCounter{}
handler := web.NewHandler(ctx, &errorTracker).Add(web.NextHTTP(metricTracking.ServeHTTP))
st := stats.ToKeeperMany(protocol.ListenerDims(name, "sfx_"+chainType), &metricTracking, &errorTracker, counter)
return handler, st
}
func setupNotFoundHandler(r *mux.Router, ctx context.Context, name string) stats.Keeper {
metricTracking := web.RequestCounter{}
r.NotFoundHandler = web.NewHandler(ctx, web.FromHTTP(http.NotFoundHandler())).Add(web.NextHTTP(metricTracking.ServeHTTP))
return stats.ToKeeperMany(map[string]string{"location": "listener", "name": name, "type": "http404"}, &metricTracking)
}