@@ -17,11 +17,12 @@ limitations under the License.
1717package openstack
1818
1919import (
20+ "context"
2021 "fmt"
2122
2223 "github.com/go-logr/logr"
23- limits "github.com/gophercloud/gophercloud/openstack/identity/v3/limits"
24- registeredlimits "github.com/gophercloud/gophercloud/openstack/identity/v3/registeredlimits"
24+ limits "github.com/gophercloud/gophercloud/v2/ openstack/identity/v3/limits"
25+ registeredlimits "github.com/gophercloud/gophercloud/v2/ openstack/identity/v3/registeredlimits"
2526)
2627
2728// Limit -
@@ -50,12 +51,13 @@ type Limit struct {
5051
5152// CreateLimit - create a limit in keystone for particular project if it does not exist
5253func (o * OpenStack ) CreateLimit (
54+ ctx context.Context ,
5355 log logr.Logger ,
5456 l Limit ,
5557) (string , error ) {
5658 var limitID string
5759
58- allPages , err := limits .List (o .osclient , limits.ListOpts {ResourceName : l .ResourceName }).AllPages ()
60+ allPages , err := limits .List (o .osclient , limits.ListOpts {ResourceName : l .ResourceName }).AllPages (ctx )
5961 if err != nil {
6062 return limitID , err
6163 }
@@ -80,7 +82,7 @@ func (o *OpenStack) CreateLimit(
8082 },
8183 }
8284 log .Info (fmt .Sprintf ("Creating limit %s" , l .ResourceName ))
83- createdLimits , err := limits .BatchCreate (o .osclient , createOpts ).Extract ()
85+ createdLimits , err := limits .BatchCreate (ctx , o .osclient , createOpts ).Extract ()
8486 if err != nil {
8587 return limitID , err
8688 }
@@ -112,12 +114,13 @@ type RegisteredLimit struct {
112114
113115// CreateOrUpdateRegisteredLimit - create or update limit in keystone (global across projects) if it does not exist
114116func (o * OpenStack ) CreateOrUpdateRegisteredLimit (
117+ ctx context.Context ,
115118 log logr.Logger ,
116119 l RegisteredLimit ,
117120) (string , error ) {
118121 var limitID string
119122
120- allPages , err := registeredlimits .List (o .osclient , registeredlimits.ListOpts {ResourceName : l .ResourceName }).AllPages ()
123+ allPages , err := registeredlimits .List (o .osclient , registeredlimits.ListOpts {ResourceName : l .ResourceName }).AllPages (ctx )
121124 if err != nil {
122125 return limitID , err
123126 }
@@ -134,7 +137,7 @@ func (o *OpenStack) CreateOrUpdateRegisteredLimit(
134137 DefaultLimit : & l .DefaultLimit ,
135138 }
136139 log .Info (fmt .Sprintf ("Updating registered limit %s" , l .ResourceName ))
137- _ , err := registeredlimits .Update (o .osclient , limitID , updateOpts ).Extract ()
140+ _ , err := registeredlimits .Update (ctx , o .osclient , limitID , updateOpts ).Extract ()
138141 if err != nil {
139142 return limitID , err
140143 }
@@ -150,7 +153,7 @@ func (o *OpenStack) CreateOrUpdateRegisteredLimit(
150153 },
151154 }
152155 log .Info (fmt .Sprintf ("Creating registered limit %s" , l .ResourceName ))
153- createdLimits , err := registeredlimits .BatchCreate (o .osclient , createOpts ).Extract ()
156+ createdLimits , err := registeredlimits .BatchCreate (ctx , o .osclient , createOpts ).Extract ()
154157 if err != nil {
155158 return limitID , err
156159 }
@@ -164,11 +167,12 @@ func (o *OpenStack) CreateOrUpdateRegisteredLimit(
164167
165168// DeleteRegisteredLimit - delete limit from keystone
166169func (o * OpenStack ) DeleteRegisteredLimit (
170+ ctx context.Context ,
167171 log logr.Logger ,
168172 registeredLimitID string ,
169173) error {
170174 log .Info (fmt .Sprintf ("Deleting registered limit %s" , registeredLimitID ))
171- err := registeredlimits .Delete (o .osclient , registeredLimitID ).ExtractErr ()
175+ err := registeredlimits .Delete (ctx , o .osclient , registeredLimitID ).ExtractErr ()
172176 if err != nil {
173177 return err
174178 }
@@ -177,11 +181,12 @@ func (o *OpenStack) DeleteRegisteredLimit(
177181
178182// GetRegisteredLimit - Get existing registered limit by ID
179183func (o * OpenStack ) GetRegisteredLimit (
184+ ctx context.Context ,
180185 log logr.Logger ,
181186 registeredLimitID string ,
182187) (* registeredlimits.RegisteredLimit , error ) {
183188 log .Info (fmt .Sprintf ("Fetching registered limit %s" , registeredLimitID ))
184- registeredLimit , err := registeredlimits .Get (o .osclient , registeredLimitID ).Extract ()
189+ registeredLimit , err := registeredlimits .Get (ctx , o .osclient , registeredLimitID ).Extract ()
185190 if err != nil {
186191 return nil , err
187192 }
@@ -190,6 +195,7 @@ func (o *OpenStack) GetRegisteredLimit(
190195
191196// ListRegisteredLimitsByResourceName - List all registered limits filtered by resource name
192197func (o * OpenStack ) ListRegisteredLimitsByResourceName (
198+ ctx context.Context ,
193199 log logr.Logger ,
194200 resourceName string ,
195201) ([]registeredlimits.RegisteredLimit , error ) {
@@ -198,7 +204,7 @@ func (o *OpenStack) ListRegisteredLimitsByResourceName(
198204 }
199205
200206 log .Info (fmt .Sprintf ("Fetching registered limit %s" , resourceName ))
201- allPages , err := registeredlimits .List (o .osclient , listOpts ).AllPages ()
207+ allPages , err := registeredlimits .List (o .osclient , listOpts ).AllPages (ctx )
202208 if err != nil {
203209 return nil , err
204210 }
@@ -212,6 +218,7 @@ func (o *OpenStack) ListRegisteredLimitsByResourceName(
212218
213219// ListRegisteredLimitsByServiceID - List all registered limits filtered by service id
214220func (o * OpenStack ) ListRegisteredLimitsByServiceID (
221+ ctx context.Context ,
215222 log logr.Logger ,
216223 serviceID string ,
217224) ([]registeredlimits.RegisteredLimit , error ) {
@@ -220,7 +227,7 @@ func (o *OpenStack) ListRegisteredLimitsByServiceID(
220227 }
221228
222229 log .Info (fmt .Sprintf ("Fetching registered limit for service %s" , serviceID ))
223- allPages , err := registeredlimits .List (o .osclient , listOpts ).AllPages ()
230+ allPages , err := registeredlimits .List (o .osclient , listOpts ).AllPages (ctx )
224231 if err != nil {
225232 return nil , err
226233 }
0 commit comments