Skip to content

Commit

Permalink
use single copy of instance cache for node_controller as well
Browse files Browse the repository at this point in the history
  • Loading branch information
rahulait committed Dec 17, 2024
1 parent bc69f45 commit b148f90
Show file tree
Hide file tree
Showing 4 changed files with 32 additions and 31 deletions.
11 changes: 6 additions & 5 deletions cloud/linode/cloud.go
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,8 @@ type linodeCloud struct {
routes cloudprovider.Routes
}

var instanceCache *instances

func init() {
cloudprovider.RegisterCloudProvider(
ProviderName,
Expand Down Expand Up @@ -97,9 +99,8 @@ func newCloud() (cloudprovider.Interface, error) {
Options.VPCNames = Options.VPCName
}

instances := newInstances(linodeClient)

routes, err := newRoutes(linodeClient, instances)
instanceCache = newInstances(linodeClient)
routes, err := newRoutes(linodeClient, instanceCache)
if err != nil {
return nil, fmt.Errorf("routes client was not created successfully: %w", err)
}
Expand All @@ -125,7 +126,7 @@ func newCloud() (cloudprovider.Interface, error) {
// create struct that satisfies cloudprovider.Interface
lcloud := &linodeCloud{
client: linodeClient,
instances: instances,
instances: instanceCache,
loadbalancers: newLoadbalancers(linodeClient, region),
routes: routes,
}
Expand All @@ -141,7 +142,7 @@ func (c *linodeCloud) Initialize(clientBuilder cloudprovider.ControllerClientBui
serviceController := newServiceController(c.loadbalancers.(*loadbalancers), serviceInformer)
go serviceController.Run(stopCh)

nodeController := newNodeController(kubeclient, c.client, nodeInformer)
nodeController := newNodeController(kubeclient, c.client, nodeInformer, instanceCache)
go nodeController.Run(stopCh)
}

Expand Down
4 changes: 2 additions & 2 deletions cloud/linode/node_controller.go
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ type nodeController struct {
queue workqueue.TypedDelayingInterface[any]
}

func newNodeController(kubeclient kubernetes.Interface, client client.Client, informer v1informers.NodeInformer) *nodeController {
func newNodeController(kubeclient kubernetes.Interface, client client.Client, informer v1informers.NodeInformer, instanceCache *instances) *nodeController {
timeout := defaultMetadataTTL
if raw, ok := os.LookupEnv("LINODE_METADATA_TTL"); ok {
if t, _ := strconv.Atoi(raw); t > 0 {
Expand All @@ -52,7 +52,7 @@ func newNodeController(kubeclient kubernetes.Interface, client client.Client, in

return &nodeController{
client: client,
instances: newInstances(client),
instances: instanceCache,
kubeclient: kubeclient,
informer: informer,
ttl: timeout,
Expand Down
4 changes: 2 additions & 2 deletions cloud/linode/route_controller.go
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@ type routes struct {
routeCache *routeCache
}

func newRoutes(client client.Client, instances *instances) (cloudprovider.Routes, error) {
func newRoutes(client client.Client, instanceCache *instances) (cloudprovider.Routes, error) {
timeout := 60
if raw, ok := os.LookupEnv("LINODE_ROUTES_CACHE_TTL_SECONDS"); ok {
if t, _ := strconv.Atoi(raw); t > 0 {
Expand All @@ -77,7 +77,7 @@ func newRoutes(client client.Client, instances *instances) (cloudprovider.Routes

return &routes{
client: client,
instances: instances,
instances: instanceCache,
routeCache: &routeCache{
routes: make(map[int][]linodego.VPCIP, 0),
ttl: time.Duration(timeout) * time.Second,
Expand Down
44 changes: 22 additions & 22 deletions cloud/linode/route_controller_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -33,8 +33,8 @@ func TestListRoutes(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

client.EXPECT().ListInstances(gomock.Any(), gomock.Any()).Times(1).Return([]linodego.Instance{}, nil)
Expand All @@ -57,8 +57,8 @@ func TestListRoutes(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

client.EXPECT().ListInstances(gomock.Any(), nil).Times(1).Return([]linodego.Instance{validInstance}, nil)
Expand All @@ -84,8 +84,8 @@ func TestListRoutes(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

client.EXPECT().ListInstances(gomock.Any(), nil).Times(1).Return([]linodego.Instance{validInstance}, nil)
Expand Down Expand Up @@ -126,8 +126,8 @@ func TestListRoutes(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

client.EXPECT().ListInstances(gomock.Any(), nil).Times(1).Return([]linodego.Instance{validInstance}, nil)
Expand Down Expand Up @@ -168,8 +168,8 @@ func TestListRoutes(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

client.EXPECT().ListInstances(gomock.Any(), nil).Times(1).Return([]linodego.Instance{validInstance}, nil)
Expand All @@ -184,8 +184,8 @@ func TestListRoutes(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

vpcIP2 := "10.0.0.3"
Expand Down Expand Up @@ -322,8 +322,8 @@ func TestCreateRoute(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

client.EXPECT().ListInstances(gomock.Any(), nil).Times(1).Return([]linodego.Instance{validInstance}, nil)
Expand All @@ -336,8 +336,8 @@ func TestCreateRoute(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

client.EXPECT().ListInstances(gomock.Any(), nil).Times(1).Return([]linodego.Instance{}, nil)
Expand Down Expand Up @@ -379,8 +379,8 @@ func TestDeleteRoute(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

client.EXPECT().ListInstances(gomock.Any(), nil).Times(1).Return([]linodego.Instance{}, nil)
Expand Down Expand Up @@ -410,8 +410,8 @@ func TestDeleteRoute(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

client.EXPECT().ListInstances(gomock.Any(), nil).Times(1).Return([]linodego.Instance{validInstance}, nil)
Expand Down Expand Up @@ -442,8 +442,8 @@ func TestDeleteRoute(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
client := mocks.NewMockClient(ctrl)
instances := newInstances(client)
routeController, err := newRoutes(client, instances)
instanceCache := newInstances(client)
routeController, err := newRoutes(client, instanceCache)
assert.NoError(t, err)

client.EXPECT().ListInstances(gomock.Any(), nil).Times(1).Return([]linodego.Instance{validInstance}, nil)
Expand Down

0 comments on commit b148f90

Please sign in to comment.