/
rabbitmq_app.go
129 lines (100 loc) · 3.77 KB
/
rabbitmq_app.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
118
119
120
121
122
123
124
125
126
127
128
129
// Copyright (c) arkade author(s) 2022. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
package apps
import (
"fmt"
"github.com/alexellis/arkade/pkg/config"
"github.com/alexellis/arkade/pkg"
"github.com/alexellis/arkade/pkg/apps"
"github.com/alexellis/arkade/pkg/types"
"github.com/spf13/cobra"
)
func MakeInstallRabbitmq() *cobra.Command {
var rabbitmq = &cobra.Command{
Use: "rabbitmq",
Short: "Install rabbitmq",
Long: "Install rabbitmq",
Example: "arkade install rabbitmq",
SilenceUsage: true,
}
rabbitmq.Flags().StringP("namespace", "n", "rabbitmq", "The namespace to install rabbitmq")
rabbitmq.Flags().Bool("update-repo", true, "Update the helm repo")
rabbitmq.Flags().Bool("persistence", false, "Make rabbitmq persistent")
rabbitmq.Flags().StringArray("set", []string{},
"Use custom flags or override existing flags \n(example --set persistence.enabled=true)")
rabbitmq.PreRunE = func(command *cobra.Command, args []string) error {
_, err := command.Flags().GetString("kubeconfig")
if err != nil {
return fmt.Errorf("error with --kubeconfig usage: %s", err)
}
_, err = command.Flags().GetBool("wait")
if err != nil {
return fmt.Errorf("error with --wait usage: %s", err)
}
_, err = command.Flags().GetBool("persistence")
if err != nil {
return fmt.Errorf("error with --persistence usage: %s", err)
}
_, err = command.Flags().GetString("namespace")
if err != nil {
return fmt.Errorf("error with --namespace usage: %s", err)
}
_, err = command.Flags().GetBool("update-repo")
if err != nil {
return fmt.Errorf("error with --update-repo usage: %s", err)
}
_, err = command.Flags().GetStringArray("set")
if err != nil {
return fmt.Errorf("error with --set usage: %s", err)
}
return nil
}
rabbitmq.RunE = func(command *cobra.Command, args []string) error {
// Get all flags
kubeConfigPath, _ := command.Flags().GetString("kubeconfig")
wait, _ := command.Flags().GetBool("wait")
persistence, _ := command.Flags().GetBool("persistence")
namespace, _ := command.Flags().GetString("namespace")
updateRepo, _ := command.Flags().GetBool("update-repo")
customFlags, _ := command.Flags().GetStringArray("set")
overrides := map[string]string{}
if persistence {
overrides["persistence.enabled"] = "true"
overrides["persistence.size"] = "2Gi"
}
// set custom flags
if err := config.MergeFlags(overrides, customFlags); err != nil {
return err
}
rabbitmqAppOptions := types.DefaultInstallOptions().
WithNamespace(namespace).
WithHelmRepo("bitnami/rabbitmq").
WithHelmURL("https://charts.bitnami.com/bitnami").
WithOverrides(overrides).
WithHelmUpdateRepo(updateRepo).
WithKubeconfigPath(kubeConfigPath).
WithWait(wait)
_, err := apps.MakeInstallChart(rabbitmqAppOptions)
if err != nil {
return err
}
fmt.Println(rabbitmqInstallMsg)
return nil
}
return rabbitmq
}
const RabbitmqInfoMsg = `
# By default the "user" username is used
# To get your password run:
export RABBITMQ_PASSWORD=$(kubectl get secret --namespace rabbitmq rabbitmq -o jsonpath="{.data.rabbitmq-password}" | base64 -d)
# To Access the RabbitMQ Management interface:
kubectl port-forward --namespace rabbitmq svc/rabbitmq 15672:15672
# To Access the RabbitMQ AMQP port:
kubectl port-forward --namespace rabbitmq svc/rabbitmq 5672:5672
# Enable persistence:
arkade install rabbitmq --persistence
`
var rabbitmqInstallMsg = `=======================================================================
= rabbitmq has been installed =
=======================================================================` +
"\n\n" + RabbitmqInfoMsg + "\n\n" + pkg.SupportMessageShort