root: simple output format on logrus for parity with cli

Signed-off-by: CrazyMax <crazy-max@users.noreply.github.com>
docker · Jan 25, 2022 · d3e56ea · d3e56ea
1 parent 278f94a
commit d3e56ea
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/commands/root.go b/commands/root.go
@@ -28,6 +28,8 @@ func NewRootCmd(name string, isPlugin bool, dockerCli command.Cli) *cobra.Comman
 		}
 	}
 
+	logrus.SetFormatter(&logutil.Formatter{})
+
 	logrus.AddHook(logutil.NewFilter([]logrus.Level{
 		logrus.DebugLevel,
 	},

diff --git a/util/logutil/format.go b/util/logutil/format.go
@@ -0,0 +1,16 @@
+package logutil
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/sirupsen/logrus"
+)
+
+type Formatter struct {
+	logrus.TextFormatter
+}
+
+func (f *Formatter) Format(entry *logrus.Entry) ([]byte, error) {
+	return []byte(fmt.Sprintf("%s: %s\n", strings.ToUpper(entry.Level.String()), entry.Message)), nil
+}