switch to go-kit/log (#1575)

Signed-off-by: yeya24 <yb532204897@gmail.com>

1 files changed, 9 insertions, 6 deletions

diff --git a/collector/sockstat_linux.go b/collector/sockstat_linux.go
index fceb0ed..c7596c9 100644
--- a/collector/sockstat_linux.go
+++ b/collector/sockstat_linux.go
@@ -19,8 +19,9 @@ import (
         "fmt"
         "os"
 
+        "github.com/go-kit/kit/log"
+        "github.com/go-kit/kit/log/level"
         "github.com/prometheus/client_golang/prometheus"
-        "github.com/prometheus/common/log"
         "github.com/prometheus/procfs"
 )
 
@@ -31,15 +32,17 @@ const (
 // Used for calculating the total memory bytes on TCP and UDP.
 var pageSize = os.Getpagesize()
 
-type sockStatCollector struct{}
+type sockStatCollector struct {
+        logger log.Logger
+}
 
 func init() {
         registerCollector(sockStatSubsystem, defaultEnabled, NewSockStatCollector)
 }
 
 // NewSockStatCollector returns a new Collector exposing socket stats.
-func NewSockStatCollector() (Collector, error) {
-        return &sockStatCollector{}, nil
+func NewSockStatCollector(logger log.Logger) (Collector, error) {
+        return &sockStatCollector{logger}, nil
 }
 
 func (c *sockStatCollector) Update(ch chan<- prometheus.Metric) error {
@@ -53,7 +56,7 @@ func (c *sockStatCollector) Update(ch chan<- prometheus.Metric) error {
         switch {
         case err == nil:
         case os.IsNotExist(err):
-                log.Debug("IPv4 sockstat statistics not found, skipping")
+                level.Debug(c.logger).Log("msg", "IPv4 sockstat statistics not found, skipping")
         default:
                 return fmt.Errorf("failed to get IPv4 sockstat data: %w", err)
         }
@@ -62,7 +65,7 @@ func (c *sockStatCollector) Update(ch chan<- prometheus.Metric) error {
         switch {
         case err == nil:
         case os.IsNotExist(err):
-                log.Debug("IPv6 sockstat statistics not found, skipping")
+                level.Debug(c.logger).Log("msg", "IPv6 sockstat statistics not found, skipping")
         default:
                 return fmt.Errorf("failed to get IPv6 sockstat data: %w", err)
         }