Skip to content

Commit

Permalink
Merge pull request #28 from deckhouse/feature/mirror-debug-logging
Browse files Browse the repository at this point in the history
[mirror] Add mirror debug logging
  • Loading branch information
name212 authored Jul 25, 2024
2 parents 8e7d774 + 74e3e98 commit 17fe746
Showing 1 changed file with 22 additions and 2 deletions.
24 changes: 22 additions & 2 deletions internal/mirror/cmd/mirror.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,14 +17,18 @@ limitations under the License.
package mirror

import (
"os"
"strconv"

"github.com/google/go-containerregistry/pkg/logs"
"github.com/spf13/cobra"
"k8s.io/component-base/logs"
"k8s.io/kubectl/pkg/util/templates"

"github.com/deckhouse/deckhouse-cli/internal/mirror/cmd/modules"
"github.com/deckhouse/deckhouse-cli/internal/mirror/cmd/pull"
"github.com/deckhouse/deckhouse-cli/internal/mirror/cmd/push"
"github.com/deckhouse/deckhouse-cli/internal/mirror/cmd/vulndb"
"github.com/deckhouse/deckhouse-cli/internal/mirror/util/log"
)

var mirrorLong = templates.LongDesc(`
Expand Down Expand Up @@ -54,6 +58,22 @@ func NewCommand() *cobra.Command {
vulndb.NewCommand(),
)

logs.AddFlags(mirrorCmd.PersistentFlags())
debugLogStr := os.Getenv("MIRROR_DEBUG_LOG")
if debugLogStr != "" {
debugLogLevel, err := strconv.Atoi(debugLogStr)
if err != nil {
log.WarnF("Invalid $MIRROR_DEBUG_LOG: %v\nUse 1 for progress logging, 2 for warnings or 3 for connection logging. Each level also enables previous ones.\n", err)
}

switch {
case debugLogLevel >= 3:
logs.Debug.SetOutput(os.Stderr)
case debugLogLevel >= 2:
logs.Warn.SetOutput(os.Stderr)
case debugLogLevel >= 1:
logs.Progress.SetOutput(os.Stderr)
}
}

return mirrorCmd
}

0 comments on commit 17fe746

Please sign in to comment.