/
portsforobject.go
33 lines (27 loc) · 789 Bytes
/
portsforobject.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
package originpolymorphichelpers
import (
"strconv"
corev1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/kubernetes/pkg/kubectl/polymorphichelpers"
appsv1 "github.com/openshift/api/apps/v1"
)
func NewPortsForObjectFn(delegate polymorphichelpers.PortsForObjectFunc) polymorphichelpers.PortsForObjectFunc {
return func(object runtime.Object) ([]string, error) {
switch t := object.(type) {
case *appsv1.DeploymentConfig:
return getPorts(t.Spec.Template.Spec), nil
default:
return delegate(object)
}
}
}
func getPorts(spec corev1.PodSpec) []string {
result := []string{}
for _, container := range spec.Containers {
for _, port := range container.Ports {
result = append(result, strconv.Itoa(int(port.ContainerPort)))
}
}
return result
}