Skip to content

Commit

Permalink
PT-2236 - pt-secure-collect, pt-pg-summary do not follow PT standard …
Browse files Browse the repository at this point in the history
…for option --version

Renamed const TOOLNAME to toolname to follow Go coding standards
  • Loading branch information
svetasmirnova committed Jul 28, 2023
1 parent 62342a2 commit 854bd8e
Show file tree
Hide file tree
Showing 11 changed files with 35 additions and 35 deletions.
4 changes: 2 additions & 2 deletions src/go/pt-k8s-debug-collector/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ import (
)

const (
TOOLNAME = "pt-k8s-debug-collector"
toolname = "pt-k8s-debug-collector"
)

// We do not set anything here, these variables are defined by the Makefile
Expand Down Expand Up @@ -38,7 +38,7 @@ func main() {
flag.Parse()

if version {
fmt.Println(TOOLNAME)
fmt.Println(toolname)
fmt.Printf("Version %s\n", Version)
fmt.Printf("Build: %s using %s\n", Build, GoVersion)
fmt.Printf("Commit: %s\n", Commit)
Expand Down
8 changes: 4 additions & 4 deletions src/go/pt-k8s-debug-collector/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -164,14 +164,14 @@ func TestResourceOption(t *testing.T) {
Option --version
*/
func TestVersionOption(t *testing.T) {
out, err := exec.Command("../../../bin/"+TOOLNAME, "--version").Output()
out, err := exec.Command("../../../bin/"+toolname, "--version").Output()
if err != nil {
t.Errorf("error executing %s --version: %s", TOOLNAME, err.Error())
t.Errorf("error executing %s --version: %s", toolname, err.Error())
}
// We are using MustCompile here, because hard-coded RE should not fail
re := regexp.MustCompile(TOOLNAME + `\n.*Version v?\d+\.\d+\.\d+\n`)
re := regexp.MustCompile(toolname + `\n.*Version v?\d+\.\d+\.\d+\n`)
if !re.Match(out) {
t.Errorf("%s --version returns wrong result:\n%s", TOOLNAME, out)
t.Errorf("%s --version returns wrong result:\n%s", toolname, out)
}
}

Expand Down
4 changes: 2 additions & 2 deletions src/go/pt-mongodb-index-check/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ type response struct {
}

const (
TOOLNAME = "pt-mongodb-index-check"
toolname = "pt-mongodb-index-check"
)

// We do not set anything here, these variables are defined by the Makefile
Expand All @@ -58,7 +58,7 @@ func main() {
kongctx := kong.Parse(&args, kong.UsageOnError())

if kongctx.Command() == "version" {
fmt.Println(TOOLNAME)
fmt.Println(toolname)
fmt.Printf("Version %s\n", Version)
fmt.Printf("Build: %s using %s\n", Build, GoVersion)
fmt.Printf("Commit: %s\n", Commit)
Expand Down
10 changes: 5 additions & 5 deletions src/go/pt-mongodb-query-digest/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ import (
)

const (
TOOLNAME = "pt-mongodb-query-digest"
toolname = "pt-mongodb-query-digest"

DEFAULT_AUTHDB = "admin"
DEFAULT_HOST = "localhost:27017"
Expand Down Expand Up @@ -90,16 +90,16 @@ func main() {
log.SetLevel(logLevel)

if opts.Version {
fmt.Println(TOOLNAME)
fmt.Println(toolname)
fmt.Printf("Version %s\n", Version)
fmt.Printf("Build: %s using %s\n", Build, GoVersion)
fmt.Printf("Commit: %s\n", Commit)
return
}

conf := config.DefaultConfig(TOOLNAME)
conf := config.DefaultConfig(toolname)
if !conf.GetBool("no-version-check") && !opts.NoVersionCheck {
advice, err := versioncheck.CheckUpdates(TOOLNAME, Version)
advice, err := versioncheck.CheckUpdates(toolname, Version)
if err != nil {
log.Infof("cannot check version updates: %s", err.Error())
} else if advice != "" {
Expand Down Expand Up @@ -361,7 +361,7 @@ func getClientOptions(opts *cliOptions) (*options.ClientOptions, error) {

func getHeaders(opts *cliOptions) []string {
h := []string{
fmt.Sprintf("%s - %s\n", TOOLNAME, time.Now().Format(time.RFC1123Z)),
fmt.Sprintf("%s - %s\n", toolname, time.Now().Format(time.RFC1123Z)),
fmt.Sprintf("Host: %s\n", opts.Host),
fmt.Sprintf("Skipping profiled queries on these collections: %v\n", opts.SkipCollections),
}
Expand Down
6 changes: 3 additions & 3 deletions src/go/pt-mongodb-query-digest/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,7 @@ func TestParseArgs(t *testing.T) {
want *cliOptions
}{
{
args: []string{TOOLNAME}, // arg[0] is the command itself
args: []string{toolname}, // arg[0] is the command itself
want: &cliOptions{
Host: "mongodb://" + DEFAULT_HOST,
LogLevel: DEFAULT_LOGLEVEL,
Expand All @@ -117,11 +117,11 @@ func TestParseArgs(t *testing.T) {
},
},
{
args: []string{TOOLNAME, "zapp.brannigan.net:27018/samples", "--help"},
args: []string{toolname, "zapp.brannigan.net:27018/samples", "--help"},
want: nil,
},
{
args: []string{TOOLNAME, "zapp.brannigan.net:27018/samples"},
args: []string{toolname, "zapp.brannigan.net:27018/samples"},
want: &cliOptions{
Host: "mongodb://zapp.brannigan.net:27018/samples",
LogLevel: DEFAULT_LOGLEVEL,
Expand Down
8 changes: 4 additions & 4 deletions src/go/pt-mongodb-summary/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ import (
)

const (
TOOLNAME = "pt-mongodb-summary"
toolname = "pt-mongodb-summary"

DefaultAuthDB = "admin"
DefaultHost = "mongodb://localhost:27017"
Expand Down Expand Up @@ -194,17 +194,17 @@ func main() {
log.SetLevel(logLevel)

if opts.Version {
fmt.Println(TOOLNAME)
fmt.Println(toolname)
fmt.Printf("Version %s\n", Version)
fmt.Printf("Build: %s using %s\n", Build, GoVersion)
fmt.Printf("Commit: %s\n", Commit)

return
}

conf := config.DefaultConfig(TOOLNAME)
conf := config.DefaultConfig(toolname)
if !conf.GetBool("no-version-check") && !opts.NoVersionCheck {
advice, err := versioncheck.CheckUpdates(TOOLNAME, Version)
advice, err := versioncheck.CheckUpdates(toolname, Version)
if err != nil {
log.Infof("cannot check version updates: %s", err.Error())
} else if advice != "" {
Expand Down
4 changes: 2 additions & 2 deletions src/go/pt-mongodb-summary/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ func TestParseArgs(t *testing.T) {
want *cliOptions
}{
{
args: []string{TOOLNAME}, // arg[0] is the command itself
args: []string{toolname}, // arg[0] is the command itself
want: &cliOptions{
Host: DefaultHost,
LogLevel: DefaultLogLevel,
Expand All @@ -112,7 +112,7 @@ func TestParseArgs(t *testing.T) {
},
},
{
args: []string{TOOLNAME, "zapp.brannigan.net:27018/samples", "--help"},
args: []string{toolname, "zapp.brannigan.net:27018/samples", "--help"},
want: nil,
},
}
Expand Down
4 changes: 2 additions & 2 deletions src/go/pt-pg-summary/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ import (
)

const (
TOOLNAME = "pt-pg-summary"
toolname = "pt-pg-summary"
)

// We do not set anything here, these variables are defined by the Makefile
Expand Down Expand Up @@ -215,7 +215,7 @@ func safeConnString(opts connOpts, dbName string) string {
}

func parseCommandLineOpts(args []string) (cliOptions, error) {
app := kingpin.New(TOOLNAME, "Percona Toolkit - PostgreSQL Summary")
app := kingpin.New(toolname, "Percona Toolkit - PostgreSQL Summary")
app.UsageWriter(os.Stdout)
// version, commit and date will be set at build time by the compiler -ldflags param
app.Version(fmt.Sprintf("%s\nVersion %s\nBuild: %s using %s\nCommit: %s",
Expand Down
8 changes: 4 additions & 4 deletions src/go/pt-pg-summary/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -129,13 +129,13 @@ func TestCollectPerDatabaseInfo(t *testing.T) {
Option --version
*/
func TestVersionOption(t *testing.T) {
out, err := exec.Command("../../../bin/"+TOOLNAME, "--version").Output()
out, err := exec.Command("../../../bin/"+toolname, "--version").Output()
if err != nil {
t.Errorf("error executing %s --version: %s", TOOLNAME, err.Error())
t.Errorf("error executing %s --version: %s", toolname, err.Error())
}
// We are using MustCompile here, because hard-coded RE should not fail
re := regexp.MustCompile(TOOLNAME + `\n.*Version v?\d+\.\d+\.\d+\n`)
re := regexp.MustCompile(toolname + `\n.*Version v?\d+\.\d+\.\d+\n`)
if !re.Match(out) {
t.Errorf("%s --version returns wrong result:\n%s", TOOLNAME, out)
t.Errorf("%s --version returns wrong result:\n%s", toolname, out)
}
}
6 changes: 3 additions & 3 deletions src/go/pt-secure-collect/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ type myDefaults struct {
}

const (
TOOLNAME = "pt-secure-collect"
toolname = "pt-secure-collect"

decryptCmd = "decrypt"
encryptCmd = "encrypt"
Expand Down Expand Up @@ -172,7 +172,7 @@ func processCliParams(baseTempPath string, usageWriter io.Writer) (*cliOptions,
}
msg += "\n "

app := kingpin.New(TOOLNAME, msg)
app := kingpin.New(toolname, msg)
if usageWriter != nil {
app.UsageWriter(usageWriter)
app.Terminate(nil)
Expand All @@ -181,7 +181,7 @@ func processCliParams(baseTempPath string, usageWriter io.Writer) (*cliOptions,
}

// Add support for --version flag
app.Version(TOOLNAME + "\nVersion " + Version + "\nBuild: " + Build + " using " + GoVersion +
app.Version(toolname + "\nVersion " + Version + "\nBuild: " + Build + " using " + GoVersion +
"\nCommit:" + Commit)

opts := &cliOptions{
Expand Down
8 changes: 4 additions & 4 deletions src/go/pt-secure-collect/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -45,13 +45,13 @@ func TestCollect(t *testing.T) {
Option --version
*/
func TestVersionOption(t *testing.T) {
out, err := exec.Command("../../../bin/"+TOOLNAME, "--version").Output()
out, err := exec.Command("../../../bin/"+toolname, "--version").Output()
if err != nil {
t.Errorf("error executing %s --version: %s", TOOLNAME, err.Error())
t.Errorf("error executing %s --version: %s", toolname, err.Error())
}
// We are using MustCompile here, because hard-coded RE should not fail
re := regexp.MustCompile(TOOLNAME + `\n.*Version v?\d+\.\d+\.\d+\n`)
re := regexp.MustCompile(toolname + `\n.*Version v?\d+\.\d+\.\d+\n`)
if !re.Match(out) {
t.Errorf("%s --version returns wrong result:\n%s", TOOLNAME, out)
t.Errorf("%s --version returns wrong result:\n%s", toolname, out)
}
}

0 comments on commit 854bd8e

Please sign in to comment.