Class: Miasma::Models::LoadBalancer::Aws

Inherits:
Miasma::Models::LoadBalancer show all
Includes:
Contrib::AwsApiCore::ApiCommon, Contrib::AwsApiCore::RequestUtils
Defined in:
lib/miasma/contrib/aws/load_balancer.rb

Overview

AWS load balancer API

Constant Summary collapse

API_SERVICE =

Service name of API

"elasticloadbalancing".freeze
API_VERSION =

Supported version of the ELB API

"2012-06-01".freeze

Instance Method Summary collapse

Methods included from Contrib::AwsApiCore::RequestUtils

#all_result_pages

Methods included from Contrib::AwsApiCore::ApiCommon

#after_setup, #api_for, #connect, #connection, #custom_setup, #endpoint, #extract_creds, #get_credential, #get_region, included, #load_aws_file, #load_ecs_credentials!, #load_instance_credentials!, #make_request, #perform_request_retry, #retryable_allowed?, #signer, #sts_assume_role!, #sts_assume_role_update_required?, #sts_attribute_update_required?, #sts_mfa_session!, #sts_mfa_session_update_required?, #update_request, #uri_escape

Instance Method Details

#balancer_all(options = {}) ⇒ Array<Models::LoadBalancer::Balancer>

Return all load balancers

Parameters:

  • options (Hash) (defaults to: {})

    filter

Returns:

  • (Array<Models::LoadBalancer::Balancer>)


205
206
207
# File 'lib/miasma/contrib/aws/load_balancer.rb', line 205

def balancer_all(options = {})
  load_balancer_data
end

#balancer_destroy(balancer) ⇒ TrueClass, FalseClass

Delete load balancer

Parameters:

  • balancer (Models::LoadBalancer::Balancer)

Returns:

  • (TrueClass, FalseClass)


182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
# File 'lib/miasma/contrib/aws/load_balancer.rb', line 182

def balancer_destroy(balancer)
  if balancer.persisted?
    request(
      :method => :post,
      :path => "/",
      :form => Smash.new(
        "Action" => "DeleteLoadBalancer",
        "LoadBalancerName" => balancer.name,
      ),
    )
    balancer.state = :pending
    balancer.status = "DELETE_IN_PROGRESS"
    balancer.valid_state
    true
  else
    false
  end
end

#balancer_health_check(balancer) ⇒ Models::LoadBalancer::Balancer

Save the load balancer health check

Parameters:

  • balancer (Models::LoadBalancer::Balancer)

Returns:

  • (Models::LoadBalancer::Balancer)


78
79
80
# File 'lib/miasma/contrib/aws/load_balancer.rb', line 78

def balancer_health_check(balancer)
  balancer
end

#balancer_reload(balancer) ⇒ Models::LoadBalancer::Balancer

Reload the balancer data from the API

Parameters:

  • balancer (Models::LoadBalancer::Balancer)

Returns:

  • (Models::LoadBalancer::Balancer)


94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
# File 'lib/miasma/contrib/aws/load_balancer.rb', line 94

def balancer_reload(balancer)
  if balancer.persisted?
    begin
      load_balancer_data(balancer)
    rescue Miasma::Error::ApiError::RequestError => e
      if e.response_error_msg.include?("LoadBalancerNotFound")
        balancer.state = :terminated
        balancer.status = "terminated"
        balancer.valid_state
      else
        raise
      end
    end
  end
  balancer
end

#balancer_save(balancer) ⇒ Models::LoadBalancer::Balancer

Save load balancer

Parameters:

  • balancer (Models::LoadBalancer::Balancer)

Returns:

  • (Models::LoadBalancer::Balancer)


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
# File 'lib/miasma/contrib/aws/load_balancer.rb', line 20

