Skip to content

Commit 4e125d0

Browse files
deryrahmanirainia
authored andcommitted
refactor: fix unproper logger print
1 parent 6a1ec5a commit 4e125d0

File tree

4 files changed

+14
-30
lines changed

4 files changed

+14
-30
lines changed

cmd/commands.go

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
11
package cmd
22

33
import (
4-
"os"
5-
64
"github.com/MakeNowJust/heredoc"
75
"github.com/odpf/salt/cmdx"
86
cli "github.com/spf13/cobra"
@@ -12,7 +10,6 @@ import (
1210
"github.com/odpf/optimus/cmd/deploy"
1311
"github.com/odpf/optimus/cmd/extension"
1412
"github.com/odpf/optimus/cmd/initialize"
15-
"github.com/odpf/optimus/cmd/internal/logger"
1613
"github.com/odpf/optimus/cmd/job"
1714
"github.com/odpf/optimus/cmd/namespace"
1815
"github.com/odpf/optimus/cmd/playground"
@@ -23,18 +20,13 @@ import (
2320
"github.com/odpf/optimus/cmd/secret"
2421
"github.com/odpf/optimus/cmd/serve"
2522
"github.com/odpf/optimus/cmd/version"
26-
"github.com/odpf/optimus/utils"
2723
)
2824

2925
// New constructs the 'root' command. It houses all other sub commands
3026
// default output of logging should go to stdout
3127
// interactive output like progress bars should go to stderr
3228
// unless the stdout/err is a tty, colors/progressbar should be disabled
3329
func New() *cli.Command {
34-
if utils.IsTerminal(os.Stdout) {
35-
logger.InitializeColor()
36-
}
37-
3830
cmd := &cli.Command{
3931
Use: "optimus <command> <subcommand> [flags]",
4032
Long: heredoc.Doc(`

cmd/deploy/deploy.go

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ func (d *deployCommand) PreRunE(_ *cobra.Command, _ []string) error {
8686
}
8787

8888
func (d *deployCommand) RunE(_ *cobra.Command, _ []string) error {
89-
d.logger.Info("Registering project [%s] to [%s]", d.clientConfig.Project.Name, d.clientConfig.Host)
89+
d.logger.Info(fmt.Sprintf("Registering project [%s] to [%s]", d.clientConfig.Project.Name, d.clientConfig.Host))
9090
if err := project.RegisterProject(d.logger, d.clientConfig.Host, d.clientConfig.Project); err != nil {
9191
return err
9292
}
@@ -102,7 +102,7 @@ func (d *deployCommand) RunE(_ *cobra.Command, _ []string) error {
102102
}
103103
d.logger.Info("validation finished!\n")
104104

105-
d.logger.Info("Registering namespaces for [%s] to [%s]", d.clientConfig.Project.Name, d.clientConfig.Host)
105+
d.logger.Info(fmt.Sprintf("Registering namespaces for [%s] to [%s]", d.clientConfig.Project.Name, d.clientConfig.Host))
106106
if err := namespace.RegisterSelectedNamespaces(d.logger, d.clientConfig.Host, d.clientConfig.Project.Name, selectedNamespaces...); err != nil {
107107
return err
108108
}
@@ -146,7 +146,7 @@ func (d *deployCommand) deployJobs(conn *connectivity.Connectivity, selectedName
146146
for _, namespace := range selectedNamespaces {
147147
namespaceNames = append(namespaceNames, namespace.Name)
148148
}
149-
d.logger.Info("\n> Deploying jobs from namespaces [%s]", strings.Join(namespaceNames, ","))
149+
d.logger.Info(fmt.Sprintf("\n> Deploying jobs from namespaces [%s]", strings.Join(namespaceNames, ",")))
150150

151151
stream, err := d.getJobStreamClient(conn)
152152
if err != nil {
@@ -455,19 +455,19 @@ func PollJobDeployment(ctx context.Context, l log.Logger, jobSpecService pb.JobS
455455
if len(resp.Failures) > 0 {
456456
for _, failedJob := range resp.Failures {
457457
if failedJob.GetJobName() != "" {
458-
l.Error("Unable to deploy job %s: %s", failedJob.GetJobName(), failedJob.GetMessage())
458+
l.Error(fmt.Sprintf("Unable to deploy job %s: %s", failedJob.GetJobName(), failedJob.GetMessage()))
459459
} else {
460-
l.Error("Job deployment failed: %s", failedJob.GetMessage())
460+
l.Error(fmt.Sprintf("Job deployment failed: %s", failedJob.GetMessage()))
461461
}
462462
}
463463
}
464464
if len(resp.UnknownDependencies) > 0 {
465465
l.Error("Unable to create sensors for below jobs:")
466466
for jobName, dependencies := range resp.UnknownDependencies {
467-
l.Error("- %s: invalid dependency name(s): %s.", jobName, dependencies)
467+
l.Error(fmt.Sprintf("- %s: invalid dependency name(s): %s.", jobName, dependencies))
468468
}
469469
}
470-
l.Error("Deployed %d/%d jobs.", resp.SuccessCount, resp.SuccessCount+resp.FailureCount)
470+
l.Error(fmt.Sprintf("Deployed %d/%d jobs.", resp.SuccessCount, resp.SuccessCount+resp.FailureCount))
471471
return errors.New("job deployment failed")
472472
}
473473

cmd/internal/logger/color.go

Lines changed: 6 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,17 @@
11
package logger
22

33
import (
4-
"fmt"
5-
64
"github.com/muesli/termenv"
75
"github.com/odpf/salt/term"
86
)
97

108
var (
11-
// ColoredSuccess format message with color for success
12-
ColoredSuccess = fmt.Sprintf
13-
14-
tp = termenv.EnvColorProfile()
15-
ColorYellow = tp.Color("#FFAF00")
16-
ColorWhite = tp.Color("#FFFFFF")
17-
ColorRed = tp.Color("#D70000")
18-
)
19-
20-
func InitializeColor() {
21-
cs := term.NewColorScheme()
9+
cs = term.NewColorScheme()
2210
ColoredSuccess = func(s string, a ...interface{}) string {
2311
return cs.Greenf(s, a...)
2412
}
25-
}
13+
14+
tp = termenv.EnvColorProfile()
15+
ColorYellow = tp.Color("#DBAB79")
16+
ColorRed = tp.Color("#E88388")
17+
)

cmd/internal/logger/logger.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ import (
1313
type colorFormatter int
1414

1515
func (*colorFormatter) Format(entry *logrus.Entry) ([]byte, error) {
16-
var colorcode = ColorWhite
16+
var colorcode termenv.Color
1717
switch entry.Level {
1818
case logrus.ErrorLevel, logrus.FatalLevel, logrus.PanicLevel:
1919
colorcode = ColorRed

0 commit comments

Comments
 (0)