1package network
2
3// Copyright (c) Microsoft Corporation. All rights reserved.
4// Licensed under the MIT License. See License.txt in the project root for license information.
5//
6// Code generated by Microsoft (R) AutoRest Code Generator.
7// Changes may cause incorrect behavior and will be lost if the code is regenerated.
8
9import (
10	"context"
11	"github.com/Azure/go-autorest/autorest"
12	"github.com/Azure/go-autorest/autorest/azure"
13	"github.com/Azure/go-autorest/autorest/validation"
14	"github.com/Azure/go-autorest/tracing"
15	"net/http"
16)
17
18// UsagesClient is the network Client
19type UsagesClient struct {
20	BaseClient
21}
22
23// NewUsagesClient creates an instance of the UsagesClient client.
24func NewUsagesClient(subscriptionID string) UsagesClient {
25	return NewUsagesClientWithBaseURI(DefaultBaseURI, subscriptionID)
26}
27
28// NewUsagesClientWithBaseURI creates an instance of the UsagesClient client using a custom endpoint.  Use this when
29// interacting with an Azure cloud that uses a non-standard base URI (sovereign clouds, Azure stack).
30func NewUsagesClientWithBaseURI(baseURI string, subscriptionID string) UsagesClient {
31	return UsagesClient{NewWithBaseURI(baseURI, subscriptionID)}
32}
33
34// List list network usages for a subscription.
35// Parameters:
36// location - the location where resource usage is queried.
37func (client UsagesClient) List(ctx context.Context, location string) (result UsagesListResultPage, err error) {
38	if tracing.IsEnabled() {
39		ctx = tracing.StartSpan(ctx, fqdn+"/UsagesClient.List")
40		defer func() {
41			sc := -1
42			if result.ulr.Response.Response != nil {
43				sc = result.ulr.Response.Response.StatusCode
44			}
45			tracing.EndSpan(ctx, sc, err)
46		}()
47	}
48	if err := validation.Validate([]validation.Validation{
49		{TargetValue: location,
50			Constraints: []validation.Constraint{{Target: "location", Name: validation.Pattern, Rule: `^[-\w\._ ]+$`, Chain: nil}}}}); err != nil {
51		return result, validation.NewError("network.UsagesClient", "List", err.Error())
52	}
53
54	result.fn = client.listNextResults
55	req, err := client.ListPreparer(ctx, location)
56	if err != nil {
57		err = autorest.NewErrorWithError(err, "network.UsagesClient", "List", nil, "Failure preparing request")
58		return
59	}
60
61	resp, err := client.ListSender(req)
62	if err != nil {
63		result.ulr.Response = autorest.Response{Response: resp}
64		err = autorest.NewErrorWithError(err, "network.UsagesClient", "List", resp, "Failure sending request")
65		return
66	}
67
68	result.ulr, err = client.ListResponder(resp)
69	if err != nil {
70		err = autorest.NewErrorWithError(err, "network.UsagesClient", "List", resp, "Failure responding to request")
71		return
72	}
73	if result.ulr.hasNextLink() && result.ulr.IsEmpty() {
74		err = result.NextWithContext(ctx)
75		return
76	}
77
78	return
79}
80
81// ListPreparer prepares the List request.
82func (client UsagesClient) ListPreparer(ctx context.Context, location string) (*http.Request, error) {
83	pathParameters := map[string]interface{}{
84		"location":       autorest.Encode("path", location),
85		"subscriptionId": autorest.Encode("path", client.SubscriptionID),
86	}
87
88	const APIVersion = "2020-11-01"
89	queryParameters := map[string]interface{}{
90		"api-version": APIVersion,
91	}
92
93	preparer := autorest.CreatePreparer(
94		autorest.AsGet(),
95		autorest.WithBaseURL(client.BaseURI),
96		autorest.WithPathParameters("/subscriptions/{subscriptionId}/providers/Microsoft.Network/locations/{location}/usages", pathParameters),
97		autorest.WithQueryParameters(queryParameters))
98	return preparer.Prepare((&http.Request{}).WithContext(ctx))
99}
100
101// ListSender sends the List request. The method will close the
102// http.Response Body if it receives an error.
103func (client UsagesClient) ListSender(req *http.Request) (*http.Response, error) {
104	return client.Send(req, azure.DoRetryWithRegistration(client.Client))
105}
106
107// ListResponder handles the response to the List request. The method always
108// closes the http.Response Body.
109func (client UsagesClient) ListResponder(resp *http.Response) (result UsagesListResult, err error) {
110	err = autorest.Respond(
111		resp,
112		azure.WithErrorUnlessStatusCode(http.StatusOK),
113		autorest.ByUnmarshallingJSON(&result),
114		autorest.ByClosing())
115	result.Response = autorest.Response{Response: resp}
116	return
117}
118
119// listNextResults retrieves the next set of results, if any.
120func (client UsagesClient) listNextResults(ctx context.Context, lastResults UsagesListResult) (result UsagesListResult, err error) {
121	req, err := lastResults.usagesListResultPreparer(ctx)
122	if err != nil {
123		return result, autorest.NewErrorWithError(err, "network.UsagesClient", "listNextResults", nil, "Failure preparing next results request")
124	}
125	if req == nil {
126		return
127	}
128	resp, err := client.ListSender(req)
129	if err != nil {
130		result.Response = autorest.Response{Response: resp}
131		return result, autorest.NewErrorWithError(err, "network.UsagesClient", "listNextResults", resp, "Failure sending next results request")
132	}
133	result, err = client.ListResponder(resp)
134	if err != nil {
135		err = autorest.NewErrorWithError(err, "network.UsagesClient", "listNextResults", resp, "Failure responding to next results request")
136	}
137	return
138}
139
140// ListComplete enumerates all values, automatically crossing page boundaries as required.
141func (client UsagesClient) ListComplete(ctx context.Context, location string) (result UsagesListResultIterator, err error) {
142	if tracing.IsEnabled() {
143		ctx = tracing.StartSpan(ctx, fqdn+"/UsagesClient.List")
144		defer func() {
145			sc := -1
146			if result.Response().Response.Response != nil {
147				sc = result.page.Response().Response.Response.StatusCode
148			}
149			tracing.EndSpan(ctx, sc, err)
150		}()
151	}
152	result.page, err = client.List(ctx, location)
153	return
154}
155