1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
# encoding: utf-8
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
module MsRestAzure
#
# Class which represents a state of Azure long running operation.
#
class PollingState
# @return [Net::HTTPRequest] the HTTP request.
attr_accessor :request
# @return the resource
attr_accessor :resource
# @return [Net::HTTPResponse] the HTTP response.
attr_accessor :response
# @return [AzureOperationError] the azure error data.
attr_accessor :error_data
# @return [String] the latest value captured from Azure-AsyncOperation header.
attr_accessor :azure_async_operation_header_link
# @return [String] the latest value captured from Location header.
attr_accessor :location_header_link
# @return [String] status of the long running operation.
attr_accessor :status
def initialize(azure_response, retry_timeout)
@retry_timeout = retry_timeout
@request = azure_response.request
update_response(azure_response.response)
@resource = azure_response.body
case @response.status
when 200
provisioning_state = get_provisioning_state
@status = provisioning_state.nil?? (AsyncOperationStatus::SUCCESS_STATUS):provisioning_state
when 201
provisioning_state = get_provisioning_state
@status = provisioning_state.nil?? (AsyncOperationStatus::IN_PROGRESS_STATUS):provisioning_state
when 202
@status = AsyncOperationStatus::IN_PROGRESS_STATUS
when 204
@status = AsyncOperationStatus::SUCCESS_STATUS
else
@status = AsyncOperationStatus::FAILED_STATUS
end
end
#
# Returns the provisioning status of the resource
#
# @return [String] provisioning status of the resource
def get_provisioning_state
# On non flattened resource, we should find provisioning_state inside 'properties'
if (!@resource.nil? && @resource.respond_to?(:properties) && @resource.properties.respond_to?(:provisioning_state) && !@resource.properties.provisioning_state.nil?)
@resource.properties.provisioning_state
# On flattened resource, we should find provisioning_state at the top level
elsif !@resource.nil? && @resource.respond_to?(:provisioning_state) && !@resource.provisioning_state.nil?
@resource.provisioning_state
else
nil
end
end
#
# Returns the amount of time in seconds for long running operation polling delay.
#
# @return [Integer] Amount of time in seconds for long running operation polling delay.
def get_delay
return @retry_timeout unless @retry_timeout.nil?
if !response.nil? && !response.headers['Retry-After'].nil?
return response.headers['Retry-After'].to_i
end
return AsyncOperationStatus::DEFAULT_DELAY
end
#
# Updates the polling state from the fields of given response object.
# @param response [Net::HTTPResponse] the HTTP response.
def update_response(response)
@response = response
unless response.nil?
@azure_async_operation_header_link = response.headers['Azure-AsyncOperation'] unless response.headers['Azure-AsyncOperation'].nil?
@location_header_link = response.headers['Location'] unless response.headers['Location'].nil?
end
end
#
# returns the Azure's response.
#
# @return [MsRestAzure::AzureOperationResponse] Azure's response.
def get_operation_response
azure_response = AzureOperationResponse.new(@request, @response, @resource)
azure_response
end
#
# Composes and returns cloud error.
#
# @return [AzureOperationError] the cloud error.
def get_operation_error
AzureOperationError.new @request, @response, @error_data, "Long running operation failed with status #{@status}"
end
def get_request(options = {})
link = @azure_async_operation_header_link || @location_header_link
options[:connection] = create_connection(options[:base_uri])
MsRest::HttpOperationRequest.new(nil, link, :get, options)
end
private
# @return [Integer] retry timeout.
attr_accessor :retry_timeout
attr_accessor :connection
def create_connection(base_url)
@connection ||= Faraday.new(:url => base_url, :ssl => MsRest.ssl_options) do |faraday|
[[MsRest::RetryPolicyMiddleware, times: 3, retry: 0.02], [:cookie_jar]].each{ |args| faraday.use(*args) }
faraday.adapter Faraday.default_adapter
faraday.headers = request.headers
logging = ENV['AZURE_HTTP_LOGGING'] || request.log
if logging
faraday.response :logger, nil, { :bodies => logging == 'full' }
end
end
end
end
end
|
