Fix the nil pointer dereference when counting RS replicas

pull/6/head
Janet Kuo 2016-02-24 16:09:20 -08:00
parent c031697848
commit 069e77d504
1 changed files with 19 additions and 13 deletions

View File

@ -325,8 +325,10 @@ func SetFromReplicaSetTemplate(deployment *extensions.Deployment, template api.P
func GetReplicaCountForReplicaSets(replicaSets []*extensions.ReplicaSet) int { func GetReplicaCountForReplicaSets(replicaSets []*extensions.ReplicaSet) int {
totalReplicaCount := 0 totalReplicaCount := 0
for _, rs := range replicaSets { for _, rs := range replicaSets {
if rs != nil {
totalReplicaCount += rs.Spec.Replicas totalReplicaCount += rs.Spec.Replicas
} }
}
return totalReplicaCount return totalReplicaCount
} }
@ -334,8 +336,10 @@ func GetReplicaCountForReplicaSets(replicaSets []*extensions.ReplicaSet) int {
func GetActualReplicaCountForReplicaSets(replicaSets []*extensions.ReplicaSet) int { func GetActualReplicaCountForReplicaSets(replicaSets []*extensions.ReplicaSet) int {
totalReplicaCount := 0 totalReplicaCount := 0
for _, rs := range replicaSets { for _, rs := range replicaSets {
if rs != nil {
totalReplicaCount += rs.Status.Replicas totalReplicaCount += rs.Status.Replicas
} }
}
return totalReplicaCount return totalReplicaCount
} }
@ -379,6 +383,7 @@ func IsPodAvailable(pod *api.Pod, minReadySeconds int) bool {
func GetPodsForReplicaSets(c clientset.Interface, replicaSets []*extensions.ReplicaSet) ([]api.Pod, error) { func GetPodsForReplicaSets(c clientset.Interface, replicaSets []*extensions.ReplicaSet) ([]api.Pod, error) {
allPods := map[string]api.Pod{} allPods := map[string]api.Pod{}
for _, rs := range replicaSets { for _, rs := range replicaSets {
if rs != nil {
selector, err := unversioned.LabelSelectorAsSelector(rs.Spec.Selector) selector, err := unversioned.LabelSelectorAsSelector(rs.Spec.Selector)
if err != nil { if err != nil {
return nil, fmt.Errorf("invalid label selector: %v", err) return nil, fmt.Errorf("invalid label selector: %v", err)
@ -392,6 +397,7 @@ func GetPodsForReplicaSets(c clientset.Interface, replicaSets []*extensions.Repl
allPods[pod.Name] = pod allPods[pod.Name] = pod
} }
} }
}
requiredPods := []api.Pod{} requiredPods := []api.Pod{}
for _, pod := range allPods { for _, pod := range allPods {
requiredPods = append(requiredPods, pod) requiredPods = append(requiredPods, pod)