1
0
Fork 0
mirror of https://github.com/fog/fog.git synced 2022-11-09 13:51:43 -05:00

Align the disk interface more closely with the API.

The caller need only specify either image_name OR snapshot + size
when creating a disk in a given zone.

This patch also removes the @default_zone variable being referenced
in several places (because it is not actually defined anywhere).
This commit is contained in:
Akshay Moghe 2013-08-11 16:17:41 -07:00
parent 0dc399f3a8
commit 7ad7183dc0
5 changed files with 36 additions and 13 deletions

View file

@ -16,17 +16,28 @@ module Fog
attribute :description, :aliases => 'description'
attribute :size_gb, :aliases => 'sizeGb'
attribute :self_link, :aliases => 'selfLink'
attribute :image_name, :aliases => 'image'
attribute :source_image, :aliases => 'sourceImage'
attribute :source_snapshot, :aliases => 'sourceSnapshot'
attribute :source_snapshot_id, :aliases => 'sourceSnapshot'
def save
data = service.insert_disk(name, size_gb, zone_name, image_name).body
requires :name
requires :zone_name
options = {}
if source_image.nil?
options['sourceSnapshot'] = source_snapshot
options['sizeGb'] = size_gb
end
data = service.insert_disk(name, zone_name, source_image, options).body
data = service.backoff_if_unfound {service.get_disk(name, zone_name).body}
service.disks.merge_attributes(data)
end
def destroy
requires :name, :zone
service.delete_disk(name, zone)
requires :name, :zone_name
service.delete_disk(name, zone_name)
end
def zone
@ -77,4 +88,4 @@ module Fog
end
end
end
end
end

View file

@ -12,7 +12,7 @@ module Fog
class Real
def delete_disk(disk_name, zone_name=@default_zone)
def delete_disk(disk_name, zone_name)
api_method = @compute.disks.delete
parameters = {
'project' => @project,

View file

@ -12,7 +12,7 @@ module Fog
class Real
def get_disk(disk_name, zone_name=@default_zone)
def get_disk(disk_name, zone_name)
if zone_name.start_with? 'http'
zone_name = zone_name.split('/')[-1]
end

View file

@ -12,7 +12,7 @@ module Fog
class Real
def insert_disk(disk_name, disk_size, zone_name=@default_zone, image_name=nil)
def insert_disk(disk_name, zone_name, image_name=nil, opts={})
api_method = @compute.disks.insert
parameters = {
'project' => @project,
@ -26,10 +26,22 @@ module Fog
parameters['sourceImage'] = @image_url
end
body_object = {
'name' => disk_name,
'sizeGb' => disk_size,
}
body_object = { 'name' => disk_name }
# These must be present if image_name is not specified
if image_name.nil?
unless opts.has_key?(:sourceSnapshot) and opts.has_key?(:sizeGb)
raise ArgumentError.new('Must specify image OR snapshot and '\
'disk size when creating a disk.')
end
body_object['sizeGb'] = opts['sizeGb'].delete
# TODO 'get' the sourceSnapshot to validate that it exists?
body_object['sourceSnapshot'] = opts['sourceSnapshot'].delete
end
# Merge in any remaining options (only 'description' should remain)
body_object.merge(opts)
result = self.build_result(api_method, parameters,
body_object)

View file

@ -12,7 +12,7 @@ module Fog
class Real
def list_disks(zone_name=@default_zone)
def list_disks(zone_name)
api_method = @compute.disks.list
parameters = {
'project' => @project,