storj/cmd/statreceiver/parser.go
Egon Elbre 6615ecc9b6 common: separate repository
Change-Id: Ibb89c42060450e3839481a7e495bbe3ad940610a
2019-12-27 14:11:15 +02:00

68 lines
1.2 KiB
Go

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package main
import (
"log"
"sync"
"time"
"github.com/zeebo/admission/admproto"
"storj.io/common/memory"
)
// Parser is a PacketDest that sends data to a MetricDest
type Parser struct {
dest MetricDest
scratch sync.Pool
}
// NewParser creates a Parser. It sends metrics to dest.
func NewParser(dest MetricDest) *Parser {
return &Parser{
dest: dest,
scratch: sync.Pool{
New: func() interface{} {
var x [10 * memory.KB]byte
return &x
},
},
}
}
// Packet implements PacketDest
func (p *Parser) Packet(data []byte, ts time.Time) (err error) {
data, err = admproto.CheckChecksum(data)
if err != nil {
return err
}
scratch := p.scratch.Get().(*[10 * memory.KB]byte)
defer p.scratch.Put(scratch)
r := admproto.NewReaderWith((*scratch)[:])
data, appb, instb, err := r.Begin(data)
if err != nil {
return err
}
app, inst := string(appb), string(instb)
var key []byte
var value float64
for len(data) > 0 {
data, key, value, err = r.Next(data)
if err != nil {
return err
}
err = p.dest.Metric(app, inst, key, value, ts)
if err != nil {
log.Printf("failed to write metric: %v", err)
continue
}
}
return nil
}