70971d37ee
We were leaking EIPs because we used to pull the eip using the nic id which results in the eip leak since the nic is already deleted at the time of eip listing via API. Now, we simply store that information in our database and release using that information without the need of extra API call.
89 lines
2.4 KiB
Ruby
89 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "aws-sdk-ec2"
|
|
class Prog::Aws::Nic < Prog::Base
|
|
subject_is :nic
|
|
|
|
label def create_network_interface
|
|
network_interface_response = client.create_network_interface({
|
|
subnet_id: nic.private_subnet.private_subnet_aws_resource.subnet_id,
|
|
private_ip_address: nic.private_ipv4.network.to_s,
|
|
ipv_6_prefix_count: 1,
|
|
groups: [
|
|
nic.private_subnet.private_subnet_aws_resource.security_group_id
|
|
],
|
|
tag_specifications: tag_specifications("network-interface")
|
|
})
|
|
network_interface_id = network_interface_response.network_interface.network_interface_id
|
|
|
|
client.assign_ipv_6_addresses({
|
|
network_interface_id:,
|
|
ipv_6_address_count: 1
|
|
})
|
|
|
|
nic.update(name: network_interface_id)
|
|
hop_wait_network_interface_created
|
|
end
|
|
|
|
label def wait_network_interface_created
|
|
network_interface_response = client.describe_network_interfaces({filters: [{name: "network-interface-id", values: [nic.name]}, {name: "tag:Ubicloud", values: ["true"]}]}).network_interfaces[0]
|
|
if network_interface_response.status == "available"
|
|
eip_response = client.allocate_address
|
|
nic.nic_aws_resource.update(eip_allocation_id: eip_response.allocation_id)
|
|
|
|
hop_attach_eip_network_interface
|
|
end
|
|
|
|
nap 1
|
|
end
|
|
|
|
label def attach_eip_network_interface
|
|
# Associate the Elastic IP with your network interface
|
|
client.associate_address({
|
|
allocation_id: nic.nic_aws_resource.eip_allocation_id,
|
|
network_interface_id: nic.name
|
|
})
|
|
pop "nic created"
|
|
end
|
|
|
|
label def destroy
|
|
ignore_invalid_nic do
|
|
client.delete_network_interface({network_interface_id: nic.name})
|
|
end
|
|
hop_release_eip
|
|
end
|
|
|
|
label def release_eip
|
|
begin
|
|
client.release_address({allocation_id: nic.nic_aws_resource.eip_allocation_id})
|
|
rescue Aws::EC2::Errors::InvalidAllocationIDNotFound
|
|
# Ignore if the address is not found
|
|
# This can happen if the address was already released or never existed
|
|
# In this case, we just pop the stack
|
|
end
|
|
pop "nic destroyed"
|
|
end
|
|
|
|
def client
|
|
@client ||= nic.private_subnet.location.location_credential.client
|
|
end
|
|
|
|
def tag_specifications(resource_type)
|
|
[
|
|
{
|
|
resource_type: resource_type,
|
|
tags: [
|
|
{key: "Ubicloud", value: "true"}
|
|
]
|
|
}
|
|
]
|
|
end
|
|
|
|
private
|
|
|
|
def ignore_invalid_nic
|
|
yield
|
|
rescue Aws::EC2::Errors::InvalidNetworkInterfaceIDNotFound
|
|
end
|
|
end
|