mirror of
https://github.com/ClusterCockpit/cc-metric-store.git
synced 2024-09-20 15:37:28 +02:00
135 lines
3.4 KiB
Go
135 lines
3.4 KiB
Go
|
package checkpoints
|
||
|
|
||
|
import (
|
||
|
"encoding/binary"
|
||
|
"io"
|
||
|
"reflect"
|
||
|
"unsafe"
|
||
|
|
||
|
"github.com/ClusterCockpit/cc-metric-store/internal/types"
|
||
|
)
|
||
|
|
||
|
const writeBufferSize int = 8192
|
||
|
|
||
|
// A single checkpoint file is defined by this structure.
|
||
|
type Checkpoint struct {
|
||
|
Reserved []byte // Reserved for future use/metadata.
|
||
|
From, To uint64 // Time covered by this checkpoint.
|
||
|
Root *CheckpointLevel // Root of the internal tree structure.
|
||
|
}
|
||
|
|
||
|
// A single level in the tree of the in-memory store is defined by this structure.
|
||
|
type CheckpointLevel struct {
|
||
|
Reserved []byte // Reserved for future use/metadata.
|
||
|
Metrics map[string]*Metrics // Payload...
|
||
|
Sublevels map[string]*CheckpointLevel // Child in the internal tree structure.
|
||
|
}
|
||
|
|
||
|
// Payload/Data
|
||
|
type Metrics struct {
|
||
|
Reserved []byte // Reserved for future use/metadata.
|
||
|
Frequency uint64 // Frequency of measurements for this metric.
|
||
|
From uint64 // Timestamp of the first measurement of this metric.
|
||
|
Data []types.Float // Payload...
|
||
|
}
|
||
|
|
||
|
func (c *Checkpoint) Serialize(w io.Writer) error {
|
||
|
// Write magic value (8 byte):
|
||
|
if _, err := w.Write([]byte("CCMSv1.0")); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
// The header:
|
||
|
buf := make([]byte, 0, writeBufferSize*2)
|
||
|
buf = encodeBytes(buf, c.Reserved)
|
||
|
buf = encodeUint64(buf, c.From)
|
||
|
buf = encodeUint64(buf, c.To)
|
||
|
|
||
|
// The rest:
|
||
|
buf, err := c.Root.serialize(w, buf)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if _, err := w.Write(buf); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (c *CheckpointLevel) serialize(w io.Writer, buf []byte) ([]byte, error) {
|
||
|
// The reserved data:
|
||
|
buf = encodeBytes(buf, c.Reserved)
|
||
|
|
||
|
var err error
|
||
|
// metrics:
|
||
|
buf = encodeUint32(buf, uint32(len(c.Metrics)))
|
||
|
for key, metric := range c.Metrics {
|
||
|
buf = encodeString(buf, key)
|
||
|
buf, err = metric.serialize(w, buf)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
if len(buf) >= writeBufferSize {
|
||
|
if _, err = w.Write(buf); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
buf = buf[0:]
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Sublevels:
|
||
|
buf = encodeUint32(buf, uint32(len(c.Sublevels)))
|
||
|
for key, sublevel := range c.Sublevels {
|
||
|
buf = encodeString(buf, key)
|
||
|
buf, err = sublevel.serialize(w, buf)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return buf, nil
|
||
|
}
|
||
|
|
||
|
func (m *Metrics) serialize(w io.Writer, buf []byte) ([]byte, error) {
|
||
|
buf = encodeBytes(m.Reserved, buf) // Reserved
|
||
|
buf = encodeUint64(buf, m.Frequency) // Frequency
|
||
|
buf = encodeUint64(buf, m.From) // First measurmenet timestamp
|
||
|
buf = encodeUint32(buf, uint32(len(m.Data))) // Number of elements
|
||
|
|
||
|
var x types.Float
|
||
|
elmsize := unsafe.Sizeof(x)
|
||
|
sh := (*reflect.SliceHeader)(unsafe.Pointer(&m.Data))
|
||
|
bytes := unsafe.Slice((*byte)(unsafe.Pointer(sh.Data)), sh.Len*int(elmsize))
|
||
|
|
||
|
buf = append(buf, bytes...)
|
||
|
return buf, nil
|
||
|
}
|
||
|
|
||
|
func encodeBytes(buf []byte, bytes []byte) []byte {
|
||
|
buf = encodeUint32(buf, uint32(len(bytes)))
|
||
|
buf = append(buf, bytes...)
|
||
|
return buf
|
||
|
}
|
||
|
|
||
|
func encodeString(buf []byte, str string) []byte {
|
||
|
buf = encodeUint32(buf, uint32(len(str)))
|
||
|
buf = append(buf, str...)
|
||
|
return buf
|
||
|
}
|
||
|
|
||
|
func encodeUint64(buf []byte, i uint64) []byte {
|
||
|
buf = append(buf, 0, 0, 0, 0, 0, 0, 0, 0) // 8 bytes
|
||
|
binary.LittleEndian.PutUint64(buf, i)
|
||
|
return buf[8:]
|
||
|
}
|
||
|
|
||
|
func encodeUint32(buf []byte, i uint32) []byte {
|
||
|
buf = append(buf, 0, 0, 0, 0) // 4 bytes
|
||
|
binary.LittleEndian.PutUint32(buf, i)
|
||
|
return buf[4:]
|
||
|
}
|