mirrors/kyverno
mirrors/kyverno
1
0
Fork 0
mirror of https://github.com/kyverno/kyverno.git synced 2025-03-31 03:45:17 +00:00
Code Activity

527 stopCh changes

Browse source
This commit is contained in:
shravan 2020-02-29 17:19:00 +05:30
parent 8c84e9af2b
commit 053ccde6b8
3 changed files with 10 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cmd/kyverno/main.go
View file

@ -248,7 +248,7 @@ func main() {
go grc.Run(1, stopCh) go grc.Run(1, stopCh)
go grcc.Run(1, stopCh) go grcc.Run(1, stopCh)
go pvgen.Run(1, stopCh) go pvgen.Run(1, stopCh)
go statusSync.Run(2) go statusSync.Run(1, stopCh)
// verifys if the admission control is enabled and active // verifys if the admission control is enabled and active
// resync: 60 seconds // resync: 60 seconds

4
pkg/generate/status.go
View file

@ -36,14 +36,14 @@ func (sc StatusControl) Failed(gr kyverno.GenerateRequest, message string, genRe
// Success sets the gr status.state to completed and clears message // Success sets the gr status.state to completed and clears message
func (sc StatusControl) Success(gr kyverno.GenerateRequest, genResources []kyverno.ResourceSpec) error { func (sc StatusControl) Success(gr kyverno.GenerateRequest, genResources []kyverno.ResourceSpec) error {
grCopy := gr.DeepCopy() oldState := gr.Status.State
gr.Status.State = kyverno.Completed gr.Status.State = kyverno.Completed
gr.Status.Message = "" gr.Status.Message = ""
// Update Generated Resources // Update Generated Resources
gr.Status.GeneratedResources = genResources gr.Status.GeneratedResources = genResources
if grCopy.Status.State != kyverno.Completed { if oldState != kyverno.Completed {
go sc.policyStatus.UpdatePolicyStatusWithGeneratedResourceCount(gr) go sc.policyStatus.UpdatePolicyStatusWithGeneratedResourceCount(gr)
} }

16
pkg/policyStatus/0_main.go
View file

@ -18,7 +18,6 @@ import (
type Sync struct { type Sync struct {
cache *cache cache *cache
listener chan statusUpdater listener chan statusUpdater
stop <-chan struct{}
client *versioned.Clientset client *versioned.Clientset
policyStore *policystore.PolicyStore policyStore *policystore.PolicyStore
} }
@ -28,35 +27,34 @@ type cache struct {
data map[string]v1.PolicyStatus data map[string]v1.PolicyStatus
} }
func NewSync(c *versioned.Clientset, sc <-chan struct{}, pms *policystore.PolicyStore) *Sync { func NewSync(c *versioned.Clientset, pms *policystore.PolicyStore) *Sync {
return &Sync{ return &Sync{
cache: &cache{ cache: &cache{
mutex: sync.RWMutex{}, mutex: sync.RWMutex{},
data: make(map[string]v1.PolicyStatus), data: make(map[string]v1.PolicyStatus),
}, },
stop: sc,
client: c, client: c,
policyStore: pms, policyStore: pms,
listener: make(chan statusUpdater), listener: make(chan statusUpdater),
} }
} }
func (s *Sync) Run(workers int) { func (s *Sync) Run(workers int, stopCh <-chan struct{}) {
for i := 0; i < workers; i++ { for i := 0; i < workers; i++ {
go s.updateStatusCache() go s.updateStatusCache(stopCh)
} }
wait.Until(s.updatePolicyStatus, 2*time.Second, s.stop) wait.Until(s.updatePolicyStatus, 2*time.Second, stopCh)
<-s.stop <-stopCh
s.updatePolicyStatus() s.updatePolicyStatus()
} }
func (s *Sync) updateStatusCache() { func (s *Sync) updateStatusCache(stopCh <-chan struct{}) {
for { for {
select { select {
case statusUpdater := <-s.listener: case statusUpdater := <-s.listener:
statusUpdater.updateStatus() statusUpdater.updateStatus()
case <-s.stop: case <-stopCh:
return return
} }
} }