-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
7 changed files
with
176 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,133 @@ | ||
| package status | ||
|
|
||
| import ( | ||
| "errors" | ||
| "fmt" | ||
| "os" | ||
| "strconv" | ||
| "strings" | ||
| "time" | ||
|
|
||
| "github.com/clevyr/kubedb/internal/config" | ||
| "github.com/clevyr/kubedb/internal/config/flags" | ||
| "github.com/clevyr/kubedb/internal/consts" | ||
| "github.com/clevyr/kubedb/internal/database" | ||
| "github.com/clevyr/kubedb/internal/kubernetes" | ||
| "github.com/clevyr/kubedb/internal/util" | ||
| "github.com/fatih/color" | ||
| log "github.com/sirupsen/logrus" | ||
| "github.com/spf13/cobra" | ||
| "github.com/spf13/viper" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| ) | ||
|
|
||
| func New() *cobra.Command { | ||
| cmd := &cobra.Command{ | ||
| Use: "status", | ||
| Short: "View connection status", | ||
| GroupID: "ro", | ||
| PreRunE: preRun, | ||
| RunE: run, | ||
| } | ||
| flags.JobPodLabels(cmd) | ||
| flags.Port(cmd) | ||
| flags.Database(cmd) | ||
| flags.Username(cmd) | ||
| flags.Password(cmd) | ||
| return cmd | ||
| } | ||
|
|
||
| var conf config.Global | ||
|
|
||
| func preRun(cmd *cobra.Command, args []string) error { | ||
| cmd.SilenceUsage = true | ||
| log.SetLevel(log.WarnLevel) | ||
| flags.BindJobPodLabels(cmd) | ||
| return nil | ||
| } | ||
|
|
||
| func run(cmd *cobra.Command, args []string) error { | ||
| prefixOk := color.GreenString(" ✓") | ||
| prefixErr := color.RedString(" ✗") | ||
| bold := color.New(color.Bold) | ||
|
|
||
| viper.Set(consts.NamespaceFilterKey, false) | ||
| defaultSetupErr := util.DefaultSetup(cmd, &conf, util.SetupOptions{Name: "status"}) | ||
|
|
||
| fmt.Println(bold.Sprintf("Cluster Info")) | ||
| if conf.Client.ClientSet != nil { | ||
| fmt.Println( | ||
| prefixOk, | ||
| "Using cluster", | ||
| bold.Sprintf(conf.Context), | ||
| "at", | ||
| bold.Sprintf(conf.Client.ClientConfig.Host), | ||
| ) | ||
| } else { | ||
| fmt.Println(prefixErr, "Failed to load cluster config:", defaultSetupErr.Error()) | ||
| os.Exit(1) | ||
| } | ||
|
|
||
| if serverVersion, err := conf.Client.Discovery.ServerVersion(); err == nil { | ||
| fmt.Println(prefixOk, "Cluster version is", bold.Sprintf(serverVersion.String())) | ||
| } else { | ||
| fmt.Println(prefixErr, "Cluster version check failed:", err.Error()) | ||
| } | ||
|
|
||
| if namespaces, err := conf.Client.Namespaces().List(cmd.Context(), metav1.ListOptions{}); err == nil { | ||
| fmt.Println(prefixOk, "Cluster has", bold.Sprintf(strconv.Itoa(len(namespaces.Items))), "namespaces") | ||
| } else { | ||
| fmt.Println(prefixErr, "Failed to list namespaces:", err.Error()) | ||
| } | ||
|
|
||
| fmt.Println(prefixOk, "Using namespace", bold.Sprintf(conf.Client.Namespace)) | ||
|
|
||
| fmt.Println(bold.Sprintf("Database Info")) | ||
| if defaultSetupErr == nil { | ||
| fmt.Println( | ||
| prefixOk, | ||
| "Found", | ||
| bold.Sprintf(conf.Dialect.Name()), | ||
| "database", | ||
| bold.Sprintf(conf.DbPod.ObjectMeta.Name), | ||
| ) | ||
| } else { | ||
| if errors.Is(defaultSetupErr, database.ErrDatabaseNotFound) { | ||
| fmt.Println(prefixErr, "Could not detect a database") | ||
| } else { | ||
| fmt.Println(prefixErr, "Failed to search for database:", defaultSetupErr.Error()) | ||
| } | ||
| os.Exit(1) | ||
| } | ||
|
|
||
| if err := util.CreateJob(cmd, &conf, util.SetupOptions{Name: "status"}); err == nil { | ||
| fmt.Println(prefixOk, "Jobs can be created") | ||
| } else { | ||
| fmt.Println(prefixErr, "Job creation failed:", err.Error()) | ||
| os.Exit(1) | ||
| } | ||
| defer util.Teardown(cmd, &conf) | ||
|
|
||
| var buf strings.Builder | ||
| listTablesCmd := conf.Dialect.ExecCommand(config.Exec{ | ||
| Global: conf, | ||
| DisableHeaders: true, | ||
| Command: conf.Dialect.ListTablesQuery(), | ||
| }) | ||
| execOpts := kubernetes.ExecOptions{ | ||
| Pod: conf.JobPod, | ||
| Cmd: listTablesCmd.String(), | ||
| Stdout: &buf, | ||
| Stderr: os.Stderr, | ||
| PingPeriod: 5 * time.Second, | ||
| } | ||
| if err := conf.Client.Exec(cmd.Context(), execOpts); err == nil { | ||
| names := strings.Split(buf.String(), "\n") | ||
| fmt.Println(prefixOk, "Database has", bold.Sprintf(strconv.Itoa(len(names))), "tables") | ||
| } else { | ||
| fmt.Println(prefixErr, "Failed to connect to database", err.Error()) | ||
| os.Exit(1) | ||
| } | ||
|
|
||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,35 @@ | ||
| ## kubedb status | ||
|
|
||
| View connection status | ||
|
|
||
| ``` | ||
| kubedb status [flags] | ||
| ``` | ||
|
|
||
| ### Options | ||
|
|
||
| ``` | ||
| -d, --dbname string Database name to use (default discovered) | ||
| -h, --help help for status | ||
| --job-pod-labels stringToString Pod labels to add to the job (default []) | ||
| -p, --password string Database password (default discovered) | ||
| --port uint16 Database port (default discovered) | ||
| -U, --username string Database username (default discovered) | ||
| ``` | ||
|
|
||
| ### Options inherited from parent commands | ||
|
|
||
| ``` | ||
| --context string Kubernetes context name | ||
| --dialect string Database dialect. One of (postgres|mariadb|mongodb) (default discovered) | ||
| --kubeconfig string Paths to the kubeconfig file (default "$HOME/.kube/config") | ||
| --log-format string Log formatter. One of (text|json) (default "text") | ||
| --log-level string Log level. One of (trace|debug|info|warning|error|fatal|panic) (default "info") | ||
| -n, --namespace string Kubernetes namespace | ||
| --pod string Perform detection from a pod instead of searching the namespace | ||
| ``` | ||
|
|
||
| ### SEE ALSO | ||
|
|
||
| * [kubedb](kubedb.md) - Painlessly work with databases in Kubernetes. | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters