mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-04 22:36:34 +00:00
179 lines
4.4 KiB
Go
179 lines
4.4 KiB
Go
|
// Copyright 2013 The Prometheus Authors
|
||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
// you may not use this file except in compliance with the License.
|
||
|
// You may obtain a copy of the License at
|
||
|
//
|
||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
//
|
||
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
// See the License for the specific language governing permissions and
|
||
|
// limitations under the License.
|
||
|
|
||
|
package model
|
||
|
|
||
|
import (
|
||
|
"encoding/json"
|
||
|
"fmt"
|
||
|
"strconv"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
type FloatString float64
|
||
|
|
||
|
func (v FloatString) String() string {
|
||
|
return strconv.FormatFloat(float64(v), 'f', -1, 64)
|
||
|
}
|
||
|
|
||
|
func (v FloatString) MarshalJSON() ([]byte, error) {
|
||
|
return json.Marshal(v.String())
|
||
|
}
|
||
|
|
||
|
func (v *FloatString) UnmarshalJSON(b []byte) error {
|
||
|
if len(b) < 2 || b[0] != '"' || b[len(b)-1] != '"' {
|
||
|
return fmt.Errorf("float value must be a quoted string")
|
||
|
}
|
||
|
f, err := strconv.ParseFloat(string(b[1:len(b)-1]), 64)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
*v = FloatString(f)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
type HistogramBucket struct {
|
||
|
Boundaries int32
|
||
|
Lower FloatString
|
||
|
Upper FloatString
|
||
|
Count FloatString
|
||
|
}
|
||
|
|
||
|
func (s HistogramBucket) MarshalJSON() ([]byte, error) {
|
||
|
b, err := json.Marshal(s.Boundaries)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
l, err := json.Marshal(s.Lower)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
u, err := json.Marshal(s.Upper)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
c, err := json.Marshal(s.Count)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return []byte(fmt.Sprintf("[%s,%s,%s,%s]", b, l, u, c)), nil
|
||
|
}
|
||
|
|
||
|
func (s *HistogramBucket) UnmarshalJSON(buf []byte) error {
|
||
|
tmp := []interface{}{&s.Boundaries, &s.Lower, &s.Upper, &s.Count}
|
||
|
wantLen := len(tmp)
|
||
|
if err := json.Unmarshal(buf, &tmp); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if gotLen := len(tmp); gotLen != wantLen {
|
||
|
return fmt.Errorf("wrong number of fields: %d != %d", gotLen, wantLen)
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (s *HistogramBucket) Equal(o *HistogramBucket) bool {
|
||
|
return s == o || (s.Boundaries == o.Boundaries && s.Lower == o.Lower && s.Upper == o.Upper && s.Count == o.Count)
|
||
|
}
|
||
|
|
||
|
func (b HistogramBucket) String() string {
|
||
|
var sb strings.Builder
|
||
|
lowerInclusive := b.Boundaries == 1 || b.Boundaries == 3
|
||
|
upperInclusive := b.Boundaries == 0 || b.Boundaries == 3
|
||
|
if lowerInclusive {
|
||
|
sb.WriteRune('[')
|
||
|
} else {
|
||
|
sb.WriteRune('(')
|
||
|
}
|
||
|
fmt.Fprintf(&sb, "%g,%g", b.Lower, b.Upper)
|
||
|
if upperInclusive {
|
||
|
sb.WriteRune(']')
|
||
|
} else {
|
||
|
sb.WriteRune(')')
|
||
|
}
|
||
|
fmt.Fprintf(&sb, ":%v", b.Count)
|
||
|
return sb.String()
|
||
|
}
|
||
|
|
||
|
type HistogramBuckets []*HistogramBucket
|
||
|
|
||
|
func (s HistogramBuckets) Equal(o HistogramBuckets) bool {
|
||
|
if len(s) != len(o) {
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
for i, bucket := range s {
|
||
|
if !bucket.Equal(o[i]) {
|
||
|
return false
|
||
|
}
|
||
|
}
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
type SampleHistogram struct {
|
||
|
Count FloatString `json:"count"`
|
||
|
Sum FloatString `json:"sum"`
|
||
|
Buckets HistogramBuckets `json:"buckets"`
|
||
|
}
|
||
|
|
||
|
func (s SampleHistogram) String() string {
|
||
|
return fmt.Sprintf("Count: %f, Sum: %f, Buckets: %v", s.Count, s.Sum, s.Buckets)
|
||
|
}
|
||
|
|
||
|
func (s *SampleHistogram) Equal(o *SampleHistogram) bool {
|
||
|
return s == o || (s.Count == o.Count && s.Sum == o.Sum && s.Buckets.Equal(o.Buckets))
|
||
|
}
|
||
|
|
||
|
type SampleHistogramPair struct {
|
||
|
Timestamp Time
|
||
|
// Histogram should never be nil, it's only stored as pointer for efficiency.
|
||
|
Histogram *SampleHistogram
|
||
|
}
|
||
|
|
||
|
func (s SampleHistogramPair) MarshalJSON() ([]byte, error) {
|
||
|
if s.Histogram == nil {
|
||
|
return nil, fmt.Errorf("histogram is nil")
|
||
|
}
|
||
|
t, err := json.Marshal(s.Timestamp)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
v, err := json.Marshal(s.Histogram)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return []byte(fmt.Sprintf("[%s,%s]", t, v)), nil
|
||
|
}
|
||
|
|
||
|
func (s *SampleHistogramPair) UnmarshalJSON(buf []byte) error {
|
||
|
tmp := []interface{}{&s.Timestamp, &s.Histogram}
|
||
|
wantLen := len(tmp)
|
||
|
if err := json.Unmarshal(buf, &tmp); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if gotLen := len(tmp); gotLen != wantLen {
|
||
|
return fmt.Errorf("wrong number of fields: %d != %d", gotLen, wantLen)
|
||
|
}
|
||
|
if s.Histogram == nil {
|
||
|
return fmt.Errorf("histogram is null")
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (s SampleHistogramPair) String() string {
|
||
|
return fmt.Sprintf("%s @[%s]", s.Histogram, s.Timestamp)
|
||
|
}
|
||
|
|
||
|
func (s *SampleHistogramPair) Equal(o *SampleHistogramPair) bool {
|
||
|
return s == o || (s.Histogram.Equal(o.Histogram) && s.Timestamp.Equal(o.Timestamp))
|
||
|
}
|