forked from CrunchyData/postgres-operator
-
Notifications
You must be signed in to change notification settings - Fork 0
/
reload.go
99 lines (81 loc) · 2.63 KB
/
reload.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
// Package cmd provides the command line functions of the crunchy CLI
package cmd
/*
Copyright 2018 - 2021 Crunchy Data Solutions, Inc.
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.
*/
import (
"fmt"
"os"
"github.com/crunchydata/postgres-operator/pgo/api"
"github.com/crunchydata/postgres-operator/pgo/util"
msgs "github.com/crunchydata/postgres-operator/pkg/apiservermsgs"
log "github.com/sirupsen/logrus"
"github.com/spf13/cobra"
)
//unused but coming soon to a theatre near you
var ConfigMapName string
var reloadCmd = &cobra.Command{
Use: "reload",
Short: "Perform a cluster reload",
Long: `RELOAD performs a PostgreSQL reload on a cluster or set of clusters. For example:
pgo reload mycluster`,
Run: func(cmd *cobra.Command, args []string) {
if Namespace == "" {
Namespace = PGONamespace
}
log.Debug("reload called")
if len(args) == 0 && Selector == "" {
fmt.Println(`Error: You must specify the cluster to reload or specify a selector flag.`)
} else {
if util.AskForConfirmation(NoPrompt, "") {
reload(args, Namespace)
} else {
fmt.Println("Aborting...")
}
}
},
}
func init() {
RootCmd.AddCommand(reloadCmd)
reloadCmd.Flags().StringVarP(&Selector, "selector", "s", "", "The selector to use for cluster filtering.")
reloadCmd.Flags().BoolVar(&NoPrompt, "no-prompt", false, "No command line confirmation.")
}
// reload ....
func reload(args []string, ns string) {
log.Debugf("reload called %v", args)
request := new(msgs.ReloadRequest)
request.Args = args
request.Selector = Selector
request.Namespace = ns
response, err := api.Reload(httpclient, &SessionCredentials, request)
if err != nil {
fmt.Println("Error: " + err.Error())
os.Exit(2)
}
if response.Status.Code == msgs.Ok {
for k := range response.Results {
fmt.Println(response.Results[k])
}
} else {
// print the error message as well as the results, since the reload might have succeeded
// for certain clusters specified, but not for others
fmt.Println(response.Status.Msg)
for k := range response.Results {
fmt.Println(response.Results[k])
}
os.Exit(2)
}
if len(response.Results) == 0 {
fmt.Println("No clusters found.")
return
}
}