Skip to content

Commit

Permalink
use single copy of instances
Browse files Browse the repository at this point in the history
  • Loading branch information
rahulait committed Dec 17, 2024
1 parent a82417c commit bc69f45
Show file tree
Hide file tree
Showing 3 changed files with 30 additions and 16 deletions.
6 changes: 4 additions & 2 deletions cloud/linode/cloud.go
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,9 @@ func newCloud() (cloudprovider.Interface, error) {
Options.VPCNames = Options.VPCName
}

routes, err := newRoutes(linodeClient)
instances := newInstances(linodeClient)

routes, err := newRoutes(linodeClient, instances)
if err != nil {
return nil, fmt.Errorf("routes client was not created successfully: %w", err)
}
Expand All @@ -123,7 +125,7 @@ func newCloud() (cloudprovider.Interface, error) {
// create struct that satisfies cloudprovider.Interface
lcloud := &linodeCloud{
client: linodeClient,
instances: newInstances(linodeClient),
instances: instances,
loadbalancers: newLoadbalancers(linodeClient, region),
routes: routes,
}
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) (cloudprovider.Routes, error) {
func newRoutes(client client.Client, instances *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) (cloudprovider.Routes, error) {

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

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

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

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

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

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

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

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

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

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

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

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

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

0 comments on commit bc69f45

Please sign in to comment.