mirror of https://github.com/grpc/grpc-go.git
xds: lrs load store (#2779)
This commit is contained in:
Menghan Li
GitHub
parent
4b60e3b6a1
commit
9949ee0c45
|
|
@ -0,0 +1,429 @@
|
||||||
|
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||||
|
// source: envoy/api/v2/endpoint/load_report.proto
|
||||||
|
|
||||||
|
package envoy_api_v2_endpoint
|
||||||
|
|
||||||
|
import proto "github.com/golang/protobuf/proto"
|
||||||
|
import fmt "fmt"
|
||||||
|
import math "math"
|
||||||
|
import duration "github.com/golang/protobuf/ptypes/duration"
|
||||||
|
import _struct "github.com/golang/protobuf/ptypes/struct"
|
||||||
|
import address "google.golang.org/grpc/balancer/xds/internal/proto/envoy/api/v2/core/address"
|
||||||
|
import base "google.golang.org/grpc/balancer/xds/internal/proto/envoy/api/v2/core/base"
|
||||||
|
import _ "google.golang.org/grpc/balancer/xds/internal/proto/validate"
|
||||||
|
|
||||||
|
// Reference imports to suppress errors if they are not otherwise used.
|
||||||
|
var _ = proto.Marshal
|
||||||
|
var _ = fmt.Errorf
|
||||||
|
var _ = math.Inf
|
||||||
|
|
||||||
|
// This is a compile-time assertion to ensure that this generated file
|
||||||
|
// is compatible with the proto package it is being compiled against.
|
||||||
|
// A compilation error at this line likely means your copy of the
|
||||||
|
// proto package needs to be updated.
|
||||||
|
const _ = proto.ProtoPackageIsVersion2 // please upgrade the proto package
|
||||||
|
|
||||||
|
type UpstreamLocalityStats struct {
|
||||||
|
Locality *base.Locality `protobuf:"bytes,1,opt,name=locality,proto3" json:"locality,omitempty"`
|
||||||
|
TotalSuccessfulRequests uint64 `protobuf:"varint,2,opt,name=total_successful_requests,json=totalSuccessfulRequests,proto3" json:"total_successful_requests,omitempty"`
|
||||||
|
TotalRequestsInProgress uint64 `protobuf:"varint,3,opt,name=total_requests_in_progress,json=totalRequestsInProgress,proto3" json:"total_requests_in_progress,omitempty"`
|
||||||
|
TotalErrorRequests uint64 `protobuf:"varint,4,opt,name=total_error_requests,json=totalErrorRequests,proto3" json:"total_error_requests,omitempty"`
|
||||||
|
LoadMetricStats []*EndpointLoadMetricStats `protobuf:"bytes,5,rep,name=load_metric_stats,json=loadMetricStats,proto3" json:"load_metric_stats,omitempty"`
|
||||||
|
UpstreamEndpointStats []*UpstreamEndpointStats `protobuf:"bytes,7,rep,name=upstream_endpoint_stats,json=upstreamEndpointStats,proto3" json:"upstream_endpoint_stats,omitempty"`
|
||||||
|
Priority uint32 `protobuf:"varint,6,opt,name=priority,proto3" json:"priority,omitempty"`
|
||||||
|
XXX_NoUnkeyedLiteral struct{} `json:"-"`
|
||||||
|
XXX_unrecognized []byte `json:"-"`
|
||||||
|
XXX_sizecache int32 `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamLocalityStats) Reset() { *m = UpstreamLocalityStats{} }
|
||||||
|
func (m *UpstreamLocalityStats) String() string { return proto.CompactTextString(m) }
|
||||||
|
func (*UpstreamLocalityStats) ProtoMessage() {}
|
||||||
|
func (*UpstreamLocalityStats) Descriptor() ([]byte, []int) {
|
||||||
|
return fileDescriptor_load_report_5485b60725c658b8, []int{0}
|
||||||
|
}
|
||||||
|
func (m *UpstreamLocalityStats) XXX_Unmarshal(b []byte) error {
|
||||||
|
return xxx_messageInfo_UpstreamLocalityStats.Unmarshal(m, b)
|
||||||
|
}
|
||||||
|
func (m *UpstreamLocalityStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
|
||||||
|
return xxx_messageInfo_UpstreamLocalityStats.Marshal(b, m, deterministic)
|
||||||
|
}
|
||||||
|
func (dst *UpstreamLocalityStats) XXX_Merge(src proto.Message) {
|
||||||
|
xxx_messageInfo_UpstreamLocalityStats.Merge(dst, src)
|
||||||
|
}
|
||||||
|
func (m *UpstreamLocalityStats) XXX_Size() int {
|
||||||
|
return xxx_messageInfo_UpstreamLocalityStats.Size(m)
|
||||||
|
}
|
||||||
|
func (m *UpstreamLocalityStats) XXX_DiscardUnknown() {
|
||||||
|
xxx_messageInfo_UpstreamLocalityStats.DiscardUnknown(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
var xxx_messageInfo_UpstreamLocalityStats proto.InternalMessageInfo
|
||||||
|
|
||||||
|
func (m *UpstreamLocalityStats) GetLocality() *base.Locality {
|
||||||
|
if m != nil {
|
||||||
|
return m.Locality
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamLocalityStats) GetTotalSuccessfulRequests() uint64 {
|
||||||
|
if m != nil {
|
||||||
|
return m.TotalSuccessfulRequests
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamLocalityStats) GetTotalRequestsInProgress() uint64 {
|
||||||
|
if m != nil {
|
||||||
|
return m.TotalRequestsInProgress
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamLocalityStats) GetTotalErrorRequests() uint64 {
|
||||||
|
if m != nil {
|
||||||
|
return m.TotalErrorRequests
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamLocalityStats) GetLoadMetricStats() []*EndpointLoadMetricStats {
|
||||||
|
if m != nil {
|
||||||
|
return m.LoadMetricStats
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamLocalityStats) GetUpstreamEndpointStats() []*UpstreamEndpointStats {
|
||||||
|
if m != nil {
|
||||||
|
return m.UpstreamEndpointStats
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamLocalityStats) GetPriority() uint32 {
|
||||||
|
if m != nil {
|
||||||
|
return m.Priority
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpstreamEndpointStats struct {
|
||||||
|
Address *address.Address `protobuf:"bytes,1,opt,name=address,proto3" json:"address,omitempty"`
|
||||||
|
Metadata *_struct.Struct `protobuf:"bytes,6,opt,name=metadata,proto3" json:"metadata,omitempty"`
|
||||||
|
TotalSuccessfulRequests uint64 `protobuf:"varint,2,opt,name=total_successful_requests,json=totalSuccessfulRequests,proto3" json:"total_successful_requests,omitempty"`
|
||||||
|
TotalRequestsInProgress uint64 `protobuf:"varint,3,opt,name=total_requests_in_progress,json=totalRequestsInProgress,proto3" json:"total_requests_in_progress,omitempty"`
|
||||||
|
TotalErrorRequests uint64 `protobuf:"varint,4,opt,name=total_error_requests,json=totalErrorRequests,proto3" json:"total_error_requests,omitempty"`
|
||||||
|
LoadMetricStats []*EndpointLoadMetricStats `protobuf:"bytes,5,rep,name=load_metric_stats,json=loadMetricStats,proto3" json:"load_metric_stats,omitempty"`
|
||||||
|
XXX_NoUnkeyedLiteral struct{} `json:"-"`
|
||||||
|
XXX_unrecognized []byte `json:"-"`
|
||||||
|
XXX_sizecache int32 `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamEndpointStats) Reset() { *m = UpstreamEndpointStats{} }
|
||||||
|
func (m *UpstreamEndpointStats) String() string { return proto.CompactTextString(m) }
|
||||||
|
func (*UpstreamEndpointStats) ProtoMessage() {}
|
||||||
|
func (*UpstreamEndpointStats) Descriptor() ([]byte, []int) {
|
||||||
|
return fileDescriptor_load_report_5485b60725c658b8, []int{1}
|
||||||
|
}
|
||||||
|
func (m *UpstreamEndpointStats) XXX_Unmarshal(b []byte) error {
|
||||||
|
return xxx_messageInfo_UpstreamEndpointStats.Unmarshal(m, b)
|
||||||
|
}
|
||||||
|
func (m *UpstreamEndpointStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
|
||||||
|
return xxx_messageInfo_UpstreamEndpointStats.Marshal(b, m, deterministic)
|
||||||
|
}
|
||||||
|
func (dst *UpstreamEndpointStats) XXX_Merge(src proto.Message) {
|
||||||
|
xxx_messageInfo_UpstreamEndpointStats.Merge(dst, src)
|
||||||
|
}
|
||||||
|
func (m *UpstreamEndpointStats) XXX_Size() int {
|
||||||
|
return xxx_messageInfo_UpstreamEndpointStats.Size(m)
|
||||||
|
}
|
||||||
|
func (m *UpstreamEndpointStats) XXX_DiscardUnknown() {
|
||||||
|
xxx_messageInfo_UpstreamEndpointStats.DiscardUnknown(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
var xxx_messageInfo_UpstreamEndpointStats proto.InternalMessageInfo
|
||||||
|
|
||||||
|
func (m *UpstreamEndpointStats) GetAddress() *address.Address {
|
||||||
|
if m != nil {
|
||||||
|
return m.Address
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamEndpointStats) GetMetadata() *_struct.Struct {
|
||||||
|
if m != nil {
|
||||||
|
return m.Metadata
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamEndpointStats) GetTotalSuccessfulRequests() uint64 {
|
||||||
|
if m != nil {
|
||||||
|
return m.TotalSuccessfulRequests
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamEndpointStats) GetTotalRequestsInProgress() uint64 {
|
||||||
|
if m != nil {
|
||||||
|
return m.TotalRequestsInProgress
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamEndpointStats) GetTotalErrorRequests() uint64 {
|
||||||
|
if m != nil {
|
||||||
|
return m.TotalErrorRequests
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *UpstreamEndpointStats) GetLoadMetricStats() []*EndpointLoadMetricStats {
|
||||||
|
if m != nil {
|
||||||
|
return m.LoadMetricStats
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type EndpointLoadMetricStats struct {
|
||||||
|
MetricName string `protobuf:"bytes,1,opt,name=metric_name,json=metricName,proto3" json:"metric_name,omitempty"`
|
||||||
|
NumRequestsFinishedWithMetric uint64 `protobuf:"varint,2,opt,name=num_requests_finished_with_metric,json=numRequestsFinishedWithMetric,proto3" json:"num_requests_finished_with_metric,omitempty"`
|
||||||
|
TotalMetricValue float64 `protobuf:"fixed64,3,opt,name=total_metric_value,json=totalMetricValue,proto3" json:"total_metric_value,omitempty"`
|
||||||
|
XXX_NoUnkeyedLiteral struct{} `json:"-"`
|
||||||
|
XXX_unrecognized []byte `json:"-"`
|
||||||
|
XXX_sizecache int32 `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *EndpointLoadMetricStats) Reset() { *m = EndpointLoadMetricStats{} }
|
||||||
|
func (m *EndpointLoadMetricStats) String() string { return proto.CompactTextString(m) }
|
||||||
|
func (*EndpointLoadMetricStats) ProtoMessage() {}
|
||||||
|
func (*EndpointLoadMetricStats) Descriptor() ([]byte, []int) {
|
||||||
|
return fileDescriptor_load_report_5485b60725c658b8, []int{2}
|
||||||
|
}
|
||||||
|
func (m *EndpointLoadMetricStats) XXX_Unmarshal(b []byte) error {
|
||||||
|
return xxx_messageInfo_EndpointLoadMetricStats.Unmarshal(m, b)
|
||||||
|
}
|
||||||
|
func (m *EndpointLoadMetricStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
|
||||||
|
return xxx_messageInfo_EndpointLoadMetricStats.Marshal(b, m, deterministic)
|
||||||
|
}
|
||||||
|
func (dst *EndpointLoadMetricStats) XXX_Merge(src proto.Message) {
|
||||||
|
xxx_messageInfo_EndpointLoadMetricStats.Merge(dst, src)
|
||||||
|
}
|
||||||
|
func (m *EndpointLoadMetricStats) XXX_Size() int {
|
||||||
|
return xxx_messageInfo_EndpointLoadMetricStats.Size(m)
|
||||||
|
}
|
||||||
|
func (m *EndpointLoadMetricStats) XXX_DiscardUnknown() {
|
||||||
|
xxx_messageInfo_EndpointLoadMetricStats.DiscardUnknown(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
var xxx_messageInfo_EndpointLoadMetricStats proto.InternalMessageInfo
|
||||||
|
|
||||||
|
func (m *EndpointLoadMetricStats) GetMetricName() string {
|
||||||
|
if m != nil {
|
||||||
|
return m.MetricName
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *EndpointLoadMetricStats) GetNumRequestsFinishedWithMetric() uint64 {
|
||||||
|
if m != nil {
|
||||||
|
return m.NumRequestsFinishedWithMetric
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *EndpointLoadMetricStats) GetTotalMetricValue() float64 {
|
||||||
|
if m != nil {
|
||||||
|
return m.TotalMetricValue
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
type ClusterStats struct {
|
||||||
|
ClusterName string `protobuf:"bytes,1,opt,name=cluster_name,json=clusterName,proto3" json:"cluster_name,omitempty"`
|
||||||
|
ClusterServiceName string `protobuf:"bytes,6,opt,name=cluster_service_name,json=clusterServiceName,proto3" json:"cluster_service_name,omitempty"`
|
||||||
|
UpstreamLocalityStats []*UpstreamLocalityStats `protobuf:"bytes,2,rep,name=upstream_locality_stats,json=upstreamLocalityStats,proto3" json:"upstream_locality_stats,omitempty"`
|
||||||
|
TotalDroppedRequests uint64 `protobuf:"varint,3,opt,name=total_dropped_requests,json=totalDroppedRequests,proto3" json:"total_dropped_requests,omitempty"`
|
||||||
|
DroppedRequests []*ClusterStats_DroppedRequests `protobuf:"bytes,5,rep,name=dropped_requests,json=droppedRequests,proto3" json:"dropped_requests,omitempty"`
|
||||||
|
LoadReportInterval *duration.Duration `protobuf:"bytes,4,opt,name=load_report_interval,json=loadReportInterval,proto3" json:"load_report_interval,omitempty"`
|
||||||
|
XXX_NoUnkeyedLiteral struct{} `json:"-"`
|
||||||
|
XXX_unrecognized []byte `json:"-"`
|
||||||
|
XXX_sizecache int32 `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *ClusterStats) Reset() { *m = ClusterStats{} }
|
||||||
|
func (m *ClusterStats) String() string { return proto.CompactTextString(m) }
|
||||||
|
func (*ClusterStats) ProtoMessage() {}
|
||||||
|
func (*ClusterStats) Descriptor() ([]byte, []int) {
|
||||||
|
return fileDescriptor_load_report_5485b60725c658b8, []int{3}
|
||||||
|
}
|
||||||
|
func (m *ClusterStats) XXX_Unmarshal(b []byte) error {
|
||||||
|
return xxx_messageInfo_ClusterStats.Unmarshal(m, b)
|
||||||
|
}
|
||||||
|
func (m *ClusterStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
|
||||||
|
return xxx_messageInfo_ClusterStats.Marshal(b, m, deterministic)
|
||||||
|
}
|
||||||
|
func (dst *ClusterStats) XXX_Merge(src proto.Message) {
|
||||||
|
xxx_messageInfo_ClusterStats.Merge(dst, src)
|
||||||
|
}
|
||||||
|
func (m *ClusterStats) XXX_Size() int {
|
||||||
|
return xxx_messageInfo_ClusterStats.Size(m)
|
||||||
|
}
|
||||||
|
func (m *ClusterStats) XXX_DiscardUnknown() {
|
||||||
|
xxx_messageInfo_ClusterStats.DiscardUnknown(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
var xxx_messageInfo_ClusterStats proto.InternalMessageInfo
|
||||||
|
|
||||||
|
func (m *ClusterStats) GetClusterName() string {
|
||||||
|
if m != nil {
|
||||||
|
return m.ClusterName
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *ClusterStats) GetClusterServiceName() string {
|
||||||
|
if m != nil {
|
||||||
|
return m.ClusterServiceName
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *ClusterStats) GetUpstreamLocalityStats() []*UpstreamLocalityStats {
|
||||||
|
if m != nil {
|
||||||
|
return m.UpstreamLocalityStats
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *ClusterStats) GetTotalDroppedRequests() uint64 {
|
||||||
|
if m != nil {
|
||||||
|
return m.TotalDroppedRequests
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *ClusterStats) GetDroppedRequests() []*ClusterStats_DroppedRequests {
|
||||||
|
if m != nil {
|
||||||
|
return m.DroppedRequests
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *ClusterStats) GetLoadReportInterval() *duration.Duration {
|
||||||
|
if m != nil {
|
||||||
|
return m.LoadReportInterval
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type ClusterStats_DroppedRequests struct {
|
||||||
|
Category string `protobuf:"bytes,1,opt,name=category,proto3" json:"category,omitempty"`
|
||||||
|
DroppedCount uint64 `protobuf:"varint,2,opt,name=dropped_count,json=droppedCount,proto3" json:"dropped_count,omitempty"`
|
||||||
|
XXX_NoUnkeyedLiteral struct{} `json:"-"`
|
||||||
|
XXX_unrecognized []byte `json:"-"`
|
||||||
|
XXX_sizecache int32 `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *ClusterStats_DroppedRequests) Reset() { *m = ClusterStats_DroppedRequests{} }
|
||||||
|
func (m *ClusterStats_DroppedRequests) String() string { return proto.CompactTextString(m) }
|
||||||
|
func (*ClusterStats_DroppedRequests) ProtoMessage() {}
|
||||||
|
func (*ClusterStats_DroppedRequests) Descriptor() ([]byte, []int) {
|
||||||
|
return fileDescriptor_load_report_5485b60725c658b8, []int{3, 0}
|
||||||
|
}
|
||||||
|
func (m *ClusterStats_DroppedRequests) XXX_Unmarshal(b []byte) error {
|
||||||
|
return xxx_messageInfo_ClusterStats_DroppedRequests.Unmarshal(m, b)
|
||||||
|
}
|
||||||
|
func (m *ClusterStats_DroppedRequests) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
|
||||||
|
return xxx_messageInfo_ClusterStats_DroppedRequests.Marshal(b, m, deterministic)
|
||||||
|
}
|
||||||
|
func (dst *ClusterStats_DroppedRequests) XXX_Merge(src proto.Message) {
|
||||||
|
xxx_messageInfo_ClusterStats_DroppedRequests.Merge(dst, src)
|
||||||
|
}
|
||||||
|
func (m *ClusterStats_DroppedRequests) XXX_Size() int {
|
||||||
|
return xxx_messageInfo_ClusterStats_DroppedRequests.Size(m)
|
||||||
|
}
|
||||||
|
func (m *ClusterStats_DroppedRequests) XXX_DiscardUnknown() {
|
||||||
|
xxx_messageInfo_ClusterStats_DroppedRequests.DiscardUnknown(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
var xxx_messageInfo_ClusterStats_DroppedRequests proto.InternalMessageInfo
|
||||||
|
|
||||||
|
func (m *ClusterStats_DroppedRequests) GetCategory() string {
|
||||||
|
if m != nil {
|
||||||
|
return m.Category
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *ClusterStats_DroppedRequests) GetDroppedCount() uint64 {
|
||||||
|
if m != nil {
|
||||||
|
return m.DroppedCount
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
proto.RegisterType((*UpstreamLocalityStats)(nil), "envoy.api.v2.endpoint.UpstreamLocalityStats")
|
||||||
|
proto.RegisterType((*UpstreamEndpointStats)(nil), "envoy.api.v2.endpoint.UpstreamEndpointStats")
|
||||||
|
proto.RegisterType((*EndpointLoadMetricStats)(nil), "envoy.api.v2.endpoint.EndpointLoadMetricStats")
|
||||||
|
proto.RegisterType((*ClusterStats)(nil), "envoy.api.v2.endpoint.ClusterStats")
|
||||||
|
proto.RegisterType((*ClusterStats_DroppedRequests)(nil), "envoy.api.v2.endpoint.ClusterStats.DroppedRequests")
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
proto.RegisterFile("envoy/api/v2/endpoint/load_report.proto", fileDescriptor_load_report_5485b60725c658b8)
|
||||||
|
}
|
||||||
|
|
||||||
|
var fileDescriptor_load_report_5485b60725c658b8 = []byte{
|
||||||
|
// 737 bytes of a gzipped FileDescriptorProto
|
||||||
|
0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x55, 0x5b, 0x6e, 0xd3, 0x4c,
|
||||||
|
0x18, 0x95, 0x93, 0xb4, 0x4d, 0x27, 0xad, 0xd2, 0x7f, 0xd4, 0xfe, 0x49, 0xfd, 0x5f, 0x1a, 0x52,
|
||||||
|
0x21, 0xf2, 0x50, 0xd9, 0x55, 0x5a, 0x09, 0x04, 0x4f, 0xa4, 0x2d, 0xa2, 0xa2, 0xa0, 0xca, 0x11,
|
||||||
|
0x20, 0x21, 0x81, 0x35, 0xb5, 0xa7, 0xe9, 0x48, 0xb6, 0xc7, 0xcc, 0x8c, 0x0d, 0x59, 0x02, 0xaf,
|
||||||
|
0x2c, 0x81, 0x25, 0xf0, 0xc8, 0x13, 0xdb, 0x60, 0x09, 0xec, 0x02, 0x79, 0x2e, 0xce, 0xb5, 0x12,
|
||||||
|
0x0b, 0xe0, 0xcd, 0xfe, 0xce, 0x39, 0xfe, 0x6e, 0x67, 0xc6, 0xe0, 0x1e, 0x4e, 0x72, 0x3a, 0x76,
|
||||||
|
0x51, 0x4a, 0xdc, 0xbc, 0xef, 0xe2, 0x24, 0x4c, 0x29, 0x49, 0x84, 0x1b, 0x51, 0x14, 0xfa, 0x0c,
|
||||||
|
0xa7, 0x94, 0x09, 0x27, 0x65, 0x54, 0x50, 0xb8, 0x23, 0x89, 0x0e, 0x4a, 0x89, 0x93, 0xf7, 0x1d,
|
||||||
|
0x43, 0xb4, 0xf7, 0x66, 0xf4, 0x01, 0x65, 0xd8, 0x45, 0x61, 0xc8, 0x30, 0xe7, 0x4a, 0x67, 0xff,
|
||||||
|
0xbb, 0x48, 0xb8, 0x42, 0x1c, 0x6b, 0xf4, 0xff, 0x11, 0xa5, 0xa3, 0x08, 0xbb, 0xf2, 0xed, 0x2a,
|
||||||
|
0xbb, 0x76, 0xc3, 0x8c, 0x21, 0x41, 0x68, 0x62, 0xd4, 0xf3, 0x38, 0x17, 0x2c, 0x0b, 0x74, 0x4d,
|
||||||
|
0x76, 0x2b, 0x47, 0x11, 0x09, 0x91, 0xc0, 0xae, 0x79, 0x50, 0x40, 0xf7, 0x47, 0x15, 0xec, 0xbc,
|
||||||
|
0x4c, 0xb9, 0x60, 0x18, 0xc5, 0x17, 0x34, 0x40, 0x11, 0x11, 0xe3, 0xa1, 0x40, 0x82, 0xc3, 0xfb,
|
||||||
|
0xa0, 0x1e, 0xe9, 0x40, 0xdb, 0xea, 0x58, 0xbd, 0x46, 0xff, 0x1f, 0x67, 0xa6, 0xb3, 0xa2, 0x42,
|
||||||
|
0xc7, 0x68, 0xbc, 0x92, 0x0c, 0x1f, 0x82, 0x5d, 0x41, 0x05, 0x8a, 0x7c, 0x9e, 0x05, 0x01, 0xe6,
|
||||||
|
0xfc, 0x3a, 0x8b, 0x7c, 0x86, 0xdf, 0x67, 0x98, 0x0b, 0xde, 0xae, 0x74, 0xac, 0x5e, 0xcd, 0x6b,
|
||||||
|
0x49, 0xc2, 0xb0, 0xc4, 0x3d, 0x0d, 0xc3, 0x47, 0xc0, 0x56, 0x5a, 0x23, 0xf0, 0x49, 0xe2, 0xa7,
|
||||||
|
0x8c, 0x8e, 0x8a, 0x39, 0xb5, 0xab, 0x53, 0x62, 0x23, 0x39, 0x4f, 0x2e, 0x35, 0x0c, 0x0f, 0xc1,
|
||||||
|
0xb6, 0x12, 0x63, 0xc6, 0x28, 0x9b, 0xe4, 0xac, 0x49, 0x19, 0x94, 0xd8, 0x59, 0x01, 0x95, 0xe9,
|
||||||
|
0xde, 0x80, 0xbf, 0xe4, 0xfe, 0x62, 0x2c, 0x18, 0x09, 0x7c, 0x5e, 0x34, 0xde, 0x5e, 0xe9, 0x54,
|
||||||
|
0x7b, 0x8d, 0xbe, 0xe3, 0x2c, 0x5d, 0xa3, 0x73, 0xa6, 0x1f, 0x2e, 0x28, 0x0a, 0x9f, 0x4b, 0x99,
|
||||||
|
0x1c, 0x97, 0xd7, 0x8c, 0x66, 0x03, 0x30, 0x04, 0xad, 0x4c, 0x0f, 0xd6, 0x37, 0x6a, 0x9d, 0x61,
|
||||||
|
0x4d, 0x66, 0x38, 0xb8, 0x25, 0x83, 0x59, 0x87, 0xc9, 0xa4, 0xbe, 0xbf, 0x93, 0x2d, 0x0b, 0x43,
|
||||||
|
0x1b, 0xd4, 0x53, 0x46, 0x28, 0x2b, 0xb6, 0xb4, 0xda, 0xb1, 0x7a, 0x9b, 0x5e, 0xf9, 0xde, 0xfd,
|
||||||
|
0x34, 0xb5, 0xdb, 0x59, 0xd5, 0x31, 0x58, 0xd3, 0xde, 0xd3, 0xab, 0xb5, 0x97, 0xac, 0xf6, 0xb1,
|
||||||
|
0x62, 0x78, 0x86, 0x0a, 0x8f, 0x40, 0x3d, 0xc6, 0x02, 0x85, 0x48, 0x20, 0x99, 0xab, 0xd1, 0x6f,
|
||||||
|
0x39, 0xca, 0x75, 0x8e, 0x71, 0x9d, 0x33, 0x94, 0xae, 0xf3, 0x4a, 0xe2, 0x1f, 0x37, 0xc8, 0x40,
|
||||||
|
0xf7, 0xab, 0x05, 0x5a, 0xb7, 0x90, 0xe1, 0x1e, 0x68, 0xe8, 0x94, 0x09, 0x8a, 0xb1, 0xdc, 0xc8,
|
||||||
|
0xba, 0x07, 0x54, 0xe8, 0x05, 0x8a, 0x31, 0x7c, 0x0a, 0xee, 0x24, 0x59, 0x3c, 0x99, 0xc2, 0x35,
|
||||||
|
0x49, 0x08, 0xbf, 0xc1, 0xa1, 0xff, 0x81, 0x88, 0x1b, 0x5d, 0xae, 0x9e, 0xe5, 0x7f, 0x49, 0x16,
|
||||||
|
0x9b, 0x86, 0x9e, 0x68, 0xda, 0x6b, 0x22, 0x6e, 0x54, 0x3e, 0x78, 0x00, 0x54, 0xe3, 0xa6, 0xc7,
|
||||||
|
0x1c, 0x45, 0x19, 0x96, 0x93, 0xb4, 0xbc, 0x2d, 0x89, 0x28, 0xe2, 0xab, 0x22, 0xde, 0xfd, 0x52,
|
||||||
|
0x03, 0x1b, 0x27, 0x51, 0xc6, 0x05, 0x66, 0xaa, 0xd2, 0x03, 0xb0, 0x11, 0xa8, 0xf7, 0xa9, 0x52,
|
||||||
|
0x07, 0xeb, 0xdf, 0x7e, 0x7e, 0xaf, 0xd6, 0x58, 0xa5, 0x63, 0x79, 0x0d, 0x0d, 0xcb, 0xb2, 0x0f,
|
||||||
|
0xc1, 0xb6, 0x61, 0x73, 0xcc, 0x72, 0x12, 0x60, 0xa5, 0x5a, 0x95, 0x0d, 0x42, 0x8d, 0x0d, 0x15,
|
||||||
|
0x24, 0x15, 0xe9, 0xd4, 0x99, 0x31, 0xf7, 0x89, 0xde, 0x43, 0xe5, 0xb7, 0xce, 0xcc, 0xcc, 0x15,
|
||||||
|
0x36, 0x00, 0x45, 0x61, 0x2b, 0x9f, 0xad, 0x4a, 0xdd, 0x9a, 0x9c, 0x9f, 0xd9, 0x5b, 0xee, 0x18,
|
||||||
|
0xfc, 0xad, 0x06, 0x12, 0x32, 0x9a, 0xa6, 0x38, 0x9c, 0xf8, 0x44, 0xd9, 0x4b, 0x79, 0xe8, 0x54,
|
||||||
|
0x81, 0xa5, 0x53, 0xde, 0x81, 0xad, 0x05, 0xbe, 0x32, 0xca, 0xd1, 0x2d, 0x05, 0x4e, 0x8f, 0xd1,
|
||||||
|
0x99, 0xfb, 0x9c, 0xd7, 0x0c, 0xe7, 0xbe, 0xff, 0x0c, 0x6c, 0x4f, 0xfd, 0x57, 0x7c, 0x92, 0x08,
|
||||||
|
0xcc, 0x72, 0x14, 0x49, 0xef, 0x36, 0xfa, 0xbb, 0x0b, 0xa7, 0xee, 0x54, 0xff, 0x0b, 0x3c, 0x58,
|
||||||
|
0xc8, 0x3c, 0xa9, 0x3a, 0xd7, 0x22, 0xfb, 0x2d, 0x68, 0xce, 0xd7, 0x7f, 0x17, 0xd4, 0x03, 0x24,
|
||||||
|
0xf0, 0x88, 0xb2, 0xf1, 0xe2, 0x0e, 0x4b, 0x08, 0xee, 0x83, 0x4d, 0xd3, 0x66, 0x40, 0xb3, 0x44,
|
||||||
|
0x68, 0x8f, 0x6d, 0xe8, 0xe0, 0x49, 0x11, 0x1b, 0x3c, 0x00, 0xfb, 0x84, 0xaa, 0xae, 0x53, 0x46,
|
||||||
|
0x3f, 0x8e, 0x97, 0x0f, 0x60, 0xd0, 0xbc, 0x28, 0x2b, 0xbb, 0x2c, 0xca, 0xbe, 0xb4, 0xae, 0x56,
|
||||||
|
0x65, 0xfd, 0x47, 0xbf, 0x02, 0x00, 0x00, 0xff, 0xff, 0xad, 0x11, 0xfc, 0x18, 0x5a, 0x07, 0x00,
|
||||||
|
0x00,
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,272 @@
|
||||||
|
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||||
|
// source: envoy/service/load_stats/v2/lrs.proto
|
||||||
|
|
||||||
|
package v2
|
||||||
|
|
||||||
|
import proto "github.com/golang/protobuf/proto"
|
||||||
|
import fmt "fmt"
|
||||||
|
import math "math"
|
||||||
|
import duration "github.com/golang/protobuf/ptypes/duration"
|
||||||
|
import base "google.golang.org/grpc/balancer/xds/internal/proto/envoy/api/v2/core/base"
|
||||||
|
import load_report "google.golang.org/grpc/balancer/xds/internal/proto/envoy/api/v2/endpoint/load_report"
|
||||||
|
import _ "google.golang.org/grpc/balancer/xds/internal/proto/validate"
|
||||||
|
|
||||||
|
import (
|
||||||
|
context "golang.org/x/net/context"
|
||||||
|
grpc "google.golang.org/grpc"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Reference imports to suppress errors if they are not otherwise used.
|
||||||
|
var _ = proto.Marshal
|
||||||
|
var _ = fmt.Errorf
|
||||||
|
var _ = math.Inf
|
||||||
|
|
||||||
|
// This is a compile-time assertion to ensure that this generated file
|
||||||
|
// is compatible with the proto package it is being compiled against.
|
||||||
|
// A compilation error at this line likely means your copy of the
|
||||||
|
// proto package needs to be updated.
|
||||||
|
const _ = proto.ProtoPackageIsVersion2 // please upgrade the proto package
|
||||||
|
|
||||||
|
type LoadStatsRequest struct {
|
||||||
|
Node *base.Node `protobuf:"bytes,1,opt,name=node,proto3" json:"node,omitempty"`
|
||||||
|
ClusterStats []*load_report.ClusterStats `protobuf:"bytes,2,rep,name=cluster_stats,json=clusterStats,proto3" json:"cluster_stats,omitempty"`
|
||||||
|
XXX_NoUnkeyedLiteral struct{} `json:"-"`
|
||||||
|
XXX_unrecognized []byte `json:"-"`
|
||||||
|
XXX_sizecache int32 `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *LoadStatsRequest) Reset() { *m = LoadStatsRequest{} }
|
||||||
|
func (m *LoadStatsRequest) String() string { return proto.CompactTextString(m) }
|
||||||
|
func (*LoadStatsRequest) ProtoMessage() {}
|
||||||
|
func (*LoadStatsRequest) Descriptor() ([]byte, []int) {
|
||||||
|
return fileDescriptor_lrs_0f8f2c1d40a1b9f7, []int{0}
|
||||||
|
}
|
||||||
|
func (m *LoadStatsRequest) XXX_Unmarshal(b []byte) error {
|
||||||
|
return xxx_messageInfo_LoadStatsRequest.Unmarshal(m, b)
|
||||||
|
}
|
||||||
|
func (m *LoadStatsRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
|
||||||
|
return xxx_messageInfo_LoadStatsRequest.Marshal(b, m, deterministic)
|
||||||
|
}
|
||||||
|
func (dst *LoadStatsRequest) XXX_Merge(src proto.Message) {
|
||||||
|
xxx_messageInfo_LoadStatsRequest.Merge(dst, src)
|
||||||
|
}
|
||||||
|
func (m *LoadStatsRequest) XXX_Size() int {
|
||||||
|
return xxx_messageInfo_LoadStatsRequest.Size(m)
|
||||||
|
}
|
||||||
|
func (m *LoadStatsRequest) XXX_DiscardUnknown() {
|
||||||
|
xxx_messageInfo_LoadStatsRequest.DiscardUnknown(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
var xxx_messageInfo_LoadStatsRequest proto.InternalMessageInfo
|
||||||
|
|
||||||
|
func (m *LoadStatsRequest) GetNode() *base.Node {
|
||||||
|
if m != nil {
|
||||||
|
return m.Node
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *LoadStatsRequest) GetClusterStats() []*load_report.ClusterStats {
|
||||||
|
if m != nil {
|
||||||
|
return m.ClusterStats
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type LoadStatsResponse struct {
|
||||||
|
Clusters []string `protobuf:"bytes,1,rep,name=clusters,proto3" json:"clusters,omitempty"`
|
||||||
|
LoadReportingInterval *duration.Duration `protobuf:"bytes,2,opt,name=load_reporting_interval,json=loadReportingInterval,proto3" json:"load_reporting_interval,omitempty"`
|
||||||
|
ReportEndpointGranularity bool `protobuf:"varint,3,opt,name=report_endpoint_granularity,json=reportEndpointGranularity,proto3" json:"report_endpoint_granularity,omitempty"`
|
||||||
|
XXX_NoUnkeyedLiteral struct{} `json:"-"`
|
||||||
|
XXX_unrecognized []byte `json:"-"`
|
||||||
|
XXX_sizecache int32 `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *LoadStatsResponse) Reset() { *m = LoadStatsResponse{} }
|
||||||
|
func (m *LoadStatsResponse) String() string { return proto.CompactTextString(m) }
|
||||||
|
func (*LoadStatsResponse) ProtoMessage() {}
|
||||||
|
func (*LoadStatsResponse) Descriptor() ([]byte, []int) {
|
||||||
|
return fileDescriptor_lrs_0f8f2c1d40a1b9f7, []int{1}
|
||||||
|
}
|
||||||
|
func (m *LoadStatsResponse) XXX_Unmarshal(b []byte) error {
|
||||||
|
return xxx_messageInfo_LoadStatsResponse.Unmarshal(m, b)
|
||||||
|
}
|
||||||
|
func (m *LoadStatsResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
|
||||||
|
return xxx_messageInfo_LoadStatsResponse.Marshal(b, m, deterministic)
|
||||||
|
}
|
||||||
|
func (dst *LoadStatsResponse) XXX_Merge(src proto.Message) {
|
||||||
|
xxx_messageInfo_LoadStatsResponse.Merge(dst, src)
|
||||||
|
}
|
||||||
|
func (m *LoadStatsResponse) XXX_Size() int {
|
||||||
|
return xxx_messageInfo_LoadStatsResponse.Size(m)
|
||||||
|
}
|
||||||
|
func (m *LoadStatsResponse) XXX_DiscardUnknown() {
|
||||||
|
xxx_messageInfo_LoadStatsResponse.DiscardUnknown(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
var xxx_messageInfo_LoadStatsResponse proto.InternalMessageInfo
|
||||||
|
|
||||||
|
func (m *LoadStatsResponse) GetClusters() []string {
|
||||||
|
if m != nil {
|
||||||
|
return m.Clusters
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *LoadStatsResponse) GetLoadReportingInterval() *duration.Duration {
|
||||||
|
if m != nil {
|
||||||
|
return m.LoadReportingInterval
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *LoadStatsResponse) GetReportEndpointGranularity() bool {
|
||||||
|
if m != nil {
|
||||||
|
return m.ReportEndpointGranularity
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
proto.RegisterType((*LoadStatsRequest)(nil), "envoy.service.load_stats.v2.LoadStatsRequest")
|
||||||
|
proto.RegisterType((*LoadStatsResponse)(nil), "envoy.service.load_stats.v2.LoadStatsResponse")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reference imports to suppress errors if they are not otherwise used.
|
||||||
|
var _ context.Context
|
||||||
|
var _ grpc.ClientConn
|
||||||
|
|
||||||
|
// This is a compile-time assertion to ensure that this generated file
|
||||||
|
// is compatible with the grpc package it is being compiled against.
|
||||||
|
const _ = grpc.SupportPackageIsVersion4
|
||||||
|
|
||||||
|
// LoadReportingServiceClient is the client API for LoadReportingService service.
|
||||||
|
//
|
||||||
|
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://godoc.org/google.golang.org/grpc#ClientConn.NewStream.
|
||||||
|
type LoadReportingServiceClient interface {
|
||||||
|
StreamLoadStats(ctx context.Context, opts ...grpc.CallOption) (LoadReportingService_StreamLoadStatsClient, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
type loadReportingServiceClient struct {
|
||||||
|
cc *grpc.ClientConn
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewLoadReportingServiceClient(cc *grpc.ClientConn) LoadReportingServiceClient {
|
||||||
|
return &loadReportingServiceClient{cc}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *loadReportingServiceClient) StreamLoadStats(ctx context.Context, opts ...grpc.CallOption) (LoadReportingService_StreamLoadStatsClient, error) {
|
||||||
|
stream, err := c.cc.NewStream(ctx, &_LoadReportingService_serviceDesc.Streams[0], "/envoy.service.load_stats.v2.LoadReportingService/StreamLoadStats", opts...)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
x := &loadReportingServiceStreamLoadStatsClient{stream}
|
||||||
|
return x, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type LoadReportingService_StreamLoadStatsClient interface {
|
||||||
|
Send(*LoadStatsRequest) error
|
||||||
|
Recv() (*LoadStatsResponse, error)
|
||||||
|
grpc.ClientStream
|
||||||
|
}
|
||||||
|
|
||||||
|
type loadReportingServiceStreamLoadStatsClient struct {
|
||||||
|
grpc.ClientStream
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *loadReportingServiceStreamLoadStatsClient) Send(m *LoadStatsRequest) error {
|
||||||
|
return x.ClientStream.SendMsg(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *loadReportingServiceStreamLoadStatsClient) Recv() (*LoadStatsResponse, error) {
|
||||||
|
m := new(LoadStatsResponse)
|
||||||
|
if err := x.ClientStream.RecvMsg(m); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return m, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadReportingServiceServer is the server API for LoadReportingService service.
|
||||||
|
type LoadReportingServiceServer interface {
|
||||||
|
StreamLoadStats(LoadReportingService_StreamLoadStatsServer) error
|
||||||
|
}
|
||||||
|
|
||||||
|
func RegisterLoadReportingServiceServer(s *grpc.Server, srv LoadReportingServiceServer) {
|
||||||
|
s.RegisterService(&_LoadReportingService_serviceDesc, srv)
|
||||||
|
}
|
||||||
|
|
||||||
|
func _LoadReportingService_StreamLoadStats_Handler(srv interface{}, stream grpc.ServerStream) error {
|
||||||
|
return srv.(LoadReportingServiceServer).StreamLoadStats(&loadReportingServiceStreamLoadStatsServer{stream})
|
||||||
|
}
|
||||||
|
|
||||||
|
type LoadReportingService_StreamLoadStatsServer interface {
|
||||||
|
Send(*LoadStatsResponse) error
|
||||||
|
Recv() (*LoadStatsRequest, error)
|
||||||
|
grpc.ServerStream
|
||||||
|
}
|
||||||
|
|
||||||
|
type loadReportingServiceStreamLoadStatsServer struct {
|
||||||
|
grpc.ServerStream
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *loadReportingServiceStreamLoadStatsServer) Send(m *LoadStatsResponse) error {
|
||||||
|
return x.ServerStream.SendMsg(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *loadReportingServiceStreamLoadStatsServer) Recv() (*LoadStatsRequest, error) {
|
||||||
|
m := new(LoadStatsRequest)
|
||||||
|
if err := x.ServerStream.RecvMsg(m); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return m, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var _LoadReportingService_serviceDesc = grpc.ServiceDesc{
|
||||||
|
ServiceName: "envoy.service.load_stats.v2.LoadReportingService",
|
||||||
|
HandlerType: (*LoadReportingServiceServer)(nil),
|
||||||
|
Methods: []grpc.MethodDesc{},
|
||||||
|
Streams: []grpc.StreamDesc{
|
||||||
|
{
|
||||||
|
StreamName: "StreamLoadStats",
|
||||||
|
Handler: _LoadReportingService_StreamLoadStats_Handler,
|
||||||
|
ServerStreams: true,
|
||||||
|
ClientStreams: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Metadata: "envoy/service/load_stats/v2/lrs.proto",
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
proto.RegisterFile("envoy/service/load_stats/v2/lrs.proto", fileDescriptor_lrs_0f8f2c1d40a1b9f7)
|
||||||
|
}
|
||||||
|
|
||||||
|
var fileDescriptor_lrs_0f8f2c1d40a1b9f7 = []byte{
|
||||||
|
// 429 bytes of a gzipped FileDescriptorProto
|
||||||
|
0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x92, 0x41, 0x8e, 0xd3, 0x30,
|
||||||
|
0x14, 0x86, 0x71, 0x0b, 0xa8, 0x78, 0x40, 0x80, 0x05, 0x6a, 0xa6, 0x83, 0x50, 0x55, 0x04, 0x04,
|
||||||
|
0x21, 0x6c, 0x14, 0xf6, 0xb3, 0x28, 0x20, 0x40, 0xaa, 0xd0, 0x90, 0xee, 0xd8, 0x54, 0x6e, 0xf2,
|
||||||
|
0x88, 0x2c, 0x05, 0xbf, 0x60, 0x3b, 0x16, 0xbd, 0x01, 0x6c, 0x58, 0x70, 0x1c, 0x56, 0x9c, 0x80,
|
||||||
|
0x7b, 0x70, 0x0b, 0x94, 0x38, 0x99, 0xc9, 0xb0, 0xa8, 0x66, 0x17, 0xeb, 0x7d, 0xff, 0xf3, 0xff,
|
||||||
|
0xff, 0x31, 0x7d, 0x08, 0xda, 0xe3, 0x4e, 0x58, 0x30, 0x5e, 0x65, 0x20, 0x4a, 0x94, 0xf9, 0xc6,
|
||||||
|
0x3a, 0xe9, 0xac, 0xf0, 0x89, 0x28, 0x8d, 0xe5, 0x95, 0x41, 0x87, 0xec, 0xa8, 0xc5, 0x78, 0x87,
|
||||||
|
0xf1, 0x33, 0x8c, 0xfb, 0x64, 0x76, 0x2f, 0xec, 0x90, 0x95, 0x6a, 0x44, 0x19, 0x1a, 0x10, 0x5b,
|
||||||
|
0x69, 0x21, 0x48, 0x67, 0x8f, 0xcf, 0x4d, 0x41, 0xe7, 0x15, 0x2a, 0xed, 0xc2, 0x4d, 0x06, 0x2a,
|
||||||
|
0x34, 0xae, 0x03, 0xef, 0x17, 0x88, 0x45, 0x09, 0xa2, 0x3d, 0x6d, 0xeb, 0x4f, 0x22, 0xaf, 0x8d,
|
||||||
|
0x74, 0x0a, 0x75, 0x37, 0x9f, 0x7a, 0x59, 0xaa, 0x5c, 0x3a, 0x10, 0xfd, 0x47, 0x18, 0x2c, 0xbe,
|
||||||
|
0x13, 0x7a, 0x6b, 0x85, 0x32, 0x5f, 0x37, 0x86, 0x52, 0xf8, 0x52, 0x83, 0x75, 0xec, 0x29, 0xbd,
|
||||||
|
0xac, 0x31, 0x87, 0x88, 0xcc, 0x49, 0x7c, 0x90, 0x4c, 0x79, 0x08, 0x20, 0x2b, 0xc5, 0x7d, 0xc2,
|
||||||
|
0x1b, 0x8f, 0xfc, 0x3d, 0xe6, 0x90, 0xb6, 0x10, 0x7b, 0x4b, 0x6f, 0x64, 0x65, 0x6d, 0x1d, 0x98,
|
||||||
|
0x90, 0x2a, 0x1a, 0xcd, 0xc7, 0xf1, 0x41, 0xf2, 0xe0, 0xbc, 0xaa, 0xf7, 0xce, 0x5f, 0x06, 0x36,
|
||||||
|
0xdc, 0x77, 0x3d, 0x1b, 0x9c, 0x16, 0x7f, 0x08, 0xbd, 0x3d, 0xf0, 0x62, 0x2b, 0xd4, 0x16, 0xd8,
|
||||||
|
0x23, 0x3a, 0xe9, 0x28, 0x1b, 0x91, 0xf9, 0x38, 0xbe, 0xb6, 0xa4, 0xbf, 0xfe, 0xfe, 0x1e, 0x5f,
|
||||||
|
0xf9, 0x49, 0x46, 0x13, 0x92, 0x9e, 0xce, 0xd8, 0x07, 0x3a, 0x1d, 0xf4, 0xa2, 0x74, 0xb1, 0x51,
|
||||||
|
0xda, 0x81, 0xf1, 0xb2, 0x8c, 0x46, 0x6d, 0x8e, 0x43, 0x1e, 0x4a, 0xe2, 0x7d, 0x49, 0xfc, 0x55,
|
||||||
|
0x57, 0x52, 0x7a, 0xb7, 0x51, 0xa6, 0xbd, 0xf0, 0x5d, 0xa7, 0x63, 0xc7, 0xf4, 0x28, 0x6c, 0xdb,
|
||||||
|
0xf4, 0xf6, 0x37, 0x85, 0x91, 0xba, 0x2e, 0xa5, 0x51, 0x6e, 0x17, 0x8d, 0xe7, 0x24, 0x9e, 0xa4,
|
||||||
|
0x87, 0x01, 0x79, 0xdd, 0x11, 0x6f, 0xce, 0x80, 0xe4, 0x07, 0xa1, 0x77, 0x56, 0xc3, 0xcd, 0xeb,
|
||||||
|
0xf0, 0x06, 0x98, 0xa7, 0x37, 0xd7, 0xce, 0x80, 0xfc, 0x7c, 0x1a, 0x97, 0x3d, 0xe3, 0x7b, 0x9e,
|
||||||
|
0x09, 0xff, 0xff, 0x17, 0xcd, 0xf8, 0x45, 0xf1, 0xd0, 0xe2, 0xe2, 0x52, 0x4c, 0x9e, 0x93, 0xe5,
|
||||||
|
0x31, 0x7d, 0xa2, 0x30, 0x28, 0x2b, 0x83, 0x5f, 0x77, 0xfb, 0x96, 0x2c, 0x27, 0x2b, 0x63, 0x4f,
|
||||||
|
0x9a, 0xaa, 0x4e, 0xc8, 0xc7, 0x91, 0x4f, 0xbe, 0x11, 0xb2, 0xbd, 0xda, 0x56, 0xf7, 0xe2, 0x5f,
|
||||||
|
0x00, 0x00, 0x00, 0xff, 0xff, 0x44, 0xa2, 0xf5, 0xb3, 0xfa, 0x02, 0x00, 0x00,
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,200 @@
|
||||||
|
/*
|
||||||
|
* Copyright 2019 gRPC 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 lrs
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"sync"
|
||||||
|
"sync/atomic"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/golang/protobuf/ptypes"
|
||||||
|
structpb "github.com/golang/protobuf/ptypes/struct"
|
||||||
|
"google.golang.org/grpc"
|
||||||
|
basepb "google.golang.org/grpc/balancer/xds/internal/proto/envoy/api/v2/core/base"
|
||||||
|
loadreportpb "google.golang.org/grpc/balancer/xds/internal/proto/envoy/api/v2/endpoint/load_report"
|
||||||
|
lrspb "google.golang.org/grpc/balancer/xds/internal/proto/envoy/service/load_stats/v2/lrs"
|
||||||
|
"google.golang.org/grpc/grpclog"
|
||||||
|
"google.golang.org/grpc/internal/backoff"
|
||||||
|
)
|
||||||
|
|
||||||
|
// lrsStore collects loads from xds balancer, and periodically sends load to the
|
||||||
|
// server.
|
||||||
|
type lrsStore struct {
|
||||||
|
serviceName string
|
||||||
|
node *basepb.Node
|
||||||
|
backoff backoff.Strategy
|
||||||
|
lastReported time.Time
|
||||||
|
|
||||||
|
drops sync.Map // map[string]*uint64
|
||||||
|
}
|
||||||
|
|
||||||
|
const grpcHostname = "com.googleapis.trafficdirector.grpc_hostname"
|
||||||
|
|
||||||
|
func newStore(serviceName string) *lrsStore {
|
||||||
|
return &lrsStore{
|
||||||
|
serviceName: serviceName,
|
||||||
|
node: &basepb.Node{
|
||||||
|
Metadata: &structpb.Struct{
|
||||||
|
Fields: map[string]*structpb.Value{
|
||||||
|
grpcHostname: {
|
||||||
|
Kind: &structpb.Value_StringValue{StringValue: serviceName},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
backoff: backoff.Exponential{
|
||||||
|
MaxDelay: 120 * time.Second,
|
||||||
|
},
|
||||||
|
lastReported: time.Now(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update functions are called by picker for each RPC. To avoid contention, all
|
||||||
|
// updates are done atomically.
|
||||||
|
|
||||||
|
func (ls *lrsStore) callDropped(category string) {
|
||||||
|
p, ok := ls.drops.Load(category)
|
||||||
|
if !ok {
|
||||||
|
tp := new(uint64)
|
||||||
|
p, _ = ls.drops.LoadOrStore(category, tp)
|
||||||
|
}
|
||||||
|
atomic.AddUint64(p.(*uint64), 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: add query counts
|
||||||
|
// callStarted(l locality)
|
||||||
|
// callFinished(l locality, err error)
|
||||||
|
|
||||||
|
func (ls *lrsStore) buildStats() []*loadreportpb.ClusterStats {
|
||||||
|
var (
|
||||||
|
totalDropped uint64
|
||||||
|
droppedReqs []*loadreportpb.ClusterStats_DroppedRequests
|
||||||
|
)
|
||||||
|
ls.drops.Range(func(category, countP interface{}) bool {
|
||||||
|
tempCount := atomic.SwapUint64(countP.(*uint64), 0)
|
||||||
|
if tempCount <= 0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
totalDropped += tempCount
|
||||||
|
droppedReqs = append(droppedReqs, &loadreportpb.ClusterStats_DroppedRequests{
|
||||||
|
Category: category.(string),
|
||||||
|
DroppedCount: tempCount,
|
||||||
|
})
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
dur := time.Since(ls.lastReported)
|
||||||
|
ls.lastReported = time.Now()
|
||||||
|
|
||||||
|
var ret []*loadreportpb.ClusterStats
|
||||||
|
ret = append(ret, &loadreportpb.ClusterStats{
|
||||||
|
ClusterName: ls.serviceName,
|
||||||
|
UpstreamLocalityStats: nil, // TODO: populate this to support per locality loads.
|
||||||
|
|
||||||
|
TotalDroppedRequests: totalDropped,
|
||||||
|
DroppedRequests: droppedReqs,
|
||||||
|
LoadReportInterval: ptypes.DurationProto(dur),
|
||||||
|
})
|
||||||
|
|
||||||
|
return ret
|
||||||
|
}
|
||||||
|
|
||||||
|
// reportTo makes a streaming lrs call to cc and blocks.
|
||||||
|
//
|
||||||
|
// It retries the call (with backoff) until ctx is canceled.
|
||||||
|
func (ls *lrsStore) reportTo(ctx context.Context, cc *grpc.ClientConn) {
|
||||||
|
c := lrspb.NewLoadReportingServiceClient(cc)
|
||||||
|
var (
|
||||||
|
retryCount int
|
||||||
|
doBackoff bool
|
||||||
|
)
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
|
||||||
|
if doBackoff {
|
||||||
|
backoffTimer := time.NewTimer(ls.backoff.Backoff(retryCount))
|
||||||
|
select {
|
||||||
|
case <-backoffTimer.C:
|
||||||
|
case <-ctx.Done():
|
||||||
|
backoffTimer.Stop()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
retryCount++
|
||||||
|
}
|
||||||
|
|
||||||
|
doBackoff = true
|
||||||
|
stream, err := c.StreamLoadStats(ctx)
|
||||||
|
if err != nil {
|
||||||
|
grpclog.Infof("lrs: failed to create stream: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if err := stream.Send(&lrspb.LoadStatsRequest{
|
||||||
|
Node: ls.node,
|
||||||
|
}); err != nil {
|
||||||
|
grpclog.Infof("lrs: failed to send first request: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
first, err := stream.Recv()
|
||||||
|
if err != nil {
|
||||||
|
grpclog.Infof("lrs: failed to receive first response: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
interval, err := ptypes.Duration(first.LoadReportingInterval)
|
||||||
|
if err != nil {
|
||||||
|
grpclog.Infof("lrs: failed to convert report interval: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(first.Clusters) != 1 || first.Clusters[0] != ls.serviceName {
|
||||||
|
grpclog.Infof("lrs: received clusters %v, expect one cluster %q", first.Clusters, ls.serviceName)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if first.ReportEndpointGranularity {
|
||||||
|
// TODO: fixme to support per endpoint loads.
|
||||||
|
grpclog.Infof("lrs: endpoint loads requested, but not supported by current implementation")
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// No backoff afterwards.
|
||||||
|
doBackoff = false
|
||||||
|
retryCount = 0
|
||||||
|
ls.sendLoads(ctx, stream, first.Clusters[0], interval)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ls *lrsStore) sendLoads(ctx context.Context, stream lrspb.LoadReportingService_StreamLoadStatsClient, clusterName string, interval time.Duration) {
|
||||||
|
tick := time.NewTicker(interval)
|
||||||
|
defer tick.Stop()
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-tick.C:
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := stream.Send(&lrspb.LoadStatsRequest{
|
||||||
|
Node: ls.node,
|
||||||
|
ClusterStats: ls.buildStats(),
|
||||||
|
}); err != nil {
|
||||||
|
grpclog.Infof("lrs: failed to send report: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,240 @@
|
||||||
|
/*
|
||||||
|
*
|
||||||
|
* Copyright 2019 gRPC 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 lrs
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"io"
|
||||||
|
"net"
|
||||||
|
"reflect"
|
||||||
|
"sort"
|
||||||
|
"sync"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/golang/protobuf/proto"
|
||||||
|
durationpb "github.com/golang/protobuf/ptypes/duration"
|
||||||
|
structpb "github.com/golang/protobuf/ptypes/struct"
|
||||||
|
"github.com/google/go-cmp/cmp"
|
||||||
|
"google.golang.org/grpc"
|
||||||
|
basepb "google.golang.org/grpc/balancer/xds/internal/proto/envoy/api/v2/core/base"
|
||||||
|
loadreportpb "google.golang.org/grpc/balancer/xds/internal/proto/envoy/api/v2/endpoint/load_report"
|
||||||
|
lrspb "google.golang.org/grpc/balancer/xds/internal/proto/envoy/service/load_stats/v2/lrs"
|
||||||
|
"google.golang.org/grpc/codes"
|
||||||
|
"google.golang.org/grpc/status"
|
||||||
|
)
|
||||||
|
|
||||||
|
const testService = "grpc.service.test"
|
||||||
|
|
||||||
|
var dropCategories = []string{"drop_for_real", "drop_for_fun"}
|
||||||
|
|
||||||
|
// equalClusterStats sorts requests and clear report internal before comparing.
|
||||||
|
func equalClusterStats(a, b []*loadreportpb.ClusterStats) bool {
|
||||||
|
for _, s := range a {
|
||||||
|
sort.Slice(s.DroppedRequests, func(i, j int) bool {
|
||||||
|
return s.DroppedRequests[i].Category < s.DroppedRequests[j].Category
|
||||||
|
})
|
||||||
|
s.LoadReportInterval = nil
|
||||||
|
}
|
||||||
|
for _, s := range b {
|
||||||
|
sort.Slice(s.DroppedRequests, func(i, j int) bool {
|
||||||
|
return s.DroppedRequests[i].Category < s.DroppedRequests[j].Category
|
||||||
|
})
|
||||||
|
s.LoadReportInterval = nil
|
||||||
|
}
|
||||||
|
return reflect.DeepEqual(a, b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test_lrsStore_buildStats(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
drops []map[string]uint64
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "one report",
|
||||||
|
drops: []map[string]uint64{{
|
||||||
|
dropCategories[0]: 31,
|
||||||
|
dropCategories[1]: 41,
|
||||||
|
}},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "two reports",
|
||||||
|
drops: []map[string]uint64{{
|
||||||
|
dropCategories[0]: 31,
|
||||||
|
dropCategories[1]: 41,
|
||||||
|
}, {
|
||||||
|
dropCategories[0]: 59,
|
||||||
|
dropCategories[1]: 26,
|
||||||
|
}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
ls := newStore(testService)
|
||||||
|
|
||||||
|
for _, ds := range tt.drops {
|
||||||
|
var (
|
||||||
|
totalDropped uint64
|
||||||
|
droppedReqs []*loadreportpb.ClusterStats_DroppedRequests
|
||||||
|
)
|
||||||
|
for cat, count := range ds {
|
||||||
|
totalDropped += count
|
||||||
|
droppedReqs = append(droppedReqs, &loadreportpb.ClusterStats_DroppedRequests{
|
||||||
|
Category: cat,
|
||||||
|
DroppedCount: count,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
want := []*loadreportpb.ClusterStats{
|
||||||
|
{
|
||||||
|
ClusterName: testService,
|
||||||
|
TotalDroppedRequests: totalDropped,
|
||||||
|
DroppedRequests: droppedReqs,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
for c, count := range ds {
|
||||||
|
for i := 0; i < int(count); i++ {
|
||||||
|
wg.Add(1)
|
||||||
|
go func(i int, c string) {
|
||||||
|
ls.callDropped(c)
|
||||||
|
wg.Done()
|
||||||
|
}(i, c)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
wg.Wait()
|
||||||
|
|
||||||
|
if got := ls.buildStats(); !equalClusterStats(got, want) {
|
||||||
|
t.Errorf("lrsStore.buildStats() = %v, want %v", got, want)
|
||||||
|
t.Errorf("%s", cmp.Diff(got, want))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type lrsServer struct {
|
||||||
|
mu sync.Mutex
|
||||||
|
dropTotal uint64
|
||||||
|
drops map[string]uint64
|
||||||
|
reportingInterval *durationpb.Duration
|
||||||
|
}
|
||||||
|
|
||||||
|
func (lrss *lrsServer) StreamLoadStats(stream lrspb.LoadReportingService_StreamLoadStatsServer) error {
|
||||||
|
req, err := stream.Recv()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !proto.Equal(req, &lrspb.LoadStatsRequest{
|
||||||
|
Node: &basepb.Node{
|
||||||
|
Metadata: &structpb.Struct{
|
||||||
|
Fields: map[string]*structpb.Value{
|
||||||
|
grpcHostname: {
|
||||||
|
Kind: &structpb.Value_StringValue{StringValue: testService},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}) {
|
||||||
|
return status.Errorf(codes.FailedPrecondition, "unexpected req: %+v", req)
|
||||||
|
}
|
||||||
|
if err := stream.Send(&lrspb.LoadStatsResponse{
|
||||||
|
Clusters: []string{testService},
|
||||||
|
LoadReportingInterval: lrss.reportingInterval,
|
||||||
|
}); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
for {
|
||||||
|
req, err := stream.Recv()
|
||||||
|
if err != nil {
|
||||||
|
if err == io.EOF {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
stats := req.ClusterStats[0]
|
||||||
|
lrss.mu.Lock()
|
||||||
|
lrss.dropTotal += stats.TotalDroppedRequests
|
||||||
|
for _, d := range stats.DroppedRequests {
|
||||||
|
lrss.drops[d.Category] += d.DroppedCount
|
||||||
|
}
|
||||||
|
lrss.mu.Unlock()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func setupServer(t *testing.T, reportingInterval *durationpb.Duration) (addr string, lrss *lrsServer, cleanup func()) {
|
||||||
|
lis, err := net.Listen("tcp", "localhost:0")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("listen failed due to: %v", err)
|
||||||
|
}
|
||||||
|
svr := grpc.NewServer()
|
||||||
|
lrss = &lrsServer{
|
||||||
|
drops: make(map[string]uint64),
|
||||||
|
reportingInterval: reportingInterval,
|
||||||
|
}
|
||||||
|
lrspb.RegisterLoadReportingServiceServer(svr, lrss)
|
||||||
|
go svr.Serve(lis)
|
||||||
|
return lis.Addr().String(), lrss, func() {
|
||||||
|
svr.Stop()
|
||||||
|
lis.Close()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test_lrsStore_reportTo(t *testing.T) {
|
||||||
|
const intervalNano = 1000 * 1000 * 50
|
||||||
|
addr, lrss, cleanup := setupServer(t, &durationpb.Duration{
|
||||||
|
Seconds: 0,
|
||||||
|
Nanos: intervalNano,
|
||||||
|
})
|
||||||
|
defer cleanup()
|
||||||
|
|
||||||
|
ls := newStore(testService)
|
||||||
|
cc, err := grpc.Dial(addr, grpc.WithInsecure())
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to dial: %v", err)
|
||||||
|
}
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
done := make(chan struct{})
|
||||||
|
go func() {
|
||||||
|
ls.reportTo(ctx, cc)
|
||||||
|
close(done)
|
||||||
|
}()
|
||||||
|
|
||||||
|
drops := map[string]uint64{
|
||||||
|
dropCategories[0]: 31,
|
||||||
|
dropCategories[1]: 41,
|
||||||
|
}
|
||||||
|
|
||||||
|
for c, d := range drops {
|
||||||
|
for i := 0; i < int(d); i++ {
|
||||||
|
ls.callDropped(c)
|
||||||
|
time.Sleep(time.Nanosecond * intervalNano / 10)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
time.Sleep(time.Nanosecond * intervalNano * 2)
|
||||||
|
cancel()
|
||||||
|
<-done
|
||||||
|
|
||||||
|
lrss.mu.Lock()
|
||||||
|
defer lrss.mu.Unlock()
|
||||||
|
if !cmp.Equal(lrss.drops, drops) {
|
||||||
|
t.Errorf("different: %v", cmp.Diff(lrss.drops, drops))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,165 @@
|
||||||
|
/*
|
||||||
|
*
|
||||||
|
* Copyright 2019 gRPC 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 primitives
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
"sync/atomic"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
type incrementUint64Map interface {
|
||||||
|
increment(string)
|
||||||
|
result(string) uint64
|
||||||
|
}
|
||||||
|
|
||||||
|
type mapWithLock struct {
|
||||||
|
mu sync.Mutex
|
||||||
|
m map[string]uint64
|
||||||
|
}
|
||||||
|
|
||||||
|
func newMapWithLock() incrementUint64Map {
|
||||||
|
return &mapWithLock{
|
||||||
|
m: make(map[string]uint64),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (mwl *mapWithLock) increment(c string) {
|
||||||
|
mwl.mu.Lock()
|
||||||
|
mwl.m[c]++
|
||||||
|
mwl.mu.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (mwl *mapWithLock) result(c string) uint64 {
|
||||||
|
return mwl.m[c]
|
||||||
|
}
|
||||||
|
|
||||||
|
type mapWithAtomicFastpath struct {
|
||||||
|
mu sync.RWMutex
|
||||||
|
m map[string]*uint64
|
||||||
|
}
|
||||||
|
|
||||||
|
func newMapWithAtomicFastpath() incrementUint64Map {
|
||||||
|
return &mapWithAtomicFastpath{
|
||||||
|
m: make(map[string]*uint64),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (mwaf *mapWithAtomicFastpath) increment(c string) {
|
||||||
|
mwaf.mu.RLock()
|
||||||
|
if p, ok := mwaf.m[c]; ok {
|
||||||
|
atomic.AddUint64(p, 1)
|
||||||
|
mwaf.mu.RUnlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
mwaf.mu.RUnlock()
|
||||||
|
|
||||||
|
mwaf.mu.Lock()
|
||||||
|
if p, ok := mwaf.m[c]; ok {
|
||||||
|
atomic.AddUint64(p, 1)
|
||||||
|
mwaf.mu.Unlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var temp uint64 = 1
|
||||||
|
mwaf.m[c] = &temp
|
||||||
|
mwaf.mu.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (mwaf *mapWithAtomicFastpath) result(c string) uint64 {
|
||||||
|
return atomic.LoadUint64(mwaf.m[c])
|
||||||
|
}
|
||||||
|
|
||||||
|
type mapWithSyncMap struct {
|
||||||
|
m sync.Map
|
||||||
|
}
|
||||||
|
|
||||||
|
func newMapWithSyncMap() incrementUint64Map {
|
||||||
|
return &mapWithSyncMap{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (mwsm *mapWithSyncMap) increment(c string) {
|
||||||
|
p, ok := mwsm.m.Load(c)
|
||||||
|
if !ok {
|
||||||
|
tp := new(uint64)
|
||||||
|
p, _ = mwsm.m.LoadOrStore(c, tp)
|
||||||
|
}
|
||||||
|
atomic.AddUint64(p.(*uint64), 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (mwsm *mapWithSyncMap) result(c string) uint64 {
|
||||||
|
p, _ := mwsm.m.Load(c)
|
||||||
|
return atomic.LoadUint64(p.(*uint64))
|
||||||
|
}
|
||||||
|
|
||||||
|
func benchmarkIncrementUint64Map(b *testing.B, f func() incrementUint64Map) {
|
||||||
|
const cat = "cat"
|
||||||
|
benches := []struct {
|
||||||
|
name string
|
||||||
|
goroutineCount int
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: " 1",
|
||||||
|
goroutineCount: 1,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: " 10",
|
||||||
|
goroutineCount: 10,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: " 100",
|
||||||
|
goroutineCount: 100,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "1000",
|
||||||
|
goroutineCount: 1000,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, bb := range benches {
|
||||||
|
b.Run(bb.name, func(b *testing.B) {
|
||||||
|
m := f()
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
wg.Add(bb.goroutineCount)
|
||||||
|
b.ResetTimer()
|
||||||
|
for i := 0; i < bb.goroutineCount; i++ {
|
||||||
|
go func() {
|
||||||
|
for j := 0; j < b.N; j++ {
|
||||||
|
m.increment(cat)
|
||||||
|
}
|
||||||
|
wg.Done()
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
wg.Wait()
|
||||||
|
b.StopTimer()
|
||||||
|
if m.result(cat) != uint64(bb.goroutineCount*b.N) {
|
||||||
|
b.Fatalf("result is %d, want %d", m.result(cat), b.N)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func BenchmarkMapWithSyncMutexContetion(b *testing.B) {
|
||||||
|
benchmarkIncrementUint64Map(b, newMapWithLock)
|
||||||
|
}
|
||||||
|
|
||||||
|
func BenchmarkMapWithAtomicFastpath(b *testing.B) {
|
||||||
|
benchmarkIncrementUint64Map(b, newMapWithAtomicFastpath)
|
||||||
|
}
|
||||||
|
|
||||||
|
func BenchmarkMapWithSyncMap(b *testing.B) {
|
||||||
|
benchmarkIncrementUint64Map(b, newMapWithSyncMap)
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue