mirror of
https://github.com/rancher/norman.git
synced 2025-07-05 11:36:46 +00:00
f5744043a6
Problem: Cluster scoped gc was taking significantly longer to complete
than before 7387aa5. This was due to a large number of list calls and
the time needed to iterate over them.
Solution: Instead of requesting every type than Rancher uses, a
seperate map appended to when a cluster scoped handler
is created. The full map of all GroupVersionResources is kept as a
fallback for the use case where user controllers may exist on another
host other than the current leader.
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package resource
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
)
|
|
|
|
//rancherTypes is a set of all types generated by rancher
|
|
//clusterScopedTypes is a set of all types that have been added by a clusterScoped handler
|
|
var (
|
|
rancherTypes = struct {
|
|
sync.RWMutex
|
|
m map[schema.GroupVersionResource]bool
|
|
}{m: make(map[schema.GroupVersionResource]bool)}
|
|
|
|
clusterScopedTypes = struct {
|
|
sync.RWMutex
|
|
m map[schema.GroupVersionResource]bool
|
|
}{m: make(map[schema.GroupVersionResource]bool)}
|
|
)
|
|
|
|
//Get returns a copy of the set of rancherTypes
|
|
func Get() map[schema.GroupVersionResource]bool {
|
|
rancherTypes.RLock()
|
|
defer rancherTypes.RUnlock()
|
|
targetMap := make(map[schema.GroupVersionResource]bool, len(rancherTypes.m))
|
|
for key, value := range rancherTypes.m {
|
|
targetMap[key] = value
|
|
}
|
|
return targetMap
|
|
}
|
|
|
|
func GetClusterScopedTypes() map[schema.GroupVersionResource]bool {
|
|
clusterScopedTypes.Lock()
|
|
defer clusterScopedTypes.Unlock()
|
|
targetMap := make(map[schema.GroupVersionResource]bool, len(clusterScopedTypes.m))
|
|
for key, value := range clusterScopedTypes.m {
|
|
targetMap[key] = value
|
|
}
|
|
return targetMap
|
|
}
|
|
|
|
//Put adds an object to the set and panic on collision
|
|
func Put(key schema.GroupVersionResource) {
|
|
rancherTypes.Lock()
|
|
defer rancherTypes.Unlock()
|
|
_, exists := rancherTypes.m[key]
|
|
if exists {
|
|
//only used in an init function
|
|
panic("key exists in rancherTypes")
|
|
}
|
|
rancherTypes.m[key] = true
|
|
}
|
|
|
|
//PutClusterScoped adds a object that contains a cluster scoped handler to the set
|
|
func PutClusterScoped(key schema.GroupVersionResource) {
|
|
clusterScopedTypes.Lock()
|
|
defer clusterScopedTypes.Unlock()
|
|
clusterScopedTypes.m[key] = true
|
|
}
|