forked from jenkins-x/jx
-
Notifications
You must be signed in to change notification settings - Fork 0
/
step_link_services.go
117 lines (105 loc) · 3.85 KB
/
step_link_services.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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
package cmd
import (
"github.com/jenkins-x/jx/pkg/jx/cmd/templates"
"github.com/jenkins-x/jx/pkg/util"
"github.com/spf13/cobra"
"io"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)
const (
fromNamespace = "from-namespace"
toNamespace = "to-namespace"
includes = "includes"
excludes = "excludes"
)
var (
StepLinkServicesLong = templates.LongDesc(`
This pipeline step helps to link microservices from different namespaces like staging/production onto a preview environment
`)
StepLinkServicesExample = templates.Examples(`
#Link services from jx-staging namespace to the current namespace
jx step link services --from-namespace jx-staging
#Link services from jx-staging namespace to the jx-prod namespace
jx step link services --from-namespace jx-staging --to-namespace jx-prod
#Link services from jx-staging namespace to the jx-prod namespace including all but the ones starting with the characters 'cheese'
jx step link services --from-namespace jx-staging --to-namespace jx-prod --includes * --excludes cheese*
`)
)
// StepLinkServicesOptions contains the command line flags
type StepLinkServicesOptions struct {
StepOptions
FromNamespace string
ToNamespace string
Includes []string
Excludes []string
}
// NewCmdStepLinkServices Creates a new Command object
func NewCmdStepLinkServices(f Factory, out io.Writer, errOut io.Writer) *cobra.Command {
options := &StepLinkServicesOptions{
StepOptions: StepOptions{
CommonOptions: CommonOptions{
Factory: f,
Out: out,
Err: errOut,
},
},
}
cmd := &cobra.Command{
Use: "service linking",
Short: "achieve service linking in preview environments",
Long: StepLinkServicesLong,
Example: StepLinkServicesExample,
Run: func(cmd *cobra.Command, args []string) {
options.Cmd = cmd
options.Args = args
err := options.Run()
CheckErr(err)
},
}
options.addCommonFlags(cmd)
cmd.Flags().StringVarP(&options.FromNamespace, fromNamespace, "f", "", "The source namespace from which the linking would happen")
cmd.Flags().StringVarP(&options.ToNamespace, toNamespace, "t", "", "The destination namespace to which the linking would happen")
cmd.Flags().StringArrayVarP(&options.Includes, includes, "i", []string{}, "What services from source namespace to include in the linking process")
cmd.Flags().StringArrayVarP(&options.Excludes, excludes, "e", []string{}, "What services from the source namespace to exclude from the linking process")
return cmd
}
// Run implements this command
func (o *StepLinkServicesOptions) Run() error {
if o.FromNamespace == "" {
return util.MissingOption(fromNamespace)
}
kubeClient, currentNs, err := o.KubeClient()
if err != nil {
return err
}
targetNamespace := o.ToNamespace
if targetNamespace == "" {
//to-namespace wasn't supplied, let's assume it is current namespace
targetNamespace = currentNs
}
if targetNamespace == "" {
//We don't want to continue if we still can't derive to-namespace
return util.MissingOption(toNamespace)
} else {
serviceList, err := kubeClient.CoreV1().Services(o.FromNamespace).List(metav1.ListOptions{})
if err != nil {
return err
} else {
for _, service := range serviceList.Items {
if util.StringMatchesAny(service.Name, o.Includes, o.Excludes) {
lookedUpServiceFromTargetNamespace, _ := kubeClient.CoreV1().Services(targetNamespace).Get(service.GetName(), metav1.GetOptions{})
//Change the namespace in the service to target namespace
targetService := service
targetService.Namespace = targetNamespace
// We would create a new service if it doesn't already exist OR update if it already exists
if lookedUpServiceFromTargetNamespace != nil {
kubeClient.CoreV1().Services(targetNamespace).Update(&targetService)
} else {
kubeClient.CoreV1().Services(targetNamespace).Create(&targetService)
}
}
}
}
}
return nil
}