Просмотр исходного кода

Intergrate custom logger where applicable

tags/v2.0.0-rc1
9seconds 5 лет назад
Родитель
Сommit
172b596cf1
4 измененных файлов: 13 добавлений и 12 удалений
  1. 5
    0
      logger/init.go
  2. 1
    5
      mtglib/proxy.go
  3. 4
    6
      stats/statsd.go
  4. 3
    1
      stats/statsd_test.go

+ 5
- 0
logger/init.go Просмотреть файл

@@ -0,0 +1,5 @@
1
+package logger
2
+
3
+type StdLikeLogger interface {
4
+	Printf(format string, args ...interface{})
5
+}

+ 1
- 5
mtglib/proxy.go Просмотреть файл

@@ -87,10 +87,6 @@ func (p *Proxy) Shutdown() {
87 87
 	p.workerPool.Release()
88 88
 }
89 89
 
90
-type antsLogger struct{}
91
-
92
-func (a antsLogger) Printf(msg string, args ...interface{}) {}
93
-
94 90
 func NewProxy(opts ProxyOpts) (*Proxy, error) {
95 91
 	switch {
96 92
 	case opts.Network == nil:
@@ -126,7 +122,7 @@ func NewProxy(opts ProxyOpts) (*Proxy, error) {
126 122
 
127 123
 	pool, err := ants.NewPoolWithFunc(int(concurrency), func(arg interface{}) {
128 124
 		proxy.ServeConn(arg.(net.Conn))
129
-	}, ants.WithLogger(antsLogger{}))
125
+	}, ants.WithLogger(opts.Logger.Named("ants")))
130 126
 	if err != nil {
131 127
 		return nil, fmt.Errorf("cannot initialize a pool: %w", err)
132 128
 	}

+ 4
- 6
stats/statsd.go Просмотреть файл

@@ -6,14 +6,11 @@ import (
6 6
 	"time"
7 7
 
8 8
 	"github.com/9seconds/mtg/v2/events"
9
+	"github.com/9seconds/mtg/v2/logger"
9 10
 	"github.com/9seconds/mtg/v2/mtglib"
10 11
 	statsd "github.com/smira/go-statsd"
11 12
 )
12 13
 
13
-type statsdFakeLogger struct{}
14
-
15
-func (s statsdFakeLogger) Printf(msg string, args ...interface{}) {}
16
-
17 14
 type statsdProcessor struct {
18 15
 	streams map[string]*streamInfo
19 16
 	client  *statsd.Client
@@ -92,10 +89,11 @@ func (s StatsdFactory) Make() events.Observer {
92 89
 	}
93 90
 }
94 91
 
95
-func NewStatsd(address, metricPrefix, tagFormat string) (StatsdFactory, error) {
92
+func NewStatsd(address string, log logger.StdLikeLogger,
93
+	metricPrefix, tagFormat string) (StatsdFactory, error) {
96 94
 	options := []statsd.Option{
97 95
 		statsd.MetricPrefix(metricPrefix),
98
-		statsd.Logger(statsdFakeLogger{}),
96
+		statsd.Logger(log),
99 97
 	}
100 98
 
101 99
 	switch strings.ToLower(tagFormat) {

+ 3
- 1
stats/statsd_test.go Просмотреть файл

@@ -8,6 +8,7 @@ import (
8 8
 	"time"
9 9
 
10 10
 	"github.com/9seconds/mtg/v2/events"
11
+	"github.com/9seconds/mtg/v2/logger"
11 12
 	"github.com/9seconds/mtg/v2/mtglib"
12 13
 	"github.com/9seconds/mtg/v2/stats"
13 14
 	statsd "github.com/smira/go-statsd"
@@ -80,7 +81,8 @@ type StatsdTestSuite struct {
80 81
 func (suite *StatsdTestSuite) SetupTest() {
81 82
 	suite.statsdServer = statsdNewFakeServer()
82 83
 
83
-	factory, err := stats.NewStatsd(suite.statsdServer.Addr(), "mtg.", "datadog")
84
+	factory, err := stats.NewStatsd(suite.statsdServer.Addr(),
85
+		logger.NewNoopLogger(), "mtg.", "datadog")
84 86
 	if err != nil {
85 87
 		panic(err)
86 88
 	}

Загрузка…
Отмена
Сохранить