Skip to content

Commit 2155667

Browse files
authored
[ws-manager-mk2] Use common-go/log (#16921)
1 parent 16b55bb commit 2155667

File tree

4 files changed

+23
-8
lines changed

4 files changed

+23
-8
lines changed

components/ws-manager-mk2/BUILD.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ packages:
1717
- components/image-builder-api/go:lib
1818
config:
1919
packaging: app
20-
buildCommand: ["go", "build", "-trimpath", "-ldflags", "-buildid= -w -s -X 'github.com/gitpod-io/gitpod/ws-manager-mk2/cmd.Version=commit-${__git_commit}'"]
20+
buildCommand: ["go", "build", "-trimpath", "-ldflags", "-buildid= -w -s -X 'main.Version=commit-${__git_commit}'"]
2121
- name: docker
2222
type: docker
2323
deps:

components/ws-manager-mk2/go.mod

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ require (
5151
github.com/Azure/go-autorest/tracing v0.6.0 // indirect
5252
github.com/asaskevich/govalidator v0.0.0-20210307081110-f21760c49a8d // indirect
5353
github.com/beorn7/perks v1.0.1 // indirect
54+
github.com/bombsimon/logrusr/v2 v2.0.1 // indirect
5455
github.com/cespare/xxhash/v2 v2.1.2 // indirect
5556
github.com/davecgh/go-spew v1.1.1 // indirect
5657
github.com/emicklei/go-restful v2.9.5+incompatible // indirect

components/ws-manager-mk2/go.sum

Lines changed: 4 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

components/ws-manager-mk2/main.go

Lines changed: 17 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -19,16 +19,17 @@ import (
1919
"google.golang.org/grpc/credentials/insecure"
2020
_ "k8s.io/client-go/plugin/pkg/client/auth"
2121

22+
"github.com/bombsimon/logrusr/v2"
2223
grpc_prometheus "github.com/grpc-ecosystem/go-grpc-prometheus"
2324
"github.com/prometheus/client_golang/prometheus"
2425
"k8s.io/apimachinery/pkg/runtime"
2526
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
2627
clientgoscheme "k8s.io/client-go/kubernetes/scheme"
28+
"k8s.io/klog/v2"
2729
ctrl "sigs.k8s.io/controller-runtime"
2830
"sigs.k8s.io/controller-runtime/pkg/cache"
2931
"sigs.k8s.io/controller-runtime/pkg/client"
3032
"sigs.k8s.io/controller-runtime/pkg/healthz"
31-
"sigs.k8s.io/controller-runtime/pkg/log/zap"
3233
"sigs.k8s.io/controller-runtime/pkg/metrics"
3334

3435
common_grpc "github.com/gitpod-io/gitpod/common-go/grpc"
@@ -50,6 +51,11 @@ import (
5051
)
5152

5253
var (
54+
// ServiceName is the name we use for tracing/logging
55+
ServiceName = "ws-manager-mk2"
56+
// Version of this service - set during build
57+
Version = ""
58+
5359
scheme = runtime.NewScheme()
5460
setupLog = ctrl.Log.WithName("setup")
5561
)
@@ -64,17 +70,21 @@ func init() {
6470
func main() {
6571
var enableLeaderElection bool
6672
var configFN string
73+
var jsonLog bool
74+
var verbose bool
6775
flag.BoolVar(&enableLeaderElection, "leader-elect", false,
6876
"Enable leader election for controller manager. "+
6977
"Enabling this will ensure there is only one active controller manager.")
7078
flag.StringVar(&configFN, "config", "", "Path to the config file")
71-
opts := zap.Options{
72-
Development: true,
73-
}
74-
opts.BindFlags(flag.CommandLine)
79+
flag.BoolVar(&jsonLog, "json-log", true, "produce JSON log output on verbose level")
80+
flag.BoolVar(&verbose, "verbose", false, "Enable verbose logging")
7581
flag.Parse()
7682

77-
ctrl.SetLogger(zap.New(zap.UseFlagOptions(&opts)))
83+
log.Init(ServiceName, Version, jsonLog, verbose)
84+
baseLogger := logrusr.New(log.Log)
85+
ctrl.SetLogger(baseLogger)
86+
// Set the logger used by k8s (e.g. client-go).
87+
klog.SetLogger(baseLogger)
7888
promrep := &tracing.PromReporter{
7989
Operations: map[string]tracing.SpanMetricMapping{
8090
"StartWorkspace": {
@@ -84,7 +94,7 @@ func main() {
8494
},
8595
},
8696
}
87-
closer := tracing.Init("ws-manager-mk2", tracing.WithPrometheusReporter(promrep))
97+
closer := tracing.Init(ServiceName, tracing.WithPrometheusReporter(promrep))
8898
if closer != nil {
8999
defer closer.Close()
90100
}

0 commit comments

Comments
 (0)