OXIESEC PANEL

Name	Size	Modified	Perms	Actions
📁 ..	-	12/05/2024 09:52:20 PM	rwxr-xr-x
📁 .circleci	-	12/05/2024 09:52:20 PM	r-xr-xr-x
📄 .gitignore	273 bytes	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 .travis.yml	107 bytes	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 LICENSE	1.05 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 README.md	3.49 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📁 circonus	-	12/05/2024 09:52:20 PM	r-xr-xr-x
📄 const_unix.go	153 bytes	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 const_windows.go	196 bytes	12/05/2024 09:52:20 PM	r--r--r--	Delete
📁 datadog	-	12/05/2024 09:52:20 PM	r-xr-xr-x
📄 go.mod	607 bytes	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 go.sum	11.68 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 inmem.go	8.91 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 inmem_endpoint.go	4.02 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 inmem_endpoint_test.go	8.21 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 inmem_signal.go	2.76 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 inmem_signal_test.go	1.78 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 inmem_test.go	4.45 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 metrics.go	7.58 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 metrics_test.go	12.26 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📁 prometheus	-	12/05/2024 09:52:20 PM	r-xr-xr-x
📄 sink.go	3.74 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 sink_test.go	6.29 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 start.go	5.07 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 start_test.go	5.78 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 statsd.go	4.03 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 statsd_test.go	3.87 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 statsite.go	3.93 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete
📄 statsite_test.go	3.74 KB	12/05/2024 09:52:20 PM	r--r--r--	Delete

Editing: statsite.go

package metrics

import (
	"bufio"
	"fmt"
	"log"
	"net"
	"net/url"
	"strings"
	"time"
)

const (
	// We force flush the statsite metrics after this period of
	// inactivity. Prevents stats from getting stuck in a buffer
	// forever.
	flushInterval = 100 * time.Millisecond
)

// NewStatsiteSinkFromURL creates an StatsiteSink from a URL. It is used
// (and tested) from NewMetricSinkFromURL.
func NewStatsiteSinkFromURL(u *url.URL) (MetricSink, error) {
	return NewStatsiteSink(u.Host)
}

// StatsiteSink provides a MetricSink that can be used with a
// statsite metrics server
type StatsiteSink struct {
	addr        string
	metricQueue chan string
}

// NewStatsiteSink is used to create a new StatsiteSink
func NewStatsiteSink(addr string) (*StatsiteSink, error) {
	s := &StatsiteSink{
		addr:        addr,
		metricQueue: make(chan string, 4096),
	}
	go s.flushMetrics()
	return s, nil
}

// Close is used to stop flushing to statsite
func (s *StatsiteSink) Shutdown() {
	close(s.metricQueue)
}

func (s *StatsiteSink) SetGauge(key []string, val float32) {
	flatKey := s.flattenKey(key)
	s.pushMetric(fmt.Sprintf("%s:%f|g\n", flatKey, val))
}

func (s *StatsiteSink) SetGaugeWithLabels(key []string, val float32, labels []Label) {
	flatKey := s.flattenKeyLabels(key, labels)
	s.pushMetric(fmt.Sprintf("%s:%f|g\n", flatKey, val))
}

func (s *StatsiteSink) EmitKey(key []string, val float32) {
	flatKey := s.flattenKey(key)
	s.pushMetric(fmt.Sprintf("%s:%f|kv\n", flatKey, val))
}

func (s *StatsiteSink) IncrCounter(key []string, val float32) {
	flatKey := s.flattenKey(key)
	s.pushMetric(fmt.Sprintf("%s:%f|c\n", flatKey, val))
}

func (s *StatsiteSink) IncrCounterWithLabels(key []string, val float32, labels []Label) {
	flatKey := s.flattenKeyLabels(key, labels)
	s.pushMetric(fmt.Sprintf("%s:%f|c\n", flatKey, val))
}

func (s *StatsiteSink) AddSample(key []string, val float32) {
	flatKey := s.flattenKey(key)
	s.pushMetric(fmt.Sprintf("%s:%f|ms\n", flatKey, val))
}

func (s *StatsiteSink) AddSampleWithLabels(key []string, val float32, labels []Label) {
	flatKey := s.flattenKeyLabels(key, labels)
	s.pushMetric(fmt.Sprintf("%s:%f|ms\n", flatKey, val))
}

// Flattens the key for formatting, removes spaces
func (s *StatsiteSink) flattenKey(parts []string) string {
	joined := strings.Join(parts, ".")
	return strings.Map(func(r rune) rune {
		switch r {
		case ':':
			fallthrough
		case ' ':
			return '_'
		default:
			return r
		}
	}, joined)
}

// Flattens the key along with labels for formatting, removes spaces
func (s *StatsiteSink) flattenKeyLabels(parts []string, labels []Label) string {
	for _, label := range labels {
		parts = append(parts, label.Value)
	}
	return s.flattenKey(parts)
}

// Does a non-blocking push to the metrics queue
func (s *StatsiteSink) pushMetric(m string) {
	select {
	case s.metricQueue <- m:
	default:
	}
}

// Flushes metrics
func (s *StatsiteSink) flushMetrics() {
	var sock net.Conn
	var err error
	var wait <-chan time.Time
	var buffered *bufio.Writer
	ticker := time.NewTicker(flushInterval)
	defer ticker.Stop()

CONNECT:
	// Attempt to connect
	sock, err = net.Dial("tcp", s.addr)
	if err != nil {
		log.Printf("[ERR] Error connecting to statsite! Err: %s", err)
		goto WAIT
	}

// Create a buffered writer
	buffered = bufio.NewWriter(sock)

for {
		select {
		case metric, ok := <-s.metricQueue:
			// Get a metric from the queue
			if !ok {
				goto QUIT
			}

// Try to send to statsite
			_, err := buffered.Write([]byte(metric))
			if err != nil {
				log.Printf("[ERR] Error writing to statsite! Err: %s", err)
				goto WAIT
			}
		case <-ticker.C:
			if err := buffered.Flush(); err != nil {
				log.Printf("[ERR] Error flushing to statsite! Err: %s", err)
				goto WAIT
			}
		}
	}

WAIT:
	// Wait for a while
	wait = time.After(time.Duration(5) * time.Second)
	for {
		select {
		// Dequeue the messages to avoid backlog
		case _, ok := <-s.metricQueue:
			if !ok {
				goto QUIT
			}
		case <-wait:
			goto CONNECT
		}
	}
QUIT:
	s.metricQueue = nil
}