kraken

1
// Copyright (c) 2016-2019 Uber Technologies, Inc.
2
//
3
// Licensed under the Apache License, Version 2.0 (the "License");
4
// you may not use this file except in compliance with the License.
5
// You may obtain a copy of the License at
6
//
7
//     http://www.apache.org/licenses/LICENSE-2.0
8
//
9
// Unless required by applicable law or agreed to in writing, software
10
// distributed under the License is distributed on an "AS IS" BASIS,
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
// See the License for the specific language governing permissions and
13
// limitations under the License.
14
package metrics
15

16
import (
17
	"io"
18
	"time"
19

20
	"github.com/cactus/go-statsd-client/statsd"
21
	"github.com/uber-go/tally"
22
	tallystatsd "github.com/uber-go/tally/statsd"
23
)
24

25
const (
26
	flushInterval = 100 * time.Millisecond
27
	flushBytes    = 512
28
	sampleRate    = 1.0
29
)
30

31
func newStatsdScope(config Config, cluster string) (tally.Scope, io.Closer, error) {
32
	statter, err := statsd.NewBufferedClient(
33
		config.Statsd.HostPort, config.Statsd.Prefix, flushInterval, flushBytes)
34
	if err != nil {
35
		return nil, nil, err
36
	}
37
	r := tallystatsd.NewReporter(statter, tallystatsd.Options{
38
		SampleRate: sampleRate,
39
	})
40
	s, c := tally.NewRootScope(tally.ScopeOptions{
41
		Reporter: r,
42
	}, time.Second)
43
	return s, c, nil
44
}
45