qb/proto/proto.go

package proto

import (
	"fmt"
	"time"

	bin "gordenko.dev/dima/bin/little"
	"gordenko.dev/dima/qb"
	"gordenko.dev/dima/qb/bufreader"
)

const (
	TypeDeleteMeasures         byte = 1
	TypeListCurrentValues      byte = 2
	TypeListInstantMeasures    byte = 3
	TypeListCumulativeMeasures byte = 33

	TypeListInstantPeriods    byte = 4
	TypeListCumulativePeriods byte = 44
	TypeGetMetric             byte = 5
	TypeAddMetric             byte = 6

	TypeListAllInstantMeasures    byte = 8
	TypeListAllCumulativeMeasures byte = 88
	TypeAppendMeasure             byte = 10
	TypeAppendMeasures            byte = 11
	TypeDeleteMetric              byte = 12

	RespPartOfValue byte = 255
	RespEndOfValue  byte = 254
	RespError       byte = 253
	RespSuccess     byte = 252
	RespValue       byte = 251

	ErrNoMetric          = 1
	ErrDuplicate         = 2
	ErrWrongMetricType   = 3
	ErrWrongFracDigits   = 4
	ErrExpiredMeasure    = 5
	ErrNonMonotonicValue = 6
	ErrEmptyMetricID     = 7
	ErrInvalidRange      = 8
	ErrUnexpected        = 9
)

func ErrorCodeToText(code uint16) string {
	switch code {
	case ErrNoMetric:
		return "NoMetric"
	case ErrDuplicate:
		return "Duplicate"
	case ErrWrongMetricType:
		return "WrongMetricType"
	case ErrWrongFracDigits:
		return "WrongFracDigits"
	case ErrExpiredMeasure:
		return "ExpiredMeasure"
	case ErrNonMonotonicValue:
		return "NonMonotonicValue"
	case ErrEmptyMetricID:
		return "EmptyMetricID"
	case ErrInvalidRange:
		return "InvalidRange"
	case ErrUnexpected:
		return "Unexpected"
	default:
		return ""
	}
}

// common

type Metric struct {
	MetricID   uint32
	MetricType qb.MetricType
	FracDigits int
}

type AppendError struct {
	MetricID  uint32
	ErrorCode uint16
}

type TimeBound struct {
	Year  int
	Month time.Month
	Day   int
}

type CumulativeMeasure struct {
	Timestamp uint32
	Value     float64
	Total     float64
}

type CumulativePeriod struct {
	Period   uint32
	Since    uint32
	Until    uint32
	EndValue float64
	Total    float64
}

type InstantMeasure struct {
	Timestamp uint32
	Value     float64
}

type InstantPeriod struct {
	Period uint32
	Since  uint32
	Until  uint32
	Min    float64
	Max    float64
	Avg    float64
}

type CurrentValue struct {
	MetricID  uint32
	Timestamp uint32
	Value     float64
}

// API reqs

type GetMetricReq struct {
	MetricID uint32
}

func ReadGetMetricReq(r *bufreader.BufferedReader) (m GetMetricReq, err error) {
	m.MetricID, err = bin.ReadUint32(r)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return
}

type ListCurrentValuesReq struct {
	MetricIDs []uint32
}

func ReadListCurrentValuesReq(r *bufreader.BufferedReader) (m ListCurrentValuesReq, err error) {
	qty, err := bin.ReadUint16(r)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}

	for i := range int(qty) {
		var metricID uint32
		metricID, err = bin.ReadUint32(r)
		if err != nil {
			err = fmt.Errorf("read metricID (#%d): %s", i, err)
			return
		}
		m.MetricIDs = append(m.MetricIDs, metricID)
	}
	return
}

type AddMetricReq struct {
	MetricID   uint32
	MetricType qb.MetricType
	FracDigits byte
}

func ReadAddMetricReq(r *bufreader.BufferedReader) (m AddMetricReq, err error) {
	arr, err := r.ReadN(6)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return UnpackAddMetricReq(arr), nil
}