def balancer_save(balancer)
  unless balancer.persisted?
    params = Smash.new(
      "LoadBalancerName" => balancer.name,
    )
    availability_zones.each_with_index do |az, i|
      params["AvailabilityZones.member.#{i + 1}"] = az
    end
    if balancer.listeners
      balancer.listeners.each_with_index do |listener, i|
        key = "Listeners.member.#{i + 1}"
        params["#{key}.Protocol"] = listener.protocol
        params["#{key}.InstanceProtocol"] = listener.instance_protocol
        params["#{key}.LoadBalancerPort"] = listener.load_balancer_port
        params["#{key}.InstancePort"] = listener.instance_port
        if listener.ssl_certificate_id
          params["#{key}.SSLCertificateId"] = listener.ssl_certificate_id
        end
      end
    end
    result = request(
      :method => :post,
      :path => "/",
      :form => params.merge(
        Smash.new(
          "Action" => "CreateLoadBalancer",
        )
      ),
    )
    balancer.public_addresses = [
      :address => result.get(:body,
                             "CreateLoadBalancerResponse", "CreateLoadBalancerResult", "DNSName"),
    ]
    balancer.load_data(:id => balancer.name).valid_state
    if balancer.health_check
      balancer_health_check(balancer)
    end
    if balancer.servers && !balancer.servers.empty?
      balancer_set_instances(balancer)
    end
  else
    if balancer.dirty?
      if balancer.dirty?(:health_check)
        balancer_health_check(balancer)
      end
      if balancer.dirty?(:servers)
        balancer_set_instances(balancer)
      end
      balancer.reload
    end
    balancer
  end
end

#balancer_set_instances(balancer) ⇒ Models::LoadBalancer::Balancer

Save the load balancer attached servers

Parameters:

  • balancer (Models::LoadBalancer::Balancer)

Returns:

  • (Models::LoadBalancer::Balancer)


86
87
88
# File 'lib/miasma/contrib/aws/load_balancer.rb', line 86

def balancer_set_instances(balancer)
  balancer
end

#load_balancer_data(balancer = nil) ⇒ Array<Models::LoadBalancer::Balancer>

Fetch balancers or update provided balancer data

Parameters:

  • balancer (Models::LoadBalancer::Balancer) (defaults to: nil)

Returns:

  • (Array<Models::LoadBalancer::Balancer>)


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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
# File 'lib/miasma/contrib/aws/load_balancer.rb', line 115

def load_balancer_data(balancer = nil)
  params = Smash.new("Action" => "DescribeLoadBalancers")
  if balancer
    params["LoadBalancerNames.member.1"] = balancer.id || balancer.name
  end
  result = all_result_pages(nil, :body,
                            "DescribeLoadBalancersResponse", "DescribeLoadBalancersResult",
                            "LoadBalancerDescriptions", "member") do |options|
    request(
      :method => :post,
      :path => "/",
      :form => options.merge(params),
    )
  end
  if balancer
    health_result = all_result_pages(nil, :body,
                                     "DescribeInstanceHealthResponse", "DescribeInstanceHealthResult",
                                     "InstanceStates", "member") do |options|
      request(
        :method => :post,
        :path => "/",
        :form => options.merge(
          "LoadBalancerName" => balancer.id || balancer.name,
          "Action" => "DescribeInstanceHealth",
        ),
      )
    end
  end
  result.map do |blr|
    (balancer || Balancer.new(self)).load_data(
      Smash.new(
        :id => blr["LoadBalancerName"],
        :name => blr["LoadBalancerName"],
        :state => :active,
        :status => "ACTIVE",
        :created => blr["CreatedTime"],
        :updated => blr["CreatedTime"],
        :public_addresses => [
          Balancer::Address.new(
            :address => blr["DNSName"],
            :version => 4,
          ),
        ],
        :servers => [blr.get("Instances", "member")].flatten(1).compact.map { |i|
          Balancer::Server.new(self.api_for(:compute), :id => i["InstanceId"])
        },
      ).merge(
        health_result.nil? ? {} : Smash.new(
          :server_states => health_result.nil? ? nil : health_result.map { |i|
            Balancer::ServerState.new(
              self.api_for(:compute),
              :id => i["InstanceId"],
              :status => i["State"],
              :reason => i["ReasonCode"],
              :state => i["State"] == "InService" ? :up : :down,
            )
          },
        )
      )
    ).valid_state
  end
end