blob: 732db051dab6fa4bff4722ce08bd142a1a020ba4 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
package document
import (
"time"
)
type Status int
const (
// StatusSuccess indicates a successful document operation.
StatusSuccess Status = iota
// StatusConditionNotMet indicates that the document operation itself was successful, but did not satisfy its
// test-and-set condition.
StatusConditionNotMet
// StatusVespaFailure indicates that Vespa failed to process the document operation.
StatusVespaFailure
// StatusTransportFailure indicates that there was failure in the transport layer error while sending the document
// operation to Vespa.
StatusTransportFailure
// StatusError is a catch-all status for any other error that might occur.
StatusError
)
// Result represents the result of a feeding operation.
type Result struct {
Id Id
Status Status
Message string
Trace string
Err error
}
// Success returns whether status s is considered a success.
func (s Status) Success() bool { return s == StatusSuccess || s == StatusConditionNotMet }
// Stats represents the summed statistics of a feeder.
type Stats struct {
Requests int64
Responses int64
ResponsesByCode map[int]int64
Errors int64
Inflight int64
TotalLatency time.Duration
MinLatency time.Duration
MaxLatency time.Duration
BytesSent int64
BytesRecv int64
}
func NewStats() Stats { return Stats{ResponsesByCode: make(map[int]int64)} }
// AvgLatency returns the average latency for a request.
func (s Stats) AvgLatency() time.Duration {
requests := s.Requests
if requests == 0 {
requests = 1
}
return s.TotalLatency / time.Duration(requests)
}
func (s Stats) Successes() int64 {
if s.ResponsesByCode == nil {
return 0
}
return s.ResponsesByCode[200]
}
// Add adds all statistics contained in other to this.
func (s *Stats) Add(other Stats) {
s.Requests += other.Requests
s.Responses += other.Responses
for code, count := range other.ResponsesByCode {
_, ok := s.ResponsesByCode[code]
if ok {
s.ResponsesByCode[code] += count
} else {
s.ResponsesByCode[code] = count
}
}
s.Errors += other.Errors
s.Inflight += other.Inflight
s.TotalLatency += other.TotalLatency
if s.MinLatency == 0 || other.MinLatency < s.MinLatency {
s.MinLatency = other.MinLatency
}
if other.MaxLatency > s.MaxLatency {
s.MaxLatency = other.MaxLatency
}
s.BytesSent += other.BytesSent
s.BytesRecv += other.BytesRecv
}
// Feeder is the interface for a consumer of documents.
type Feeder interface {
Send(Document) Result
Stats() Stats
}
|