func UnpackAddMetricReq(arr []byte) (m AddMetricReq) {
	m.MetricID, _ = bin.GetUint32(arr)
	m.MetricType = qb.MetricType(arr[4])
	m.FracDigits = arr[5]
	return
}

type DeleteMetricReq struct {
	MetricID uint32
}

func ReadDeleteMetricReq(r *bufreader.BufferedReader) (m DeleteMetricReq, err error) {
	m.MetricID, err = bin.ReadUint32(r)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return
}

type DeleteMeasuresReq struct {
	MetricID uint32
	Since    uint32 // timestamp (optional)
}

func ReadDeleteMeasuresReq(r *bufreader.BufferedReader) (m DeleteMeasuresReq, err error) {
	arr, err := r.ReadN(8)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return UnpackDeleteMeasuresReq(arr), nil
}

func UnpackDeleteMeasuresReq(arr []byte) (m DeleteMeasuresReq) {
	m.MetricID, _ = bin.GetUint32(arr)
	m.Since, _ = bin.GetUint32(arr[4:])
	return
}

type AppendMeasureReq struct {
	MetricID  uint32
	Timestamp uint32
	Value     float64
}

func ReadAppendMeasureReq(r *bufreader.BufferedReader) (m AppendMeasureReq, err error) {
	arr, err := r.ReadN(16)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return UnpackAppendMeasureReq(arr), nil
}

func UnpackAppendMeasureReq(arr []byte) (m AppendMeasureReq) {
	m.MetricID, _ = bin.GetUint32(arr)
	m.Timestamp, _ = bin.GetUint32(arr[4:])
	m.Value, _ = bin.GetFloat64(arr[8:])
	return
}

type AppendMeasuresReq struct {
	MetricID uint32
	Measures []Measure
}

type Measure struct {
	Timestamp uint32
	Value     float64
}

func ReadAppendMeasuresReq(r *bufreader.BufferedReader) (m AppendMeasuresReq, err error) {
	prefix, err := bin.ReadN(r, 6) // metricID + measures qty
	if err != nil {
		err = fmt.Errorf("read prefix: %s", err)
		return
	}

	m.MetricID, _ = bin.GetUint32(prefix[0:])
	qty, _ := bin.GetUint16(prefix[4:])

	for i := range int(qty) {
		var measure Measure
		measure.Timestamp, err = bin.ReadUint32(r)
		if err != nil {
			err = fmt.Errorf("read timestamp (#%d): %s", i, err)
			return
		}
		measure.Value, err = bin.ReadFloat64(r)
		if err != nil {
			err = fmt.Errorf("read value (#%d): %s", i, err)
			return
		}
		m.Measures = append(m.Measures, measure)
	}
	return
}

type MetricMeasure struct {
	MetricID  uint32
	Timestamp uint32
	Value     float64
}

func ReadAppendMeasurePerMetricReq(r *bufreader.BufferedReader) (measures []MetricMeasure, err error) {
	count, err := bin.ReadUint16(r)
	if err != nil {
		return
	}
	for range count {
		var m MetricMeasure
		m.MetricID, err = bin.ReadUint32(r)
		if err != nil {
			return
		}
		m.Timestamp, err = bin.ReadUint32(r)
		if err != nil {
			return
		}
		m.Value, err = bin.ReadFloat64(r)
		if err != nil {
			return
		}
		measures = append(measures, m)
	}
	return
}

type ListAllInstantMetricMeasuresReq struct {
	MetricID uint32
}

func ReadListAllInstantMeasuresReq(r *bufreader.BufferedReader) (m ListAllInstantMetricMeasuresReq, err error) {
	m.MetricID, err = bin.ReadUint32(r)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return
}

type ListAllCumulativeMeasuresReq struct {
	MetricID uint32
}

