mirror of
https://github.com/FairwindsOps/polaris.git
synced 2026-05-09 10:47:05 +00:00
75f70352ba
**Changes** - Refactored the way controllers work to be an interface - Added configurable controllers to include in scans - Added daemonsets, jobs and cronjobs in scans - Added `ReplicationController` type controllers to the supported list - Adjusted logic for failed YAML parsing to bubble up errors - Added better logic for calculating summaries on cluster wide results - Relocated responsibilities for counting types into validators vs spreading it around more packages - Fixed bug where cronjob parsing was using wrong KIND - Added fixtures for mocking new controller types - Added example yamls to test scanning files - Added functions to NamespacedResult(s) to reduce code complexity deep set iterations - Refactored how results get added to namespacedresults so adding more later is easier - Minor signature changes for interface implementing structs for controllers
70 lines
2.3 KiB
Go
70 lines
2.3 KiB
Go
package controllers
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/fairwindsops/polaris/pkg/config"
|
|
"github.com/fairwindsops/polaris/pkg/kube"
|
|
kubeAPICoreV1 "k8s.io/api/core/v1"
|
|
)
|
|
|
|
// Interface is an interface for k8s controllers (e.g. Deployments and StatefulSets)
|
|
type Interface interface {
|
|
GetName() string
|
|
GetNamespace() string
|
|
GetPodTemplate() *kubeAPICoreV1.PodTemplateSpec
|
|
GetPodSpec() *kubeAPICoreV1.PodSpec
|
|
GetType() config.SupportedController
|
|
}
|
|
|
|
// GenericController is a base implementation with some free methods for inherited structs
|
|
type GenericController struct {
|
|
Name string
|
|
Namespace string
|
|
}
|
|
|
|
// GetName is inherited by all controllers using generic controller to get the name of the controller
|
|
func (g GenericController) GetName() string {
|
|
return g.Name
|
|
}
|
|
|
|
// GetNamespace is inherited by all controllers using generic controller to get the namespace of the controller
|
|
func (g GenericController) GetNamespace() string {
|
|
return g.Namespace
|
|
}
|
|
|
|
// LoadControllersByType loads a list of controllers from the kubeResources by detecting their type
|
|
func LoadControllersByType(controllerType config.SupportedController, kubeResources *kube.ResourceProvider) ([]Interface, error) {
|
|
interfaces := []Interface{}
|
|
switch controllerType {
|
|
case config.Deployments:
|
|
for _, deploy := range kubeResources.Deployments {
|
|
interfaces = append(interfaces, NewDeploymentController(deploy))
|
|
}
|
|
case config.StatefulSets:
|
|
for _, statefulSet := range kubeResources.StatefulSets {
|
|
interfaces = append(interfaces, NewStatefulSetController(statefulSet))
|
|
}
|
|
case config.DaemonSets:
|
|
for _, daemonSet := range kubeResources.DaemonSets {
|
|
interfaces = append(interfaces, NewDaemonSetController(daemonSet))
|
|
}
|
|
case config.Jobs:
|
|
for _, job := range kubeResources.Jobs {
|
|
interfaces = append(interfaces, NewJobController(job))
|
|
}
|
|
case config.CronJobs:
|
|
for _, cronJob := range kubeResources.CronJobs {
|
|
interfaces = append(interfaces, NewCronJobController(cronJob))
|
|
}
|
|
case config.ReplicationControllers:
|
|
for _, replicationController := range kubeResources.ReplicationControllers {
|
|
interfaces = append(interfaces, NewReplicationControllerController(replicationController))
|
|
}
|
|
}
|
|
if len(interfaces) > 0 {
|
|
return interfaces, nil
|
|
}
|
|
return nil, fmt.Errorf("Controller type (%s) does not have a generator", controllerType)
|
|
}
|