129 lines
3.9 KiB
Go
129 lines
3.9 KiB
Go
/*
|
|
Copyright 2022 The Kruise Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package deployment
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
apps "k8s.io/api/apps/v1"
|
|
corev1 "k8s.io/api/core/v1"
|
|
v1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
|
|
"github.com/openkruise/rollouts/api/v1beta1"
|
|
batchcontext "github.com/openkruise/rollouts/pkg/controller/batchrelease/context"
|
|
"github.com/openkruise/rollouts/pkg/controller/batchrelease/control"
|
|
"github.com/openkruise/rollouts/pkg/controller/batchrelease/control/canarystyle"
|
|
"github.com/openkruise/rollouts/pkg/util"
|
|
utilclient "github.com/openkruise/rollouts/pkg/util/client"
|
|
)
|
|
|
|
type realController struct {
|
|
realStableController
|
|
realCanaryController
|
|
}
|
|
|
|
func NewController(cli client.Client, key types.NamespacedName) canarystyle.Interface {
|
|
return &realController{
|
|
realStableController: newStable(cli, key),
|
|
realCanaryController: newCanary(cli, key),
|
|
}
|
|
}
|
|
|
|
func (rc *realController) BuildStableController() (canarystyle.StableInterface, error) {
|
|
if rc.stableObject != nil {
|
|
return rc, nil
|
|
}
|
|
|
|
object := &apps.Deployment{}
|
|
err := rc.stableClient.Get(context.TODO(), rc.stableKey, object)
|
|
if err != nil {
|
|
return rc, err
|
|
}
|
|
rc.stableObject = object
|
|
rc.stableInfo = util.ParseWorkload(object)
|
|
return rc, nil
|
|
}
|
|
|
|
func (rc *realController) BuildCanaryController(release *v1beta1.BatchRelease) (canarystyle.CanaryInterface, error) {
|
|
if rc.canaryObject != nil {
|
|
return rc, nil
|
|
}
|
|
|
|
ds, err := rc.listDeployment(release, client.InNamespace(rc.stableKey.Namespace), utilclient.DisableDeepCopy)
|
|
if err != nil {
|
|
return rc, err
|
|
}
|
|
|
|
template, err := rc.getLatestTemplate()
|
|
if client.IgnoreNotFound(err) != nil {
|
|
return rc, err
|
|
}
|
|
rc.canaryObject = filterCanaryDeployment(release, util.FilterActiveDeployment(ds), template)
|
|
if rc.canaryObject == nil {
|
|
return rc, control.GenerateNotFoundError(fmt.Sprintf("%v-canary", rc.stableKey), "Deployment")
|
|
}
|
|
|
|
rc.canaryInfo = util.ParseWorkload(rc.canaryObject)
|
|
return rc, nil
|
|
}
|
|
|
|
func (rc *realController) CalculateBatchContext(release *v1beta1.BatchRelease) (*batchcontext.BatchContext, error) {
|
|
rolloutID := release.Spec.ReleasePlan.RolloutID
|
|
if rolloutID != "" {
|
|
// if rollout-id is set, the pod will be patched batch label,
|
|
// so we have to list pod here.
|
|
if _, err := rc.ListOwnedPods(); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
replicas := *rc.stableObject.Spec.Replicas
|
|
currentBatch := release.Status.CanaryStatus.CurrentBatch
|
|
desiredUpdate := int32(control.CalculateBatchReplicas(release, int(replicas), int(currentBatch)))
|
|
|
|
return &batchcontext.BatchContext{
|
|
Pods: rc.canaryPods,
|
|
RolloutID: rolloutID,
|
|
Replicas: replicas,
|
|
UpdateRevision: release.Status.UpdateRevision,
|
|
CurrentBatch: currentBatch,
|
|
DesiredUpdatedReplicas: desiredUpdate,
|
|
FailureThreshold: release.Spec.ReleasePlan.FailureThreshold,
|
|
UpdatedReplicas: rc.canaryObject.Status.Replicas,
|
|
UpdatedReadyReplicas: rc.canaryObject.Status.AvailableReplicas,
|
|
}, nil
|
|
}
|
|
|
|
func (rc *realController) getLatestTemplate() (*v1.PodTemplateSpec, error) {
|
|
_, err := rc.BuildStableController()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &rc.stableObject.Spec.Template, nil
|
|
}
|
|
|
|
func (rc *realController) ListOwnedPods() ([]*corev1.Pod, error) {
|
|
if rc.canaryPods != nil {
|
|
return rc.canaryPods, nil
|
|
}
|
|
var err error
|
|
rc.canaryPods, err = util.ListOwnedPods(rc.canaryClient, rc.canaryObject)
|
|
return rc.canaryPods, err
|
|
}
|