Update heartbeat to extend lease of active workers

internal/base/base.go

@@ -615,7 +615,7 @@ type Lease struct {
 	once sync.Once
 	ch   chan struct{}
 
-	clock timeutil.Clock
+	Clock timeutil.Clock
 
 	mu       sync.Mutex
 	expireAt time.Time // guarded by mu
@@ -625,7 +625,7 @@ func NewLease(expirationTime time.Time) *Lease {
 	return &Lease{
 		ch:       make(chan struct{}),
 		expireAt: expirationTime,
-		clock:    timeutil.NewRealClock(),
+		Clock:    timeutil.NewRealClock(),
 	}
 }
 
@@ -670,7 +670,7 @@ func (l *Lease) Deadline() time.Time {
 // IsValid returns true if the lease's expieration time is in the future or equals to the current time,
 // returns false otherwise.
 func (l *Lease) IsValid() bool {
-	now := l.clock.Now()
+	now := l.Clock.Now()
 	l.mu.Lock()
 	defer l.mu.Unlock()
 	return l.expireAt.After(now) || l.expireAt.Equal(now)

internal/base/base_test.go

@@ -651,7 +651,7 @@ func TestLeaseReset(t *testing.T) {
 	clock := timeutil.NewSimulatedClock(now)
 
 	l := NewLease(now.Add(30 * time.Second))
-	l.clock = clock
+	l.Clock = clock
 
 	// Check initial state
 	if !l.IsValid() {
@@ -686,7 +686,7 @@ func TestLeaseNotifyExpiration(t *testing.T) {
 	clock := timeutil.NewSimulatedClock(now)
 
 	l := NewLease(now.Add(30 * time.Second))
-	l.clock = clock
+	l.Clock = clock
 
 	select {
 	case <-l.Done():