Skip to content

Commit 9a1f8c9

Browse files
refactor: remove debug flag from pkg/server (#2239)
1 parent f44a435 commit 9a1f8c9

File tree

4 files changed

+6
-9
lines changed

4 files changed

+6
-9
lines changed

cmd/tracee-ebpf/main.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ func main() {
206206
}
207207

208208
if server.ShouldStart(c) {
209-
httpServer := server.New(c.String(server.ListenEndpointFlag), debug)
209+
httpServer := server.New(c.String(server.ListenEndpointFlag))
210210

211211
if c.Bool(server.MetricsEndpointFlag) {
212212
err := t.Stats().RegisterPrometheus()

cmd/tracee-rules/main.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ func main() {
139139
}
140140

141141
if server.ShouldStart(c) {
142-
httpServer := server.New(c.String(server.ListenEndpointFlag), false)
142+
httpServer := server.New(c.String(server.ListenEndpointFlag))
143143

144144
if c.Bool(server.MetricsEndpointFlag) {
145145
err := e.Stats().RegisterPrometheus()

pkg/server/server.go

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import (
66
"net/http/pprof"
77
"os"
88

9+
"github.com/aquasecurity/tracee/pkg/logger"
910
"github.com/prometheus/client_golang/prometheus/promhttp"
1011
"github.com/urfave/cli/v2"
1112
)
@@ -21,15 +22,13 @@ const (
2122
type Server struct {
2223
mux *http.ServeMux
2324
listenAddr string
24-
debug bool
2525
}
2626

2727
// New creates a new server
28-
func New(listenAddr string, debug bool) *Server {
28+
func New(listenAddr string) *Server {
2929
return &Server{
3030
mux: http.NewServeMux(),
3131
listenAddr: listenAddr,
32-
debug: debug,
3332
}
3433
}
3534

@@ -47,9 +46,7 @@ func (s *Server) EnableHealthzEndpoint() {
4746

4847
// Start starts the http server on the listen addr
4948
func (s *Server) Start() {
50-
if s.debug {
51-
fmt.Fprintf(os.Stdout, "Serving metrics endpoint at %s\n", s.listenAddr)
52-
}
49+
logger.Debug(fmt.Sprintf("Serving metrics endpoint at %s\n", s.listenAddr))
5350

5451
if err := http.ListenAndServe(s.listenAddr, s.mux); err != http.ErrServerClosed {
5552
fmt.Fprintf(os.Stderr, "Error serving metrics endpoint: %v\n", err)

pkg/server/server_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ import (
1010
)
1111

1212
func TestServer(t *testing.T) {
13-
httpServer := New("", false)
13+
httpServer := New("")
1414

1515
httpServer.EnableMetricsEndpoint()
1616
httpServer.EnableHealthzEndpoint()

0 commit comments

Comments
 (0)