leaderelection: Instrument for when slowpath is exercised

Signed-off-by: Eric Lin <exlin@google.com>

Kubernetes-commit: 1e54c050936be1a1e3e5758718ebca86096dbaea
This commit is contained in:
Eric Lin 2023-11-27 13:10:24 +00:00 committed by Kubernetes Publisher
parent 3c7c00d2d6
commit 2a48f1ee02
2 changed files with 22 additions and 10 deletions

View File

@ -315,6 +315,7 @@ func testTryAcquireOrRenew(t *testing.T, objectType string) {
observedRawRecord: observedRawRecord, observedRawRecord: observedRawRecord,
observedTime: test.observedTime, observedTime: test.observedTime,
clock: clock, clock: clock,
metrics: globalMetricsFactory.newLeaderMetrics(),
} }
if test.expectSuccess != le.tryAcquireOrRenew(context.Background()) { if test.expectSuccess != le.tryAcquireOrRenew(context.Background()) {
if test.retryAfter != 0 { if test.retryAfter != 0 {
@ -491,6 +492,7 @@ func testReleaseLease(t *testing.T, objectType string) {
observedRawRecord: observedRawRecord, observedRawRecord: observedRawRecord,
observedTime: test.observedTime, observedTime: test.observedTime,
clock: clock.RealClock{}, clock: clock.RealClock{},
metrics: globalMetricsFactory.newLeaderMetrics(),
} }
if !le.tryAcquireOrRenew(context.Background()) { if !le.tryAcquireOrRenew(context.Background()) {
t.Errorf("unexpected result of tryAcquireOrRenew: [succeeded=%v]", true) t.Errorf("unexpected result of tryAcquireOrRenew: [succeeded=%v]", true)

View File

@ -26,24 +26,26 @@ import (
type leaderMetricsAdapter interface { type leaderMetricsAdapter interface {
leaderOn(name string) leaderOn(name string)
leaderOff(name string) leaderOff(name string)
slowpathExercised(name string)
} }
// GaugeMetric represents a single numerical value that can arbitrarily go up // LeaderMetric instruments metrics used in leader election.
// and down. type LeaderMetric interface {
type SwitchMetric interface {
On(name string) On(name string)
Off(name string) Off(name string)
SlowpathExercised(name string)
} }
type noopMetric struct{} type noopMetric struct{}
func (noopMetric) On(name string) {} func (noopMetric) On(name string) {}
func (noopMetric) Off(name string) {} func (noopMetric) Off(name string) {}
func (noopMetric) SlowpathExercised(name string) {}
// defaultLeaderMetrics expects the caller to lock before setting any metrics. // defaultLeaderMetrics expects the caller to lock before setting any metrics.
type defaultLeaderMetrics struct { type defaultLeaderMetrics struct {
// leader's value indicates if the current process is the owner of name lease // leader's value indicates if the current process is the owner of name lease
leader SwitchMetric leader LeaderMetric
} }
func (m *defaultLeaderMetrics) leaderOn(name string) { func (m *defaultLeaderMetrics) leaderOn(name string) {
@ -60,19 +62,27 @@ func (m *defaultLeaderMetrics) leaderOff(name string) {
m.leader.Off(name) m.leader.Off(name)
} }
func (m *defaultLeaderMetrics) slowpathExercised(name string) {
if m == nil {
return
}
m.leader.SlowpathExercised(name)
}
type noMetrics struct{} type noMetrics struct{}
func (noMetrics) leaderOn(name string) {} func (noMetrics) leaderOn(name string) {}
func (noMetrics) leaderOff(name string) {} func (noMetrics) leaderOff(name string) {}
func (noMetrics) slowpathExercised(name string) {}
// MetricsProvider generates various metrics used by the leader election. // MetricsProvider generates various metrics used by the leader election.
type MetricsProvider interface { type MetricsProvider interface {
NewLeaderMetric() SwitchMetric NewLeaderMetric() LeaderMetric
} }
type noopMetricsProvider struct{} type noopMetricsProvider struct{}
func (_ noopMetricsProvider) NewLeaderMetric() SwitchMetric { func (noopMetricsProvider) NewLeaderMetric() LeaderMetric {
return noopMetric{} return noopMetric{}
} }