Skip to content

Commit 9045ee9

Browse files
authored
Merge pull request #1158 from srm09/add-logger-to-session
Adds log statements while get/create session
2 parents ef90eed + 64a95bd commit 9045ee9

File tree

7 files changed

+16
-9
lines changed

7 files changed

+16
-9
lines changed

controllers/vspherecluster_controller.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -335,7 +335,7 @@ func cloudProviderConfigurationAvailable(ctx *context.ClusterContext) bool {
335335
}
336336

337337
func (r clusterReconciler) reconcileVCenterConnectivity(ctx *context.ClusterContext) error {
338-
_, err := session.GetOrCreate(ctx, ctx.VSphereCluster.Spec.Server,
338+
_, err := session.GetOrCreate(ctx, r.Logger, ctx.VSphereCluster.Spec.Server,
339339
ctx.VSphereCluster.Spec.CloudProviderConfiguration.Workspace.Datacenter, ctx.Username, ctx.Password, ctx.VSphereCluster.Spec.Thumbprint)
340340
return err
341341
}

controllers/vspherevm_controller.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ func (r vmReconciler) Reconcile(req ctrl.Request) (_ ctrl.Result, reterr error)
138138

139139
// Get or create an authenticated session to the vSphere endpoint.
140140
authSession, err := session.GetOrCreate(r.Context,
141-
vsphereVM.Spec.Server, vsphereVM.Spec.Datacenter,
141+
r.Logger, vsphereVM.Spec.Server, vsphereVM.Spec.Datacenter,
142142
r.ControllerManagerContext.Username, r.ControllerManagerContext.Password, vsphereVM.Spec.Thumbprint)
143143
if err != nil {
144144
return reconcile.Result{}, errors.Wrap(err, "failed to create vSphere session")

go.mod

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ go 1.15
44

55
require (
66
github.com/antihax/optional v1.0.0
7-
github.com/go-logr/logr v0.1.0
7+
github.com/go-logr/logr v0.4.0
88
github.com/google/go-cmp v0.4.1
99
github.com/google/uuid v1.1.1
1010
github.com/onsi/ginkgo v1.12.1

go.sum

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,8 +109,9 @@ github.com/go-kit/kit v0.8.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2
109109
github.com/go-kit/kit v0.9.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=
110110
github.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE=
111111
github.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk=
112-
github.com/go-logr/logr v0.1.0 h1:M1Tv3VzNlEHg6uyACnRdtrploV2P7wZqH8BoQMtz0cg=
113112
github.com/go-logr/logr v0.1.0/go.mod h1:ixOQHD9gLJUVQQ2ZOR7zLEifBX6tGkNJF4QyIY7sIas=
113+
github.com/go-logr/logr v0.4.0 h1:K7/B1jt6fIBQVd4Owv2MqGQClcgf0R266+7C/QjRcLc=
114+
github.com/go-logr/logr v0.4.0/go.mod h1:z6/tIYblkpsD+a4lm/fGIIU9mZ+XfAiaFtq7xTgseGU=
114115
github.com/go-logr/zapr v0.1.0 h1:h+WVe9j6HAA01niTJPA/kKH0i7e0rLZBCwauQFcRE54=
115116
github.com/go-logr/zapr v0.1.0/go.mod h1:tabnROwaDl0UNxkVeFRbY8bwB37GwRv0P8lg6aAiEnk=
116117
github.com/go-openapi/analysis v0.0.0-20180825180245-b006789cd277/go.mod h1:k70tL6pCuVxPJOHXQ+wIac1FUrvNkHolPie/cLEU6hI=

pkg/services/govmomi/create_test.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ import (
2020
"crypto/tls"
2121
"testing"
2222

23+
"github.com/go-logr/logr"
2324
"github.com/vmware/govmomi/object"
2425
"github.com/vmware/govmomi/simulator"
2526
"github.com/vmware/govmomi/vim25/types"
@@ -48,6 +49,7 @@ func TestCreate(t *testing.T) {
4849

4950
authSession, err := session.GetOrCreate(
5051
vmContext,
52+
logr.DiscardLogger{},
5153
vmContext.VSphereVM.Spec.Server, "",
5254
s.URL.User.Username(), pass, "")
5355
if err != nil {

pkg/services/govmomi/vcenter/clone_test.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ import (
2121
"crypto/tls"
2222
"testing"
2323

24+
"github.com/go-logr/logr"
2425
"github.com/vmware/govmomi/object"
2526
"github.com/vmware/govmomi/simulator"
2627
"github.com/vmware/govmomi/vim25/types"
@@ -143,6 +144,7 @@ func initSimulator(t *testing.T) (*simulator.Model, *session.Session, *simulator
143144

144145
authSession, err := session.GetOrCreate(
145146
ctx.TODO(),
147+
logr.DiscardLogger{},
146148
server.URL.Host, "",
147149
server.URL.User.Username(), pass, "")
148150
if err != nil {

pkg/session/session.go

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ import (
2121
"net/url"
2222
"sync"
2323

24+
"github.com/go-logr/logr"
2425
"github.com/pkg/errors"
2526
"github.com/vmware/govmomi"
2627
"github.com/vmware/govmomi/find"
@@ -46,15 +47,17 @@ type Session struct {
4647
// already exist.
4748
func GetOrCreate(
4849
ctx context.Context,
50+
logger logr.Logger,
4951
server, datacenter, username, password string, thumbprint string) (*Session, error) {
5052

5153
sessionMU.Lock()
5254
defer sessionMU.Unlock()
5355

5456
sessionKey := server + username + datacenter
55-
if session, ok := sessionCache[sessionKey]; ok {
56-
if ok, _ := session.SessionManager.SessionIsActive(ctx); ok {
57-
return &session, nil
57+
if cachedSession, ok := sessionCache[sessionKey]; ok {
58+
if ok, _ := cachedSession.SessionManager.SessionIsActive(ctx); ok {
59+
logger.V(2).Info("found active cached vSphere client session", "server", server, "datacenter", datacenter)
60+
return &cachedSession, nil
5861
}
5962
}
6063

@@ -89,8 +92,7 @@ func GetOrCreate(
8992
// Cache the session.
9093
sessionCache[sessionKey] = session
9194

92-
// TODO(akutz) Reintroduce the logger.
93-
//ctx.Logger.V(2).Info("cached vSphere client session", "server", server, "datacenter", datacenter)
95+
logger.V(2).Info("cached vSphere client session", "server", server, "datacenter", datacenter)
9496

9597
return &session, nil
9698
}

0 commit comments

Comments
 (0)