mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-22 19:31:44 +00:00
Add informers to deployment controller
This commit is contained in:
parent
08c2cba266
commit
b838d8ce18
@ -74,6 +74,7 @@ type CMServer struct {
|
|||||||
ConcurrentDSCSyncs int
|
ConcurrentDSCSyncs int
|
||||||
ConcurrentJobSyncs int
|
ConcurrentJobSyncs int
|
||||||
ConcurrentResourceQuotaSyncs int
|
ConcurrentResourceQuotaSyncs int
|
||||||
|
ConcurrentDeploymentSyncs int
|
||||||
ServiceSyncPeriod time.Duration
|
ServiceSyncPeriod time.Duration
|
||||||
NodeSyncPeriod time.Duration
|
NodeSyncPeriod time.Duration
|
||||||
ResourceQuotaSyncPeriod time.Duration
|
ResourceQuotaSyncPeriod time.Duration
|
||||||
@ -116,6 +117,7 @@ func NewCMServer() *CMServer {
|
|||||||
ConcurrentDSCSyncs: 2,
|
ConcurrentDSCSyncs: 2,
|
||||||
ConcurrentJobSyncs: 5,
|
ConcurrentJobSyncs: 5,
|
||||||
ConcurrentResourceQuotaSyncs: 5,
|
ConcurrentResourceQuotaSyncs: 5,
|
||||||
|
ConcurrentDeploymentSyncs: 5,
|
||||||
ServiceSyncPeriod: 5 * time.Minute,
|
ServiceSyncPeriod: 5 * time.Minute,
|
||||||
NodeSyncPeriod: 10 * time.Second,
|
NodeSyncPeriod: 10 * time.Second,
|
||||||
ResourceQuotaSyncPeriod: 5 * time.Minute,
|
ResourceQuotaSyncPeriod: 5 * time.Minute,
|
||||||
@ -189,6 +191,7 @@ func (s *CMServer) AddFlags(fs *pflag.FlagSet) {
|
|||||||
fs.IntVar(&s.ConcurrentEndpointSyncs, "concurrent-endpoint-syncs", s.ConcurrentEndpointSyncs, "The number of endpoint syncing operations that will be done concurrently. Larger number = faster endpoint updating, but more CPU (and network) load")
|
fs.IntVar(&s.ConcurrentEndpointSyncs, "concurrent-endpoint-syncs", s.ConcurrentEndpointSyncs, "The number of endpoint syncing operations that will be done concurrently. Larger number = faster endpoint updating, but more CPU (and network) load")
|
||||||
fs.IntVar(&s.ConcurrentRCSyncs, "concurrent_rc_syncs", s.ConcurrentRCSyncs, "The number of replication controllers that are allowed to sync concurrently. Larger number = more reponsive replica management, but more CPU (and network) load")
|
fs.IntVar(&s.ConcurrentRCSyncs, "concurrent_rc_syncs", s.ConcurrentRCSyncs, "The number of replication controllers that are allowed to sync concurrently. Larger number = more reponsive replica management, but more CPU (and network) load")
|
||||||
fs.IntVar(&s.ConcurrentResourceQuotaSyncs, "concurrent-resource-quota-syncs", s.ConcurrentResourceQuotaSyncs, "The number of resource quotas that are allowed to sync concurrently. Larger number = more responsive quota management, but more CPU (and network) load")
|
fs.IntVar(&s.ConcurrentResourceQuotaSyncs, "concurrent-resource-quota-syncs", s.ConcurrentResourceQuotaSyncs, "The number of resource quotas that are allowed to sync concurrently. Larger number = more responsive quota management, but more CPU (and network) load")
|
||||||
|
fs.IntVar(&s.ConcurrentDeploymentSyncs, "concurrent-deployment-syncs", s.ConcurrentDeploymentSyncs, "The number of deployment objects that are allowed to sync concurrently. Larger number = more reponsive deployments, but more CPU (and network) load")
|
||||||
fs.DurationVar(&s.ServiceSyncPeriod, "service-sync-period", s.ServiceSyncPeriod, "The period for syncing services with their external load balancers")
|
fs.DurationVar(&s.ServiceSyncPeriod, "service-sync-period", s.ServiceSyncPeriod, "The period for syncing services with their external load balancers")
|
||||||
fs.DurationVar(&s.NodeSyncPeriod, "node-sync-period", s.NodeSyncPeriod, ""+
|
fs.DurationVar(&s.NodeSyncPeriod, "node-sync-period", s.NodeSyncPeriod, ""+
|
||||||
"The period for syncing nodes from cloudprovider. Longer periods will result in "+
|
"The period for syncing nodes from cloudprovider. Longer periods will result in "+
|
||||||
@ -383,8 +386,8 @@ func (s *CMServer) Run(_ []string) error {
|
|||||||
|
|
||||||
if containsResource(resources, "deployments") {
|
if containsResource(resources, "deployments") {
|
||||||
glog.Infof("Starting deployment controller")
|
glog.Infof("Starting deployment controller")
|
||||||
deployment.New(clientForUserAgentOrDie(*kubeconfig, "deployment-controller")).
|
deployment.NewDeploymentController(clientForUserAgentOrDie(*kubeconfig, "deployment-controller")).
|
||||||
Run(s.DeploymentControllerSyncPeriod)
|
Run(s.ConcurrentDeploymentSyncs, util.NeverStop)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -235,6 +235,7 @@ function start_apiserver {
|
|||||||
APISERVER_LOG=/tmp/kube-apiserver.log
|
APISERVER_LOG=/tmp/kube-apiserver.log
|
||||||
sudo -E "${GO_OUT}/kube-apiserver" ${priv_arg} ${runtime_config}\
|
sudo -E "${GO_OUT}/kube-apiserver" ${priv_arg} ${runtime_config}\
|
||||||
--v=${LOG_LEVEL} \
|
--v=${LOG_LEVEL} \
|
||||||
|
--runtime-config=experimental/v1=true \
|
||||||
--cert-dir="${CERT_DIR}" \
|
--cert-dir="${CERT_DIR}" \
|
||||||
--service-account-key-file="${SERVICE_ACCOUNT_KEY}" \
|
--service-account-key-file="${SERVICE_ACCOUNT_KEY}" \
|
||||||
--service-account-lookup="${SERVICE_ACCOUNT_LOOKUP}" \
|
--service-account-lookup="${SERVICE_ACCOUNT_LOOKUP}" \
|
||||||
@ -255,6 +256,7 @@ function start_controller_manager {
|
|||||||
CTLRMGR_LOG=/tmp/kube-controller-manager.log
|
CTLRMGR_LOG=/tmp/kube-controller-manager.log
|
||||||
sudo -E "${GO_OUT}/kube-controller-manager" \
|
sudo -E "${GO_OUT}/kube-controller-manager" \
|
||||||
--v=${LOG_LEVEL} \
|
--v=${LOG_LEVEL} \
|
||||||
|
--enable-deployment-controller \
|
||||||
--service-account-private-key-file="${SERVICE_ACCOUNT_KEY}" \
|
--service-account-private-key-file="${SERVICE_ACCOUNT_KEY}" \
|
||||||
--root-ca-file="${ROOT_CA_FILE}" \
|
--root-ca-file="${ROOT_CA_FILE}" \
|
||||||
--enable-hostpath-provisioner="${ENABLE_HOSTPATH_PROVISIONER}" \
|
--enable-hostpath-provisioner="${ENABLE_HOSTPATH_PROVISIONER}" \
|
||||||
|
@ -45,6 +45,7 @@ cluster-dns
|
|||||||
cluster-domain
|
cluster-domain
|
||||||
cluster-name
|
cluster-name
|
||||||
cluster-tag
|
cluster-tag
|
||||||
|
concurrent-deployment-syncs
|
||||||
concurrent-endpoint-syncs
|
concurrent-endpoint-syncs
|
||||||
concurrent-resource-quota-syncs
|
concurrent-resource-quota-syncs
|
||||||
config-sync-period
|
config-sync-period
|
||||||
|
54
pkg/client/cache/listers.go
vendored
54
pkg/client/cache/listers.go
vendored
@ -191,6 +191,60 @@ func (s *StoreToReplicationControllerLister) GetPodControllers(pod *api.Pod) (co
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// StoreToDeploymentLister gives a store List and Exists methods. The store must contain only Deployments.
|
||||||
|
type StoreToDeploymentLister struct {
|
||||||
|
Store
|
||||||
|
}
|
||||||
|
|
||||||
|
// Exists checks if the given deployment exists in the store.
|
||||||
|
func (s *StoreToDeploymentLister) Exists(deployment *extensions.Deployment) (bool, error) {
|
||||||
|
_, exists, err := s.Store.Get(deployment)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
return exists, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// StoreToDeploymentLister lists all deployments in the store.
|
||||||
|
// TODO: converge on the interface in pkg/client
|
||||||
|
func (s *StoreToDeploymentLister) List() (deployments []extensions.Deployment, err error) {
|
||||||
|
for _, c := range s.Store.List() {
|
||||||
|
deployments = append(deployments, *(c.(*extensions.Deployment)))
|
||||||
|
}
|
||||||
|
return deployments, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetDeploymentsForRC returns a list of deployments managing a replication controller. Returns an error only if no matching deployments are found.
|
||||||
|
func (s *StoreToDeploymentLister) GetDeploymentsForRC(rc *api.ReplicationController) (deployments []extensions.Deployment, err error) {
|
||||||
|
var selector labels.Selector
|
||||||
|
var d extensions.Deployment
|
||||||
|
|
||||||
|
if len(rc.Labels) == 0 {
|
||||||
|
err = fmt.Errorf("No controllers found for replication controller %v because it has no labels", rc.Name)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: MODIFY THIS METHOD so that it checks for the podTemplateSpecHash label
|
||||||
|
for _, m := range s.Store.List() {
|
||||||
|
d = *m.(*extensions.Deployment)
|
||||||
|
if d.Namespace != rc.Namespace {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
labelSet := labels.Set(d.Spec.Selector)
|
||||||
|
selector = labels.Set(d.Spec.Selector).AsSelector()
|
||||||
|
|
||||||
|
// If an rc with a nil or empty selector creeps in, it should match nothing, not everything.
|
||||||
|
if labelSet.AsSelector().Empty() || !selector.Matches(labels.Set(rc.Labels)) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
deployments = append(deployments, d)
|
||||||
|
}
|
||||||
|
if len(deployments) == 0 {
|
||||||
|
err = fmt.Errorf("Could not find deployments set for replication controller %s in namespace %s with labels: %v", rc.Name, rc.Namespace, rc.Labels)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// StoreToDaemonSetLister gives a store List and Exists methods. The store must contain only DaemonSets.
|
// StoreToDaemonSetLister gives a store List and Exists methods. The store must contain only DaemonSets.
|
||||||
type StoreToDaemonSetLister struct {
|
type StoreToDaemonSetLister struct {
|
||||||
Store
|
Store
|
||||||
|
@ -223,6 +223,34 @@ func NewControllerExpectations() *ControllerExpectations {
|
|||||||
return &ControllerExpectations{cache.NewTTLStore(ExpKeyFunc, ExpectationsTimeout)}
|
return &ControllerExpectations{cache.NewTTLStore(ExpKeyFunc, ExpectationsTimeout)}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RCControlInterface is an interface that knows how to add or delete
|
||||||
|
// replication controllers, as well as increment or decrement them. It is used
|
||||||
|
// by the deployment controller to ease testing of actions that it takes.
|
||||||
|
// @TODO Decide if we want to use this or use the kube client directly.
|
||||||
|
type RCControlInterface interface {
|
||||||
|
// CreateRC creates a new replication controller.
|
||||||
|
CreateRC(controller *api.ReplicationController) (*api.ReplicationController, error)
|
||||||
|
// ChangeReplicaCount increments or decrements a replication controller by
|
||||||
|
// a given amount. For decrementing, use negative numbers.
|
||||||
|
AdjustReplicaCount(controller *api.ReplicationController, count int) error
|
||||||
|
}
|
||||||
|
|
||||||
|
// RealPodControl is the default implementation of PodControllerInterface.
|
||||||
|
type RealRCControl struct {
|
||||||
|
KubeClient client.Interface
|
||||||
|
Recorder record.EventRecorder
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r RealRCControl) CreateRC(controller *api.ReplicationController) (*api.ReplicationController, error) {
|
||||||
|
return r.KubeClient.ReplicationControllers(controller.Namespace).Create(controller)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r RealRCControl) AdjustReplicaCount(controller *api.ReplicationController, count int) error {
|
||||||
|
controller.Spec.Replicas += count
|
||||||
|
_, err := r.KubeClient.ReplicationControllers(controller.Namespace).Update(controller)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
// PodControlInterface is an interface that knows how to add or delete pods
|
// PodControlInterface is an interface that knows how to add or delete pods
|
||||||
// created as an interface to allow testing.
|
// created as an interface to allow testing.
|
||||||
type PodControlInterface interface {
|
type PodControlInterface interface {
|
||||||
|
@ -22,77 +22,308 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/golang/glog"
|
"github.com/golang/glog"
|
||||||
|
|
||||||
"k8s.io/kubernetes/pkg/api"
|
"k8s.io/kubernetes/pkg/api"
|
||||||
"k8s.io/kubernetes/pkg/apis/extensions"
|
"k8s.io/kubernetes/pkg/apis/extensions"
|
||||||
|
"k8s.io/kubernetes/pkg/client/cache"
|
||||||
"k8s.io/kubernetes/pkg/client/record"
|
"k8s.io/kubernetes/pkg/client/record"
|
||||||
client "k8s.io/kubernetes/pkg/client/unversioned"
|
client "k8s.io/kubernetes/pkg/client/unversioned"
|
||||||
|
"k8s.io/kubernetes/pkg/controller"
|
||||||
|
"k8s.io/kubernetes/pkg/controller/framework"
|
||||||
|
"k8s.io/kubernetes/pkg/fields"
|
||||||
|
"k8s.io/kubernetes/pkg/labels"
|
||||||
|
"k8s.io/kubernetes/pkg/runtime"
|
||||||
"k8s.io/kubernetes/pkg/util"
|
"k8s.io/kubernetes/pkg/util"
|
||||||
deploymentutil "k8s.io/kubernetes/pkg/util/deployment"
|
deploymentutil "k8s.io/kubernetes/pkg/util/deployment"
|
||||||
|
"k8s.io/kubernetes/pkg/util/workqueue"
|
||||||
|
"k8s.io/kubernetes/pkg/watch"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// We'll attempt to recompute the required replicas of all deployments
|
||||||
|
// that have fulfilled their expectations at least this often. This recomputation
|
||||||
|
// happens based on contents in the local caches.
|
||||||
|
FullDeploymentResyncPeriod = 30 * time.Second
|
||||||
|
|
||||||
|
// We'll keep replication controller watches open up to this long. In the unlikely case
|
||||||
|
// that a watch misdelivers info about an RC, it'll take this long for
|
||||||
|
// that mistake to be rectified.
|
||||||
|
ControllerRelistPeriod = 5 * time.Minute
|
||||||
|
|
||||||
|
// We'll keep pod watches open up to this long. In the unlikely case
|
||||||
|
// that a watch misdelivers info about a pod, it'll take this long for
|
||||||
|
// that mistake to be rectified.
|
||||||
|
PodRelistPeriod = 5 * time.Minute
|
||||||
)
|
)
|
||||||
|
|
||||||
type DeploymentController struct {
|
type DeploymentController struct {
|
||||||
client client.Interface
|
client client.Interface
|
||||||
expClient client.ExtensionsInterface
|
expClient client.ExtensionsInterface
|
||||||
eventRecorder record.EventRecorder
|
eventRecorder record.EventRecorder
|
||||||
|
rcControl controller.RCControlInterface
|
||||||
|
|
||||||
|
// To allow injection of syncDeployment for testing.
|
||||||
|
syncHandler func(dKey string) error
|
||||||
|
|
||||||
|
// A store of deployments, populated by the dController
|
||||||
|
dStore cache.StoreToDeploymentLister
|
||||||
|
// Watches changes to all deployments
|
||||||
|
dController *framework.Controller
|
||||||
|
// A store of replication controllers, populated by the rcController
|
||||||
|
rcStore cache.StoreToReplicationControllerLister
|
||||||
|
// Watches changes to all replication controllers
|
||||||
|
rcController *framework.Controller
|
||||||
|
// rcStoreSynced returns true if the RC store has been synced at least once.
|
||||||
|
// Added as a member to the struct to allow injection for testing.
|
||||||
|
rcStoreSynced func() bool
|
||||||
|
// A store of pods, populated by the podController
|
||||||
|
podStore cache.StoreToPodLister
|
||||||
|
// Watches changes to all pods
|
||||||
|
podController *framework.Controller
|
||||||
|
// podStoreSynced returns true if the pod store has been synced at least once.
|
||||||
|
// Added as a member to the struct to allow injection for testing.
|
||||||
|
podStoreSynced func() bool
|
||||||
|
|
||||||
|
// Deployments that need to be synced
|
||||||
|
queue *workqueue.Type
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(client client.Interface) *DeploymentController {
|
func NewDeploymentController(client client.Interface) *DeploymentController {
|
||||||
eventBroadcaster := record.NewBroadcaster()
|
eventBroadcaster := record.NewBroadcaster()
|
||||||
eventBroadcaster.StartLogging(glog.Infof)
|
eventBroadcaster.StartLogging(glog.Infof)
|
||||||
eventBroadcaster.StartRecordingToSink(client.Events(""))
|
eventBroadcaster.StartRecordingToSink(client.Events(""))
|
||||||
|
|
||||||
return &DeploymentController{
|
dc := &DeploymentController{
|
||||||
client: client,
|
client: client,
|
||||||
expClient: client.Extensions(),
|
expClient: client.Extensions(),
|
||||||
eventRecorder: eventBroadcaster.NewRecorder(api.EventSource{Component: "deployment-controller"}),
|
eventRecorder: eventBroadcaster.NewRecorder(api.EventSource{Component: "deployment-controller"}),
|
||||||
|
queue: workqueue.New(),
|
||||||
|
}
|
||||||
|
|
||||||
|
dc.dStore.Store, dc.dController = framework.NewInformer(
|
||||||
|
&cache.ListWatch{
|
||||||
|
ListFunc: func() (runtime.Object, error) {
|
||||||
|
return dc.expClient.Deployments(api.NamespaceAll).List(labels.Everything(), fields.Everything())
|
||||||
|
},
|
||||||
|
WatchFunc: func(options api.ListOptions) (watch.Interface, error) {
|
||||||
|
return dc.expClient.Deployments(api.NamespaceAll).Watch(labels.Everything(), fields.Everything(), options)
|
||||||
|
},
|
||||||
|
},
|
||||||
|
&extensions.Deployment{},
|
||||||
|
FullDeploymentResyncPeriod,
|
||||||
|
framework.ResourceEventHandlerFuncs{
|
||||||
|
AddFunc: dc.enqueueDeployment,
|
||||||
|
UpdateFunc: func(old, cur interface{}) {
|
||||||
|
// Resync on deployment object relist.
|
||||||
|
dc.enqueueDeployment(cur)
|
||||||
|
},
|
||||||
|
// This will enter the sync loop and no-op, because the deployment has been deleted from the store.
|
||||||
|
// Note that deleting a controller immediately after scaling it to 0 will not work. The recommended
|
||||||
|
// way of achieving this is by performing a `stop` operation on the deployment.
|
||||||
|
DeleteFunc: dc.enqueueDeployment,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
dc.rcStore.Store, dc.rcController = framework.NewInformer(
|
||||||
|
&cache.ListWatch{
|
||||||
|
ListFunc: func() (runtime.Object, error) {
|
||||||
|
return dc.client.ReplicationControllers(api.NamespaceAll).List(labels.Everything(), fields.Everything())
|
||||||
|
},
|
||||||
|
WatchFunc: func(options api.ListOptions) (watch.Interface, error) {
|
||||||
|
return dc.client.ReplicationControllers(api.NamespaceAll).Watch(labels.Everything(), fields.Everything(), options)
|
||||||
|
},
|
||||||
|
},
|
||||||
|
&api.ReplicationController{},
|
||||||
|
ControllerRelistPeriod,
|
||||||
|
framework.ResourceEventHandlerFuncs{
|
||||||
|
AddFunc: dc.addRC,
|
||||||
|
UpdateFunc: dc.updateRC,
|
||||||
|
DeleteFunc: dc.deleteRC,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
// We do not event on anything from the podController, but we use the local
|
||||||
|
// podStore to make queries about the current state of pods (e.g. whether
|
||||||
|
// they are ready or not) more efficient.
|
||||||
|
dc.podStore.Store, dc.podController = framework.NewInformer(
|
||||||
|
&cache.ListWatch{
|
||||||
|
ListFunc: func() (runtime.Object, error) {
|
||||||
|
return dc.client.Pods(api.NamespaceAll).List(labels.Everything(), fields.Everything())
|
||||||
|
},
|
||||||
|
WatchFunc: func(options api.ListOptions) (watch.Interface, error) {
|
||||||
|
return dc.client.Pods(api.NamespaceAll).Watch(labels.Everything(), fields.Everything(), options)
|
||||||
|
},
|
||||||
|
},
|
||||||
|
&api.Pod{},
|
||||||
|
PodRelistPeriod,
|
||||||
|
framework.ResourceEventHandlerFuncs{},
|
||||||
|
)
|
||||||
|
|
||||||
|
dc.syncHandler = dc.syncDeployment
|
||||||
|
return dc
|
||||||
|
}
|
||||||
|
|
||||||
|
// When an RC is created, enqueue the deployment that manages it.
|
||||||
|
func (dc *DeploymentController) addRC(obj interface{}) {
|
||||||
|
rc := obj.(*api.ReplicationController)
|
||||||
|
if d := dc.getDeploymentForRC(rc); rc != nil {
|
||||||
|
dc.enqueueDeployment(d)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) Run(syncPeriod time.Duration) {
|
// getDeploymentForRC returns the deployment managing the given RC.
|
||||||
go util.Until(func() {
|
// TODO: Surface that we are ignoring multiple deployments for a given controller.
|
||||||
errs := d.reconcileDeployments()
|
func (dc *DeploymentController) getDeploymentForRC(rc *api.ReplicationController) *extensions.Deployment {
|
||||||
for _, err := range errs {
|
deployments, err := dc.dStore.GetDeploymentsForRC(rc)
|
||||||
glog.Errorf("Failed to reconcile: %v", err)
|
|
||||||
}
|
|
||||||
}, syncPeriod, util.NeverStop)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *DeploymentController) reconcileDeployments() []error {
|
|
||||||
list, err := d.expClient.Deployments(api.NamespaceAll).List(api.ListOptions{})
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return []error{fmt.Errorf("error listing deployments: %v", err)}
|
glog.V(4).Infof("No deployments found for replication controller %v, deployment controller will avoid syncing", rc.Name)
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
errs := []error{}
|
// Because all RC's belonging to a deployment should have a unique label key,
|
||||||
for _, deployment := range list.Items {
|
// there should never be more than one deployment returned by the above method.
|
||||||
if err := d.reconcileDeployment(&deployment); err != nil {
|
// If that happens we should probably dynamically repair the situation by ultimately
|
||||||
errs = append(errs, fmt.Errorf("error in reconciling deployment %s: %v", deployment.Name, err))
|
// trying to clean up one of the controllers, for now we just return one of the two,
|
||||||
|
// likely randomly.
|
||||||
|
return &deployments[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
// When a controller is updated, figure out what deployment/s manage it and wake them
|
||||||
|
// up. If the labels of the controller have changed we need to awaken both the old
|
||||||
|
// and new deployments. old and cur must be *api.ReplicationController types.
|
||||||
|
func (dc *DeploymentController) updateRC(old, cur interface{}) {
|
||||||
|
if api.Semantic.DeepEqual(old, cur) {
|
||||||
|
// A periodic relist will send update events for all known controllers.
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// TODO: Write a unittest for this case
|
||||||
|
curRC := cur.(*api.ReplicationController)
|
||||||
|
if d := dc.getDeploymentForRC(curRC); d != nil {
|
||||||
|
dc.enqueueDeployment(d)
|
||||||
|
}
|
||||||
|
// A number of things could affect the old deployment: labels changing,
|
||||||
|
// pod template changing, etc.
|
||||||
|
oldRC := old.(*api.ReplicationController)
|
||||||
|
// TODO: Is this the right way to check this, or is checking names sufficient?
|
||||||
|
if !api.Semantic.DeepEqual(oldRC, curRC) {
|
||||||
|
if oldD := dc.getDeploymentForRC(oldRC); oldD != nil {
|
||||||
|
dc.enqueueDeployment(oldD)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return errs
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) reconcileDeployment(deployment *extensions.Deployment) error {
|
// When a controller is deleted, enqueue the deployment that manages it.
|
||||||
switch deployment.Spec.Strategy.Type {
|
// obj could be an *api.ReplicationController, or a DeletionFinalStateUnknown
|
||||||
case extensions.RecreateDeploymentStrategyType:
|
// marker item.
|
||||||
return d.reconcileRecreateDeployment(*deployment)
|
func (dc *DeploymentController) deleteRC(obj interface{}) {
|
||||||
case extensions.RollingUpdateDeploymentStrategyType:
|
rc, ok := obj.(*api.ReplicationController)
|
||||||
return d.reconcileRollingUpdateDeployment(*deployment)
|
|
||||||
|
// When a delete is dropped, the relist will notice a pod in the store not
|
||||||
|
// in the list, leading to the insertion of a tombstone object which contains
|
||||||
|
// the deleted key/value. Note that this value might be stale. If the RC
|
||||||
|
// changed labels the new deployment will not be woken up till the periodic resync.
|
||||||
|
if !ok {
|
||||||
|
tombstone, ok := obj.(cache.DeletedFinalStateUnknown)
|
||||||
|
if !ok {
|
||||||
|
glog.Errorf("Couldn't get object from tombstone %+v, could take up to %v before a deployment recreates/updates controllers", obj, FullDeploymentResyncPeriod)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
rc, ok = tombstone.Obj.(*api.ReplicationController)
|
||||||
|
if !ok {
|
||||||
|
glog.Errorf("Tombstone contained object that is not an rc %+v, could take up to %v before a deployment recreates/updates controllers", obj, FullDeploymentResyncPeriod)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if d := dc.getDeploymentForRC(rc); d != nil {
|
||||||
|
dc.enqueueDeployment(d)
|
||||||
}
|
}
|
||||||
return fmt.Errorf("unexpected deployment strategy type: %s", deployment.Spec.Strategy.Type)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) reconcileRecreateDeployment(deployment extensions.Deployment) error {
|
// obj could be an *api.Deployment, or a DeletionFinalStateUnknown marker item.
|
||||||
|
func (dc *DeploymentController) enqueueDeployment(obj interface{}) {
|
||||||
|
key, err := controller.KeyFunc(obj)
|
||||||
|
if err != nil {
|
||||||
|
glog.Errorf("Couldn't get key for object %+v: %v", obj, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: Handle overlapping deployments better. Either disallow them at admission time or
|
||||||
|
// deterministically avoid syncing deployments that fight over RC's. Currently, we only
|
||||||
|
// ensure that the same deployment is synced for a given RC. When we periodically relist
|
||||||
|
// all deployments there will still be some RC instability. One way to handle this is
|
||||||
|
// by querying the store for all deployments that this deployment overlaps, as well as all
|
||||||
|
// deployments that overlap this deployments, and sorting them.
|
||||||
|
dc.queue.Add(key)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *DeploymentController) Run(workers int, stopCh <-chan struct{}) {
|
||||||
|
defer util.HandleCrash()
|
||||||
|
go dc.dController.Run(stopCh)
|
||||||
|
go dc.rcController.Run(stopCh)
|
||||||
|
go dc.podController.Run(stopCh)
|
||||||
|
for i := 0; i < workers; i++ {
|
||||||
|
go util.Until(dc.worker, time.Second, stopCh)
|
||||||
|
}
|
||||||
|
<-stopCh
|
||||||
|
glog.Infof("Shutting down deployment controller")
|
||||||
|
dc.queue.ShutDown()
|
||||||
|
}
|
||||||
|
|
||||||
|
// worker runs a worker thread that just dequeues items, processes them, and marks them done.
|
||||||
|
// It enforces that the syncHandler is never invoked concurrently with the same key.
|
||||||
|
func (dc *DeploymentController) worker() {
|
||||||
|
for {
|
||||||
|
func() {
|
||||||
|
key, quit := dc.queue.Get()
|
||||||
|
if quit {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer dc.queue.Done(key)
|
||||||
|
err := dc.syncHandler(key.(string))
|
||||||
|
if err != nil {
|
||||||
|
glog.Errorf("Error syncing deployment: %v", err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *DeploymentController) syncDeployment(key string) error {
|
||||||
|
startTime := time.Now()
|
||||||
|
defer func() {
|
||||||
|
glog.V(4).Infof("Finished syncing deployment %q (%v)", key, time.Now().Sub(startTime))
|
||||||
|
}()
|
||||||
|
|
||||||
|
obj, exists, err := dc.dStore.Store.GetByKey(key)
|
||||||
|
if !exists {
|
||||||
|
glog.Infof("Deployment has been deleted %v", key)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
glog.Infof("Unable to retrieve deployment %v from store: %v", key, err)
|
||||||
|
dc.queue.Add(key)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d := *obj.(*extensions.Deployment)
|
||||||
|
switch d.Spec.Strategy.Type {
|
||||||
|
case extensions.RecreateDeploymentStrategyType:
|
||||||
|
return dc.syncRecreateDeployment(d)
|
||||||
|
case extensions.RollingUpdateDeploymentStrategyType:
|
||||||
|
return dc.syncRollingUpdateDeployment(d)
|
||||||
|
}
|
||||||
|
return fmt.Errorf("Unexpected deployment strategy type: %s", d.Spec.Strategy.Type)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *DeploymentController) syncRecreateDeployment(deployment extensions.Deployment) error {
|
||||||
// TODO: implement me.
|
// TODO: implement me.
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) reconcileRollingUpdateDeployment(deployment extensions.Deployment) error {
|
func (dc *DeploymentController) syncRollingUpdateDeployment(deployment extensions.Deployment) error {
|
||||||
newRC, err := d.getNewRC(deployment)
|
newRC, err := dc.getNewRC(deployment)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
oldRCs, err := d.getOldRCs(deployment)
|
oldRCs, err := dc.getOldRCs(deployment)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@ -100,36 +331,32 @@ func (d *DeploymentController) reconcileRollingUpdateDeployment(deployment exten
|
|||||||
allRCs := append(oldRCs, newRC)
|
allRCs := append(oldRCs, newRC)
|
||||||
|
|
||||||
// Scale up, if we can.
|
// Scale up, if we can.
|
||||||
scaledUp, err := d.reconcileNewRC(allRCs, newRC, deployment)
|
scaledUp, err := dc.reconcileNewRC(allRCs, newRC, deployment)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if scaledUp {
|
if scaledUp {
|
||||||
// Update DeploymentStatus
|
// Update DeploymentStatus
|
||||||
return d.updateDeploymentStatus(allRCs, newRC, deployment)
|
return dc.updateDeploymentStatus(allRCs, newRC, deployment)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Scale down, if we can.
|
// Scale down, if we can.
|
||||||
scaledDown, err := d.reconcileOldRCs(allRCs, oldRCs, newRC, deployment)
|
scaledDown, err := dc.reconcileOldRCs(allRCs, oldRCs, newRC, deployment)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if scaledDown {
|
if scaledDown {
|
||||||
// Update DeploymentStatus
|
// Update DeploymentStatus
|
||||||
return d.updateDeploymentStatus(allRCs, newRC, deployment)
|
return dc.updateDeploymentStatus(allRCs, newRC, deployment)
|
||||||
}
|
}
|
||||||
// TODO: raise an event, neither scaled up nor down.
|
// TODO: raise an event, neither scaled up nor down.
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) getOldRCs(deployment extensions.Deployment) ([]*api.ReplicationController, error) {
|
|
||||||
return deploymentutil.GetOldRCs(deployment, d.client)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Returns an RC that matches the intent of the given deployment.
|
// Returns an RC that matches the intent of the given deployment.
|
||||||
// It creates a new RC if required.
|
// It creates a new RC if required.
|
||||||
func (d *DeploymentController) getNewRC(deployment extensions.Deployment) (*api.ReplicationController, error) {
|
func (dc *DeploymentController) getNewRC(deployment extensions.Deployment) (*api.ReplicationController, error) {
|
||||||
existingNewRC, err := deploymentutil.GetNewRC(deployment, d.client)
|
existingNewRC, err := deploymentutil.GetNewRC(deployment, dc.client)
|
||||||
if err != nil || existingNewRC != nil {
|
if err != nil || existingNewRC != nil {
|
||||||
return existingNewRC, err
|
return existingNewRC, err
|
||||||
}
|
}
|
||||||
@ -151,21 +378,55 @@ func (d *DeploymentController) getNewRC(deployment extensions.Deployment) (*api.
|
|||||||
Template: &newRCTemplate,
|
Template: &newRCTemplate,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
createdRC, err := d.client.ReplicationControllers(namespace).Create(&newRC)
|
createdRC, err := dc.client.ReplicationControllers(namespace).Create(&newRC)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("error creating replication controller: %v", err)
|
return nil, fmt.Errorf("error creating replication controller: %v", err)
|
||||||
}
|
}
|
||||||
return createdRC, nil
|
return createdRC, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) reconcileNewRC(allRCs []*api.ReplicationController, newRC *api.ReplicationController, deployment extensions.Deployment) (bool, error) {
|
func (dc *DeploymentController) getOldRCs(deployment extensions.Deployment) ([]*api.ReplicationController, error) {
|
||||||
|
// TODO: (janet) HEAD >>> return deploymentutil.GetOldRCs(deployment, d.client)
|
||||||
|
namespace := deployment.ObjectMeta.Namespace
|
||||||
|
// 1. Find all pods whose labels match deployment.Spec.Selector
|
||||||
|
podList, err := dc.podStore.Pods(api.NamespaceAll).List(labels.SelectorFromSet(deployment.Spec.Selector))
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error listing pods: %v", err)
|
||||||
|
}
|
||||||
|
// 2. Find the corresponding RCs for pods in podList.
|
||||||
|
oldRCs := map[string]api.ReplicationController{}
|
||||||
|
rcList, err := dc.rcStore.List()
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error listing replication controllers: %v", err)
|
||||||
|
}
|
||||||
|
for _, pod := range podList.Items {
|
||||||
|
podLabelsSelector := labels.Set(pod.ObjectMeta.Labels)
|
||||||
|
for _, rc := range rcList {
|
||||||
|
rcLabelsSelector := labels.SelectorFromSet(rc.Spec.Selector)
|
||||||
|
if rcLabelsSelector.Matches(podLabelsSelector) {
|
||||||
|
// Filter out RC that has the same pod template spec as the deployment - that is the new RC.
|
||||||
|
if api.Semantic.DeepEqual(rc.Spec.Template, deploymentutil.GetNewRCTemplate(deployment)) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
oldRCs[rc.ObjectMeta.Name] = rc
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
rcSlice := []*api.ReplicationController{}
|
||||||
|
for _, value := range oldRCs {
|
||||||
|
rcSlice = append(rcSlice, &value)
|
||||||
|
}
|
||||||
|
return rcSlice, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *DeploymentController) reconcileNewRC(allRCs []*api.ReplicationController, newRC *api.ReplicationController, deployment extensions.Deployment) (bool, error) {
|
||||||
if newRC.Spec.Replicas == deployment.Spec.Replicas {
|
if newRC.Spec.Replicas == deployment.Spec.Replicas {
|
||||||
// Scaling not required.
|
// Scaling not required.
|
||||||
return false, nil
|
return false, nil
|
||||||
}
|
}
|
||||||
if newRC.Spec.Replicas > deployment.Spec.Replicas {
|
if newRC.Spec.Replicas > deployment.Spec.Replicas {
|
||||||
// Scale down.
|
// Scale down.
|
||||||
_, err := d.scaleRCAndRecordEvent(newRC, deployment.Spec.Replicas, deployment)
|
_, err := dc.scaleRCAndRecordEvent(newRC, deployment.Spec.Replicas, deployment)
|
||||||
return true, err
|
return true, err
|
||||||
}
|
}
|
||||||
// Check if we can scale up.
|
// Check if we can scale up.
|
||||||
@ -188,11 +449,11 @@ func (d *DeploymentController) reconcileNewRC(allRCs []*api.ReplicationControlle
|
|||||||
// Do not exceed the number of desired replicas.
|
// Do not exceed the number of desired replicas.
|
||||||
scaleUpCount = int(math.Min(float64(scaleUpCount), float64(deployment.Spec.Replicas-newRC.Spec.Replicas)))
|
scaleUpCount = int(math.Min(float64(scaleUpCount), float64(deployment.Spec.Replicas-newRC.Spec.Replicas)))
|
||||||
newReplicasCount := newRC.Spec.Replicas + scaleUpCount
|
newReplicasCount := newRC.Spec.Replicas + scaleUpCount
|
||||||
_, err = d.scaleRCAndRecordEvent(newRC, newReplicasCount, deployment)
|
_, err = dc.scaleRCAndRecordEvent(newRC, newReplicasCount, deployment)
|
||||||
return true, err
|
return true, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) reconcileOldRCs(allRCs []*api.ReplicationController, oldRCs []*api.ReplicationController, newRC *api.ReplicationController, deployment extensions.Deployment) (bool, error) {
|
func (dc *DeploymentController) reconcileOldRCs(allRCs []*api.ReplicationController, oldRCs []*api.ReplicationController, newRC *api.ReplicationController, deployment extensions.Deployment) (bool, error) {
|
||||||
oldPodsCount := deploymentutil.GetReplicaCountForRCs(oldRCs)
|
oldPodsCount := deploymentutil.GetReplicaCountForRCs(oldRCs)
|
||||||
if oldPodsCount == 0 {
|
if oldPodsCount == 0 {
|
||||||
// Cant scale down further
|
// Cant scale down further
|
||||||
@ -209,7 +470,7 @@ func (d *DeploymentController) reconcileOldRCs(allRCs []*api.ReplicationControll
|
|||||||
minAvailable := deployment.Spec.Replicas - maxUnavailable
|
minAvailable := deployment.Spec.Replicas - maxUnavailable
|
||||||
minReadySeconds := deployment.Spec.Strategy.RollingUpdate.MinReadySeconds
|
minReadySeconds := deployment.Spec.Strategy.RollingUpdate.MinReadySeconds
|
||||||
// Find the number of ready pods.
|
// Find the number of ready pods.
|
||||||
readyPodCount, err := deploymentutil.GetAvailablePodsForRCs(d.client, allRCs, minReadySeconds)
|
readyPodCount, err := deploymentutil.GetAvailablePodsForRCs(dc.client, allRCs, minReadySeconds)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, fmt.Errorf("could not find available pods: %v", err)
|
return false, fmt.Errorf("could not find available pods: %v", err)
|
||||||
}
|
}
|
||||||
@ -231,7 +492,7 @@ func (d *DeploymentController) reconcileOldRCs(allRCs []*api.ReplicationControll
|
|||||||
// Scale down.
|
// Scale down.
|
||||||
scaleDownCount := int(math.Min(float64(targetRC.Spec.Replicas), float64(totalScaleDownCount)))
|
scaleDownCount := int(math.Min(float64(targetRC.Spec.Replicas), float64(totalScaleDownCount)))
|
||||||
newReplicasCount := targetRC.Spec.Replicas - scaleDownCount
|
newReplicasCount := targetRC.Spec.Replicas - scaleDownCount
|
||||||
_, err = d.scaleRCAndRecordEvent(targetRC, newReplicasCount, deployment)
|
_, err = dc.scaleRCAndRecordEvent(targetRC, newReplicasCount, deployment)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
@ -240,7 +501,7 @@ func (d *DeploymentController) reconcileOldRCs(allRCs []*api.ReplicationControll
|
|||||||
return true, err
|
return true, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) updateDeploymentStatus(allRCs []*api.ReplicationController, newRC *api.ReplicationController, deployment extensions.Deployment) error {
|
func (dc *DeploymentController) updateDeploymentStatus(allRCs []*api.ReplicationController, newRC *api.ReplicationController, deployment extensions.Deployment) error {
|
||||||
totalReplicas := deploymentutil.GetReplicaCountForRCs(allRCs)
|
totalReplicas := deploymentutil.GetReplicaCountForRCs(allRCs)
|
||||||
updatedReplicas := deploymentutil.GetReplicaCountForRCs([]*api.ReplicationController{newRC})
|
updatedReplicas := deploymentutil.GetReplicaCountForRCs([]*api.ReplicationController{newRC})
|
||||||
newDeployment := deployment
|
newDeployment := deployment
|
||||||
@ -249,29 +510,29 @@ func (d *DeploymentController) updateDeploymentStatus(allRCs []*api.ReplicationC
|
|||||||
Replicas: totalReplicas,
|
Replicas: totalReplicas,
|
||||||
UpdatedReplicas: updatedReplicas,
|
UpdatedReplicas: updatedReplicas,
|
||||||
}
|
}
|
||||||
_, err := d.client.Extensions().Deployments(deployment.ObjectMeta.Namespace).UpdateStatus(&newDeployment)
|
_, err := dc.client.Extensions().Deployments(api.NamespaceAll).UpdateStatus(&newDeployment)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) scaleRCAndRecordEvent(rc *api.ReplicationController, newScale int, deployment extensions.Deployment) (*api.ReplicationController, error) {
|
func (dc *DeploymentController) scaleRCAndRecordEvent(rc *api.ReplicationController, newScale int, deployment extensions.Deployment) (*api.ReplicationController, error) {
|
||||||
scalingOperation := "down"
|
scalingOperation := "down"
|
||||||
if rc.Spec.Replicas < newScale {
|
if rc.Spec.Replicas < newScale {
|
||||||
scalingOperation = "up"
|
scalingOperation = "up"
|
||||||
}
|
}
|
||||||
newRC, err := d.scaleRC(rc, newScale)
|
newRC, err := dc.scaleRC(rc, newScale)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
d.eventRecorder.Eventf(&deployment, api.EventTypeNormal, "ScalingRC", "Scaled %s rc %s to %d", scalingOperation, rc.Name, newScale)
|
d.eventRecorder.Eventf(&deployment, api.EventTypeNormal, "ScalingRC", "Scaled %s rc %s to %d", scalingOperation, rc.Name, newScale)
|
||||||
}
|
}
|
||||||
return newRC, err
|
return newRC, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) scaleRC(rc *api.ReplicationController, newScale int) (*api.ReplicationController, error) {
|
func (dc *DeploymentController) scaleRC(rc *api.ReplicationController, newScale int) (*api.ReplicationController, error) {
|
||||||
// TODO: Using client for now, update to use store when it is ready.
|
// TODO: Using client for now, update to use store when it is ready.
|
||||||
rc.Spec.Replicas = newScale
|
rc.Spec.Replicas = newScale
|
||||||
return d.client.ReplicationControllers(rc.ObjectMeta.Namespace).Update(rc)
|
return dc.client.ReplicationControllers(rc.ObjectMeta.Namespace).Update(rc)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DeploymentController) updateDeployment(deployment *extensions.Deployment) (*extensions.Deployment, error) {
|
func (dc *DeploymentController) updateDeployment(deployment *extensions.Deployment) (*extensions.Deployment, error) {
|
||||||
// TODO: Using client for now, update to use store when it is ready.
|
// TODO: Using client for now, update to use store when it is ready.
|
||||||
return d.client.Extensions().Deployments(deployment.ObjectMeta.Namespace).Update(deployment)
|
return dc.client.Extensions().Deployments(api.NamespaceAll).Update(deployment)
|
||||||
}
|
}
|
||||||
|
@ -21,10 +21,14 @@ import (
|
|||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"k8s.io/kubernetes/pkg/api"
|
"k8s.io/kubernetes/pkg/api"
|
||||||
|
"k8s.io/kubernetes/pkg/api/testapi"
|
||||||
|
"k8s.io/kubernetes/pkg/api/unversioned"
|
||||||
exp "k8s.io/kubernetes/pkg/apis/extensions"
|
exp "k8s.io/kubernetes/pkg/apis/extensions"
|
||||||
"k8s.io/kubernetes/pkg/client/record"
|
"k8s.io/kubernetes/pkg/client/record"
|
||||||
"k8s.io/kubernetes/pkg/client/unversioned/testclient"
|
"k8s.io/kubernetes/pkg/client/unversioned/testclient"
|
||||||
|
"k8s.io/kubernetes/pkg/controller"
|
||||||
"k8s.io/kubernetes/pkg/runtime"
|
"k8s.io/kubernetes/pkg/runtime"
|
||||||
|
"k8s.io/kubernetes/pkg/util"
|
||||||
"k8s.io/kubernetes/pkg/util/intstr"
|
"k8s.io/kubernetes/pkg/util/intstr"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -258,3 +262,159 @@ func deployment(name string, replicas int, maxSurge, maxUnavailable intstr.IntOr
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var alwaysReady = func() bool { return true }
|
||||||
|
|
||||||
|
func newDeployment(replicas int) *exp.Deployment {
|
||||||
|
d := exp.Deployment{
|
||||||
|
TypeMeta: unversioned.TypeMeta{APIVersion: testapi.Default.Version()},
|
||||||
|
ObjectMeta: api.ObjectMeta{
|
||||||
|
UID: util.NewUUID(),
|
||||||
|
Name: "foobar",
|
||||||
|
Namespace: api.NamespaceDefault,
|
||||||
|
ResourceVersion: "18",
|
||||||
|
},
|
||||||
|
Spec: exp.DeploymentSpec{
|
||||||
|
Strategy: exp.DeploymentStrategy{
|
||||||
|
Type: exp.RollingUpdateDeploymentStrategyType,
|
||||||
|
RollingUpdate: &exp.RollingUpdateDeployment{},
|
||||||
|
},
|
||||||
|
Replicas: replicas,
|
||||||
|
Selector: map[string]string{"foo": "bar"},
|
||||||
|
Template: api.PodTemplateSpec{
|
||||||
|
ObjectMeta: api.ObjectMeta{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"name": "foo",
|
||||||
|
"type": "production",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Spec: api.PodSpec{
|
||||||
|
Containers: []api.Container{
|
||||||
|
{
|
||||||
|
Image: "foo/bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
return &d
|
||||||
|
}
|
||||||
|
|
||||||
|
func getKey(d *exp.Deployment, t *testing.T) string {
|
||||||
|
if key, err := controller.KeyFunc(d); err != nil {
|
||||||
|
t.Errorf("Unexpected error getting key for deployment %v: %v", d.Name, err)
|
||||||
|
return ""
|
||||||
|
} else {
|
||||||
|
return key
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func newReplicationController(d *exp.Deployment, name string, replicas int) *api.ReplicationController {
|
||||||
|
return &api.ReplicationController{
|
||||||
|
ObjectMeta: api.ObjectMeta{
|
||||||
|
Name: name,
|
||||||
|
Namespace: api.NamespaceDefault,
|
||||||
|
},
|
||||||
|
Spec: api.ReplicationControllerSpec{
|
||||||
|
Replicas: 0,
|
||||||
|
Template: &d.Spec.Template,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
type fixture struct {
|
||||||
|
t *testing.T
|
||||||
|
|
||||||
|
client *testclient.Fake
|
||||||
|
|
||||||
|
// Objects to put in the store.
|
||||||
|
dStore []*exp.Deployment
|
||||||
|
rcStore []*api.ReplicationController
|
||||||
|
podStore []*api.Pod
|
||||||
|
|
||||||
|
// Actions expected to happen on the client. Objects from here are also
|
||||||
|
// preloaded into NewSimpleFake.
|
||||||
|
actions []testclient.Action
|
||||||
|
objects *api.List
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *fixture) expectUpdateDeploymentAction(d *exp.Deployment) {
|
||||||
|
f.actions = append(f.actions, testclient.NewUpdateAction("deployments", d.Namespace, d))
|
||||||
|
f.objects.Items = append(f.objects.Items, d)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *fixture) expectCreateRCAction(rc *api.ReplicationController) {
|
||||||
|
f.actions = append(f.actions, testclient.NewCreateAction("replicationcontrollers", rc.Namespace, rc))
|
||||||
|
f.objects.Items = append(f.objects.Items, rc)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *fixture) expectUpdateRCAction(rc *api.ReplicationController) {
|
||||||
|
f.actions = append(f.actions, testclient.NewUpdateAction("replicationcontrollers", rc.Namespace, rc))
|
||||||
|
f.objects.Items = append(f.objects.Items, rc)
|
||||||
|
}
|
||||||
|
|
||||||
|
func newFixture(t *testing.T) *fixture {
|
||||||
|
f := &fixture{}
|
||||||
|
f.t = t
|
||||||
|
f.objects = &api.List{}
|
||||||
|
return f
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *fixture) run(deploymentName string) {
|
||||||
|
f.client = testclient.NewSimpleFake(f.objects)
|
||||||
|
c := NewDeploymentController(f.client)
|
||||||
|
c.rcStoreSynced = alwaysReady
|
||||||
|
c.podStoreSynced = alwaysReady
|
||||||
|
for _, d := range f.dStore {
|
||||||
|
c.dStore.Store.Add(d)
|
||||||
|
}
|
||||||
|
for _, rc := range f.rcStore {
|
||||||
|
c.rcStore.Store.Add(rc)
|
||||||
|
}
|
||||||
|
for _, pod := range f.podStore {
|
||||||
|
c.podStore.Store.Add(pod)
|
||||||
|
}
|
||||||
|
|
||||||
|
err := c.syncDeployment(deploymentName)
|
||||||
|
if err != nil {
|
||||||
|
f.t.Errorf("error syncing deployment: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
actions := f.client.Actions()
|
||||||
|
for i, action := range actions {
|
||||||
|
if len(f.actions) < i+1 {
|
||||||
|
f.t.Errorf("%d unexpected actions: %+v", len(actions)-len(f.actions), actions[i:])
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
expectedAction := f.actions[i]
|
||||||
|
if !expectedAction.Matches(action.GetVerb(), action.GetResource()) {
|
||||||
|
f.t.Errorf("Expected\n\t%#v\ngot\n\t%#v", expectedAction, action)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(f.actions) > len(actions) {
|
||||||
|
f.t.Errorf("%d additional expected actions:%+v", len(f.actions)-len(actions), f.actions[len(actions):])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSyncDeploymentCreatesRC(t *testing.T) {
|
||||||
|
f := newFixture(t)
|
||||||
|
|
||||||
|
d := newDeployment(1)
|
||||||
|
f.dStore = append(f.dStore, d)
|
||||||
|
|
||||||
|
// expect that one rc with zero replicas is created
|
||||||
|
// then is updated to 1 replica
|
||||||
|
rc := newReplicationController(d, "deploymentrc-4186632231", 0)
|
||||||
|
updatedRC := newReplicationController(d, "deploymentrc-4186632231", 1)
|
||||||
|
|
||||||
|
f.expectCreateRCAction(rc)
|
||||||
|
f.expectUpdateRCAction(updatedRC)
|
||||||
|
f.expectUpdateDeploymentAction(d)
|
||||||
|
|
||||||
|
f.run(getKey(d, t))
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user