mirror of
https://github.com/kyverno/kyverno.git
synced 2025-03-09 17:37:12 +00:00
85 lines
2.8 KiB
Go
85 lines
2.8 KiB
Go
package generate
|
|
|
|
import (
|
|
"github.com/golang/glog"
|
|
kyverno "github.com/nirmata/kyverno/pkg/api/kyverno/v1"
|
|
kyvernoclient "github.com/nirmata/kyverno/pkg/client/clientset/versioned"
|
|
"github.com/nirmata/kyverno/pkg/policyStatus"
|
|
)
|
|
|
|
//StatusControlInterface provides interface to update status subresource
|
|
type StatusControlInterface interface {
|
|
Failed(gr kyverno.GenerateRequest, message string, genResources []kyverno.ResourceSpec) error
|
|
Success(gr kyverno.GenerateRequest, genResources []kyverno.ResourceSpec) error
|
|
}
|
|
|
|
// StatusControl is default implementaation of GRStatusControlInterface
|
|
type StatusControl struct {
|
|
client kyvernoclient.Interface
|
|
policyStatus *policyStatus.Sync
|
|
}
|
|
|
|
//Failed sets gr status.state to failed with message
|
|
func (sc StatusControl) Failed(gr kyverno.GenerateRequest, message string, genResources []kyverno.ResourceSpec) error {
|
|
gr.Status.State = kyverno.Failed
|
|
gr.Status.Message = message
|
|
// Update Generated Resources
|
|
gr.Status.GeneratedResources = genResources
|
|
_, err := sc.client.KyvernoV1().GenerateRequests("kyverno").UpdateStatus(&gr)
|
|
if err != nil {
|
|
glog.V(4).Infof("FAILED: updated gr %s status to %s", gr.Name, string(kyverno.Failed))
|
|
return err
|
|
}
|
|
glog.V(4).Infof("updated gr %s status to %s", gr.Name, string(kyverno.Failed))
|
|
return nil
|
|
}
|
|
|
|
// Success sets the gr status.state to completed and clears message
|
|
func (sc StatusControl) Success(gr kyverno.GenerateRequest, genResources []kyverno.ResourceSpec) error {
|
|
oldState := gr.Status.State
|
|
|
|
gr.Status.State = kyverno.Completed
|
|
gr.Status.Message = ""
|
|
// Update Generated Resources
|
|
gr.Status.GeneratedResources = genResources
|
|
|
|
if oldState != kyverno.Completed {
|
|
go func() {
|
|
sc.policyStatus.Listener <- updatePolicyStatusWithGeneratedResourceCount(gr)
|
|
}()
|
|
}
|
|
|
|
_, err := sc.client.KyvernoV1().GenerateRequests("kyverno").UpdateStatus(&gr)
|
|
if err != nil {
|
|
glog.V(4).Infof("FAILED: updated gr %s status to %s", gr.Name, string(kyverno.Completed))
|
|
return err
|
|
}
|
|
glog.V(4).Infof("updated gr %s status to %s", gr.Name, string(kyverno.Completed))
|
|
return nil
|
|
}
|
|
|
|
type generatedResourceCount struct {
|
|
generateRequest kyverno.GenerateRequest
|
|
}
|
|
|
|
func updatePolicyStatusWithGeneratedResourceCount(generateRequest kyverno.GenerateRequest) *generatedResourceCount {
|
|
return &generatedResourceCount{
|
|
generateRequest: generateRequest,
|
|
}
|
|
}
|
|
|
|
func (vc *generatedResourceCount) UpdateStatus(s *policyStatus.Sync) {
|
|
s.Cache.Mutex.Lock()
|
|
status, exist := s.Cache.Data[vc.generateRequest.Spec.Policy]
|
|
if !exist {
|
|
policy, _ := s.PolicyStore.Get(vc.generateRequest.Spec.Policy)
|
|
if policy != nil {
|
|
status = policy.Status
|
|
}
|
|
}
|
|
|
|
status.ResourcesGeneratedCount += len(vc.generateRequest.Status.GeneratedResources)
|
|
|
|
s.Cache.Data[vc.generateRequest.Spec.Policy] = status
|
|
s.Cache.Mutex.Unlock()
|
|
}
|