From fe3d720efbe51d08df9706b2e72d2ed92b995da8 Mon Sep 17 00:00:00 2001
From: Fan Shang Xiang <shafan@microsoft.com>
Date: Fri, 13 Dec 2024 12:50:40 +0800
Subject: [PATCH] init sub repo in storage account package

---
 pkg/provider/storage/azure_storageaccount.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/pkg/provider/storage/azure_storageaccount.go b/pkg/provider/storage/azure_storageaccount.go
index 4720cbad82..a031f7d956 100644
--- a/pkg/provider/storage/azure_storageaccount.go
+++ b/pkg/provider/storage/azure_storageaccount.go
@@ -30,7 +30,6 @@ import (
 	"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/network/armnetwork/v6"
 	privatedns "github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/privatedns/armprivatedns"
 	"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/storage/armstorage"
-	"github.com/Azure/go-autorest/autorest/azure"
 	"github.com/google/uuid"
 	"k8s.io/klog/v2"
 	"k8s.io/utils/ptr"
@@ -102,7 +101,7 @@ type accountWithLocation struct {
 }
 type AccountRepo struct {
 	azureconfig.Config
-	Environment          azure.Environment
+	Environment          *azclient.Environment
 	ComputeClientFactory azclient.ClientFactory
 	NetworkClientFactory azclient.ClientFactory
 	subnetRepo           subnet.Repository
@@ -111,7 +110,7 @@ type AccountRepo struct {
 	lockMap              *lockmap.LockMap
 }
 
-func NewRepository(config azureconfig.Config, env azure.Environment, computeClientFactory azclient.ClientFactory, networkClientFactory azclient.ClientFactory) (*AccountRepo, error) {
+func NewRepository(config azureconfig.Config, env *azclient.Environment, computeClientFactory azclient.ClientFactory, networkClientFactory azclient.ClientFactory) (*AccountRepo, error) {
 	getter := func(_ context.Context, _ string) (*armstorage.Account, error) { return nil, nil }
 	storageAccountCache, err := cache.NewTimedCache(time.Minute, getter, config.DisableAPICallCache)
 	if err != nil {
@@ -121,12 +120,17 @@ func NewRepository(config azureconfig.Config, env azure.Environment, computeClie
 	if err != nil {
 		return nil, err
 	}
+	subnetRepo, err := subnet.NewRepo(networkClientFactory.GetSubnetClient())
+	if err != nil {
+		return nil, err
+	}
 	return &AccountRepo{
 		Config:               config,
 		Environment:          env,
 		fileServiceRepo:      fileserviceRepo,
 		ComputeClientFactory: computeClientFactory,
 		NetworkClientFactory: networkClientFactory,
+		subnetRepo:           subnetRepo,
 		storageAccountCache:  storageAccountCache,
 		lockMap:              lockmap.NewLockMap(),
 	}, nil