| /* |
| Copyright 2018 The Kubernetes 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 polymorphichelpers |
| |
| import ( |
| "fmt" |
| "strconv" |
| |
| appsv1 "k8s.io/api/apps/v1" |
| appsv1beta1 "k8s.io/api/apps/v1beta1" |
| appsv1beta2 "k8s.io/api/apps/v1beta2" |
| corev1 "k8s.io/api/core/v1" |
| extensionsv1beta1 "k8s.io/api/extensions/v1beta1" |
| "k8s.io/apimachinery/pkg/runtime" |
| ) |
| |
| func protocolsForObject(object runtime.Object) (map[string]string, error) { |
| // TODO: replace with a swagger schema based approach (identify pod selector via schema introspection) |
| switch t := object.(type) { |
| case *corev1.ReplicationController: |
| return getProtocols(t.Spec.Template.Spec), nil |
| |
| case *corev1.Pod: |
| return getProtocols(t.Spec), nil |
| |
| case *corev1.Service: |
| return getServiceProtocols(t.Spec), nil |
| |
| case *extensionsv1beta1.Deployment: |
| return getProtocols(t.Spec.Template.Spec), nil |
| case *appsv1.Deployment: |
| return getProtocols(t.Spec.Template.Spec), nil |
| case *appsv1beta2.Deployment: |
| return getProtocols(t.Spec.Template.Spec), nil |
| case *appsv1beta1.Deployment: |
| return getProtocols(t.Spec.Template.Spec), nil |
| |
| case *extensionsv1beta1.ReplicaSet: |
| return getProtocols(t.Spec.Template.Spec), nil |
| case *appsv1.ReplicaSet: |
| return getProtocols(t.Spec.Template.Spec), nil |
| case *appsv1beta2.ReplicaSet: |
| return getProtocols(t.Spec.Template.Spec), nil |
| |
| default: |
| return nil, fmt.Errorf("cannot extract protocols from %T", object) |
| } |
| } |
| |
| func getProtocols(spec corev1.PodSpec) map[string]string { |
| result := make(map[string]string) |
| for _, container := range spec.Containers { |
| for _, port := range container.Ports { |
| // Empty protocol must be defaulted (TCP) |
| if len(port.Protocol) == 0 { |
| port.Protocol = corev1.ProtocolTCP |
| } |
| result[strconv.Itoa(int(port.ContainerPort))] = string(port.Protocol) |
| } |
| } |
| return result |
| } |
| |
| // Extracts the protocols exposed by a service from the given service spec. |
| func getServiceProtocols(spec corev1.ServiceSpec) map[string]string { |
| result := make(map[string]string) |
| for _, servicePort := range spec.Ports { |
| // Empty protocol must be defaulted (TCP) |
| if len(servicePort.Protocol) == 0 { |
| servicePort.Protocol = corev1.ProtocolTCP |
| } |
| result[strconv.Itoa(int(servicePort.Port))] = string(servicePort.Protocol) |
| } |
| return result |
| } |