kyverno/pkg/controllers/webhook/recorder.go

package webhook

import (
	"sync"
)

type Recorder struct {
	lock       sync.Mutex
	data       map[string]bool
	NotifyChan chan string
}

type StateRecorder interface {
	Ready(string) (bool, bool)
	Record(string)
	Reset()
}

func NewStateRecorder(notifyChan chan string) StateRecorder {
	return &Recorder{
		data:       make(map[string]bool),
		NotifyChan: notifyChan,
	}
}

func (s *Recorder) Ready(key string) (bool, bool) {
	s.lock.Lock()
	defer s.lock.Unlock()
	ready, ok := s.data[key]
	return ready, ok
}

func (s *Recorder) Record(key string) {
	s.lock.Lock()
	defer s.lock.Unlock()
	s.data[key] = true
	if s.NotifyChan != nil {
		s.NotifyChan <- key
	}
}

func (s *Recorder) Reset() {
	s.lock.Lock()
	defer s.lock.Unlock()
	for d := range s.data {
		s.data[d] = false
	}
}