func ReadListAllCumulativeMeasuresReq(r *bufreader.BufferedReader) (m ListAllCumulativeMeasuresReq, err error) {
	m.MetricID, err = bin.ReadUint32(r)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return
}

type ListInstantMeasuresReq struct {
	MetricID uint32
	Since    uint32
	Until    uint32
}

func ReadListInstantMeasuresReq(r *bufreader.BufferedReader) (m ListInstantMeasuresReq, err error) {
	arr, err := r.ReadN(12)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return UnpackListInstantMeasuresReq(arr), nil
}

func UnpackListInstantMeasuresReq(arr []byte) (m ListInstantMeasuresReq) {
	m.MetricID, _ = bin.GetUint32(arr[0:])
	m.Since, _ = bin.GetUint32(arr[4:])
	m.Until, _ = bin.GetUint32(arr[8:])
	return
}

type ListCumulativeMeasuresReq struct {
	MetricID uint32
	Since    uint32
	Until    uint32
}

func ReadListCumulativeMeasuresReq(r *bufreader.BufferedReader) (m ListCumulativeMeasuresReq, err error) {
	arr, err := r.ReadN(12)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return UnpackListCumulativeMeasuresReq(arr), nil
}

func UnpackListCumulativeMeasuresReq(arr []byte) (m ListCumulativeMeasuresReq) {
	m.MetricID, _ = bin.GetUint32(arr)
	m.Since, _ = bin.GetUint32(arr[4:])
	m.Until, _ = bin.GetUint32(arr[8:])
	return
}

type ListInstantPeriodsReq struct {
	MetricID       uint32
	Since          TimeBound
	Until          TimeBound
	GroupBy        qb.GroupBy
	AggregateFuncs byte
	FirstHourOfDay int
}

func ReadListInstantPeriodsReq(r *bufreader.BufferedReader) (m ListInstantPeriodsReq, err error) {
	arr, err := r.ReadN(15)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return UnpackListInstantPeriodsReq(arr), nil
}

func UnpackListInstantPeriodsReq(arr []byte) (m ListInstantPeriodsReq) {
	m.MetricID, _ = bin.GetUint32(arr)
	sinceYear, _ := bin.GetUint16(arr[4:])
	untilYear, _ := bin.GetUint16(arr[8:])
	m.Since = TimeBound{
		Year:  int(sinceYear),
		Month: time.Month(arr[6]),
		Day:   int(arr[7]),
	}
	m.Until = TimeBound{
		Year:  int(untilYear),
		Month: time.Month(arr[10]),
		Day:   int(arr[11]),
	}
	m.GroupBy = qb.GroupBy(arr[12])
	m.AggregateFuncs = arr[13]
	m.FirstHourOfDay = int(arr[14])
	return
}

type ListCumulativePeriodsReq struct {
	MetricID       uint32
	Since          TimeBound
	Until          TimeBound
	GroupBy        qb.GroupBy
	FirstHourOfDay int
}

func ReadListCumulativePeriodsReq(r *bufreader.BufferedReader) (m ListCumulativePeriodsReq, err error) {
	arr, err := r.ReadN(14)
	if err != nil {
		err = fmt.Errorf("read req: %s", err)
		return
	}
	return UnpackListCumulativePeriodsReq(arr), nil
}

func UnpackListCumulativePeriodsReq(arr []byte) (m ListCumulativePeriodsReq) {
	m.MetricID, _ = bin.GetUint32(arr[0:])
	sinceYear, _ := bin.GetUint16(arr[4:])
	untilYear, _ := bin.GetUint16(arr[8:])
	m.Since = TimeBound{
		Year:  int(sinceYear),
		Month: time.Month(arr[6]),
		Day:   int(arr[7]),
	}
	m.Until = TimeBound{
		Year:  int(untilYear),
		Month: time.Month(arr[10]),
		Day:   int(arr[11]),
	}
	m.GroupBy = qb.GroupBy(arr[12])
	m.FirstHourOfDay = int(arr[13])
	return
}