Enable Debug via Flag or Env

Cobra Example

Use cobra to create a golang command line tool, enabling debug mode with global --debug=true flag or DEBUG=true environment variable:

package main

import (
	"fmt"
	"github.com/imroc/req/v3"
	"github.com/spf13/cobra"
	"log"
	"os"
)

var globalClient = req.C()

var enableDebug bool

var uuidCmd = cobra.Command{
	Use: "uuid",
	PersistentPreRun: func(cmd *cobra.Command, args []string) {
		if enableDebug { // Enable debug mode if `--enableDebug=true` or `DEBUG=true`.
			globalClient.EnableDumpAll()  // Dump all requests.
			globalClient.EnableDebugLog() // Output debug log.
		}
	},
	RunE: func(cmd *cobra.Command, args []string) error {
		var result struct {
			Uuid string `json:"uuid"`
		}

		resp, err := globalClient.R().
			SetSuccessResult(&result). // Read uuid response into struct.
			Get("https://httpbin.org/uuid")

		if err != nil {
			return err
		}

		if resp.IsSuccessState() { // Print uuid returned by the API.
			fmt.Println(result.Uuid)
		} else {
			fmt.Println("bad response")
		}
		return nil
	},
}

func init() {
	uuidCmd.PersistentFlags().BoolVar(&enableDebug, "debug", os.Getenv("DEBUG") == "true", "Enable debug mode")
}

func main() {
	if err := uuidCmd.Execute(); err != nil {
		log.Fatalln(err)
	}
}

Let’s build and take a look at the --help:

$ go build -o uuid
$ ./uuid --help
Usage:
  uuid [flags]

Flags:
      --debug   Enable debug mode
  -h, --help    help for uuid

Run it normally:

$ ./uuid
d576aaa4-56ad-4095-8929-97e2ee7fbb5d

Enable debug mode with flag --debug=true:

$ ./uuid --debug
2022/05/25 10:00:33.093949 DEBUG [req] HTTP/2 GET https://httpbin.org/uuid
:authority: httpbin.org
:method: GET
:path: /uuid
:scheme: https
accept-encoding: gzip
user-agent: req/v3 (https://github.com/imroc/req)

:status: 200
date: Wed, 25 May 2022 02:00:33 GMT
content-type: application/json
content-length: 53
server: gunicorn/19.9.0
access-control-allow-origin: *
access-control-allow-credentials: true

{
  "uuid": "d4ea37b6-455d-4ad4-a3ad-251fa606560a"
}

d4ea37b6-455d-4ad4-a3ad-251fa606560a

Enable debug mode with environment variable DEBUG=true:

$ DEBUG=true ./uuid
2022/05/25 10:13:59.303260 DEBUG [req] HTTP/2 GET https://httpbin.org/uuid
:authority: httpbin.org
:method: GET
:path: /uuid
:scheme: https
accept-encoding: gzip
user-agent: req/v3 (https://github.com/imroc/req)

:status: 200
date: Wed, 25 May 2022 02:13:59 GMT
content-type: application/json
content-length: 53
server: gunicorn/19.9.0
access-control-allow-origin: *
access-control-allow-credentials: true

{
  "uuid": "52ab1a27-a777-4d77-8eb5-81f7c0d9f694"
}

52ab1a27-a777-4d77-8eb5-81f7c0d9f694

Go Flag Example

Use the flag package of the go standard library to create a golang command line tool, enabling debug mode with global --debug=true flag or DEBUG=true environment variable:

package main

import (
	"flag"
	"fmt"
	"github.com/imroc/req/v3"
	"os"
)

var globalClient = req.C()

var enableDebug bool

func init() {
	flag.BoolVar(&enableDebug, "debug", os.Getenv("DEBUG") == "true", "Enable debug mode")
}

func main() {
	flag.Parse()
	if enableDebug { // Enable debug mode if `--enableDebug=true` or `DEBUG=true`.
		globalClient.EnableDumpAll()  // Dump all requests.
		globalClient.EnableDebugLog() // Output debug log.
	}
	var result struct {
		Uuid string `json:"uuid"`
	}

	resp, err := globalClient.R().
		SetSuccessResult(&result). // Read uuid response into struct.
		Get("https://httpbin.org/uuid")

	if err != nil {
		panic(err)
	}

	if resp.IsSuccessState() { // Print uuid returned by the API.
		fmt.Println(result.Uuid)
	} else {
		fmt.Println("bad response")
	}
}