Skip to content

Commit

Permalink
Pass logger as params to the new exporter function
Browse files Browse the repository at this point in the history
Signed-off-by: sowmiyamuthuraman <[email protected]>
  • Loading branch information
sowmiyamuthuraman committed Oct 17, 2019
1 parent e1d68a6 commit 0b54ba7
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 10 deletions.
16 changes: 8 additions & 8 deletions consul_exporter.go
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,7 @@ type promHTTPLogger struct {
}

func (l promHTTPLogger) Println(v ...interface{}) {
level.Error(l.logger).Log("msg", v)
level.Error(l.logger).Log("msg", fmt.Sprintf("%v", v))
}

// Exporter collects Consul stats from the given server and exports them using
Expand All @@ -131,7 +131,7 @@ type consulOpts struct {
}

// NewExporter returns an initialized Exporter.
func NewExporter(opts consulOpts, kvPrefix, kvFilter string, healthSummary bool) (*Exporter, error) {
func NewExporter(opts consulOpts, kvPrefix, kvFilter string, healthSummary bool, logger log.Logger) (*Exporter, error) {
uri := opts.uri
if !strings.Contains(uri, "://") {
uri = "http://" + uri
Expand Down Expand Up @@ -177,7 +177,7 @@ func NewExporter(opts consulOpts, kvPrefix, kvFilter string, healthSummary bool)
kvPrefix: kvPrefix,
kvFilter: regexp.MustCompile(kvFilter),
healthSummary: healthSummary,
logger: log.NewLogfmtLogger(os.Stdout),
logger: logger,
}, nil
}

Expand Down Expand Up @@ -338,7 +338,7 @@ func (e *Exporter) collectHealthSummary(ch chan<- prometheus.Metric, serviceName
func (e *Exporter) collectOneHealthSummary(ch chan<- prometheus.Metric, serviceName string) {
// See https://github.com/hashicorp/consul/issues/1096.
if strings.HasPrefix(serviceName, "/") {
level.Warn(e.logger).Log("msg", "Skipping service because it starts with a slash", "service name", serviceName)
level.Warn(e.logger).Log("msg", "Skipping service because it starts with a slash", "service_name", serviceName)
return
}
level.Debug(e.logger).Log("msg", "Fetching health summary", "serviceName", serviceName)
Expand Down Expand Up @@ -431,11 +431,11 @@ func main() {
logger := promlog.New(promlogConfig)

level.Info(logger).Log("msg", "Starting consul_exporter", "version", version.Info())
level.Info(logger).Log("msg", "Build context", version.BuildContext())
level.Info(logger).Log("build_context", version.BuildContext())

exporter, err := NewExporter(opts, *kvPrefix, *kvFilter, *healthSummary)
exporter, err := NewExporter(opts, *kvPrefix, *kvFilter, *healthSummary, logger)
if err != nil {
level.Error(logger).Log("msg", "Error creating an exporter", "err", err)
level.Error(logger).Log("msg", "Error creating the exporter", "err", err)
os.Exit(1)
}
prometheus.MustRegister(exporter)
Expand All @@ -453,7 +453,7 @@ func main() {
prometheus.DefaultGatherer,
promhttp.HandlerOpts{
ErrorLog: &promHTTPLogger{
logger: log.NewLogfmtLogger(os.Stderr),
logger: logger,
},
},
),
Expand Down
5 changes: 3 additions & 2 deletions consul_exporter_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ import (
"text/template"
"time"

"github.com/go-kit/kit/log"
consul_api "github.com/hashicorp/consul/api"
"github.com/prometheus/client_golang/prometheus/testutil"
"github.com/prometheus/common/expfmt"
Expand All @@ -39,7 +40,7 @@ func TestNewExporter(t *testing.T) {
}

for _, test := range cases {
_, err := NewExporter(consulOpts{uri: test.uri}, "", ".*", true)
_, err := NewExporter(consulOpts{uri: test.uri}, "", ".*", true, log.NewNopLogger())
if test.ok && err != nil {
t.Errorf("expected no error w/ %q, but got %q", test.uri, err)
}
Expand Down Expand Up @@ -140,7 +141,7 @@ consul_catalog_services 3
},
} {
t.Run(tc.name, func(t *testing.T) {
exporter, err := NewExporter(consulOpts{uri: addr, timeout: time.Duration(time.Second)}, "", "", true)
exporter, err := NewExporter(consulOpts{uri: addr, timeout: time.Duration(time.Second)}, "", "", true, log.NewNopLogger())
if err != nil {
t.Errorf("expected no error but got %q", err)
}
Expand Down

0 comments on commit 0b54ba7

Please sign in to comment.