1package machinelearningservices
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/tracing"
14	"net/http"
15)
16
17// WorkspaceClient is the these APIs allow end users to operate on Azure Machine Learning Workspace resources.
18type WorkspaceClient struct {
19	BaseClient
20}
21
22// NewWorkspaceClient creates an instance of the WorkspaceClient client.
23func NewWorkspaceClient(subscriptionID string) WorkspaceClient {
24	return NewWorkspaceClientWithBaseURI(DefaultBaseURI, subscriptionID)
25}
26
27// NewWorkspaceClientWithBaseURI creates an instance of the WorkspaceClient client using a custom endpoint.  Use this
28// when interacting with an Azure cloud that uses a non-standard base URI (sovereign clouds, Azure stack).
29func NewWorkspaceClientWithBaseURI(baseURI string, subscriptionID string) WorkspaceClient {
30	return WorkspaceClient{NewWithBaseURI(baseURI, subscriptionID)}
31}
32
33// ListSkus lists all skus with associated features
34func (client WorkspaceClient) ListSkus(ctx context.Context) (result SkuListResultPage, err error) {
35	if tracing.IsEnabled() {
36		ctx = tracing.StartSpan(ctx, fqdn+"/WorkspaceClient.ListSkus")
37		defer func() {
38			sc := -1
39			if result.slr.Response.Response != nil {
40				sc = result.slr.Response.Response.StatusCode
41			}
42			tracing.EndSpan(ctx, sc, err)
43		}()
44	}
45	result.fn = client.listSkusNextResults
46	req, err := client.ListSkusPreparer(ctx)
47	if err != nil {
48		err = autorest.NewErrorWithError(err, "machinelearningservices.WorkspaceClient", "ListSkus", nil, "Failure preparing request")
49		return
50	}
51
52	resp, err := client.ListSkusSender(req)
53	if err != nil {
54		result.slr.Response = autorest.Response{Response: resp}
55		err = autorest.NewErrorWithError(err, "machinelearningservices.WorkspaceClient", "ListSkus", resp, "Failure sending request")
56		return
57	}
58
59	result.slr, err = client.ListSkusResponder(resp)
60	if err != nil {
61		err = autorest.NewErrorWithError(err, "machinelearningservices.WorkspaceClient", "ListSkus", resp, "Failure responding to request")
62		return
63	}
64	if result.slr.hasNextLink() && result.slr.IsEmpty() {
65		err = result.NextWithContext(ctx)
66		return
67	}
68
69	return
70}
71
72// ListSkusPreparer prepares the ListSkus request.
73func (client WorkspaceClient) ListSkusPreparer(ctx context.Context) (*http.Request, error) {
74	pathParameters := map[string]interface{}{
75		"subscriptionId": autorest.Encode("path", client.SubscriptionID),
76	}
77
78	const APIVersion = "2021-04-01"
79	queryParameters := map[string]interface{}{
80		"api-version": APIVersion,
81	}
82
83	preparer := autorest.CreatePreparer(
84		autorest.AsGet(),
85		autorest.WithBaseURL(client.BaseURI),
86		autorest.WithPathParameters("/subscriptions/{subscriptionId}/providers/Microsoft.MachineLearningServices/workspaces/skus", pathParameters),
87		autorest.WithQueryParameters(queryParameters))
88	return preparer.Prepare((&http.Request{}).WithContext(ctx))
89}
90
91// ListSkusSender sends the ListSkus request. The method will close the
92// http.Response Body if it receives an error.
93func (client WorkspaceClient) ListSkusSender(req *http.Request) (*http.Response, error) {
94	return client.Send(req, azure.DoRetryWithRegistration(client.Client))
95}
96
97// ListSkusResponder handles the response to the ListSkus request. The method always
98// closes the http.Response Body.
99func (client WorkspaceClient) ListSkusResponder(resp *http.Response) (result SkuListResult, err error) {
100	err = autorest.Respond(
101		resp,
102		azure.WithErrorUnlessStatusCode(http.StatusOK),
103		autorest.ByUnmarshallingJSON(&result),
104		autorest.ByClosing())
105	result.Response = autorest.Response{Response: resp}
106	return
107}
108
109// listSkusNextResults retrieves the next set of results, if any.
110func (client WorkspaceClient) listSkusNextResults(ctx context.Context, lastResults SkuListResult) (result SkuListResult, err error) {
111	req, err := lastResults.skuListResultPreparer(ctx)
112	if err != nil {
113		return result, autorest.NewErrorWithError(err, "machinelearningservices.WorkspaceClient", "listSkusNextResults", nil, "Failure preparing next results request")
114	}
115	if req == nil {
116		return
117	}
118	resp, err := client.ListSkusSender(req)
119	if err != nil {
120		result.Response = autorest.Response{Response: resp}
121		return result, autorest.NewErrorWithError(err, "machinelearningservices.WorkspaceClient", "listSkusNextResults", resp, "Failure sending next results request")
122	}
123	result, err = client.ListSkusResponder(resp)
124	if err != nil {
125		err = autorest.NewErrorWithError(err, "machinelearningservices.WorkspaceClient", "listSkusNextResults", resp, "Failure responding to next results request")
126	}
127	return
128}
129
130// ListSkusComplete enumerates all values, automatically crossing page boundaries as required.
131func (client WorkspaceClient) ListSkusComplete(ctx context.Context) (result SkuListResultIterator, err error) {
132	if tracing.IsEnabled() {
133		ctx = tracing.StartSpan(ctx, fqdn+"/WorkspaceClient.ListSkus")
134		defer func() {
135			sc := -1
136			if result.Response().Response.Response != nil {
137				sc = result.page.Response().Response.Response.StatusCode
138			}
139			tracing.EndSpan(ctx, sc, err)
140		}()
141	}
142	result.page, err = client.ListSkus(ctx)
143	return
144}
145