cmd/bpa-operator/vendor/k8s.io/client-go/tools/leaderelection/healthzadaptor.go

   1 /*
   2 Copyright 2015 The Kubernetes Authors.
   3
   4 Licensed under the Apache License, Version 2.0 (the "License");
   5 you may not use this file except in compliance with the License.
   6 You may obtain a copy of the License at
   7
   8     http://www.apache.org/licenses/LICENSE-2.0
   9
  10 Unless required by applicable law or agreed to in writing, software
  11 distributed under the License is distributed on an "AS IS" BASIS,
  12 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 See the License for the specific language governing permissions and
  14 limitations under the License.
  15 */
  16
  17 package leaderelection
  18
  19 import (
  20         "net/http"
  21         "sync"
  22         "time"
  23 )
  24
  25 // HealthzAdaptor associates the /healthz endpoint with the LeaderElection object.
  26 // It helps deal with the /healthz endpoint being set up prior to the LeaderElection.
  27 // This contains the code needed to act as an adaptor between the leader
  28 // election code the health check code. It allows us to provide health
  29 // status about the leader election. Most specifically about if the leader
  30 // has failed to renew without exiting the process. In that case we should
  31 // report not healthy and rely on the kubelet to take down the process.
  32 type HealthzAdaptor struct {
  33         pointerLock sync.Mutex
  34         le          *LeaderElector
  35         timeout     time.Duration
  36 }
  37
  38 // Name returns the name of the health check we are implementing.
  39 func (l *HealthzAdaptor) Name() string {
  40         return "leaderElection"
  41 }
  42
  43 // Check is called by the healthz endpoint handler.
  44 // It fails (returns an error) if we own the lease but had not been able to renew it.
  45 func (l *HealthzAdaptor) Check(req *http.Request) error {
  46         l.pointerLock.Lock()
  47         defer l.pointerLock.Unlock()
  48         if l.le == nil {
  49                 return nil
  50         }
  51         return l.le.Check(l.timeout)
  52 }
  53
  54 // SetLeaderElection ties a leader election object to a HealthzAdaptor
  55 func (l *HealthzAdaptor) SetLeaderElection(le *LeaderElector) {
  56         l.pointerLock.Lock()
  57         defer l.pointerLock.Unlock()
  58         l.le = le
  59 }
  60
  61 // NewLeaderHealthzAdaptor creates a basic healthz adaptor to monitor a leader election.
  62 // timeout determines the time beyond the lease expiry to be allowed for timeout.
  63 // checks within the timeout period after the lease expires will still return healthy.
  64 func NewLeaderHealthzAdaptor(timeout time.Duration) *HealthzAdaptor {
  65         result := &HealthzAdaptor{
  66                 timeout: timeout,
  67         }
  68         return result
  69 }