cloudflock 0.6.1 → 0.7.0

data/lib/cloudflock/target/servers/platform.rb

@@ -1,133 +1,110 @@
 require 'cpe'
 require 'cloudflock'
+require 'fog'
 
-# Public: Serves as a small class to easily map host specifications to Image
-# and Flavor IDs in Rackspace Cloud.
-#
-# Examples
-#
-#   # Build platform data for a given CPE object
-#   platform = Platform.new(cpe)
-#   platform.image_id
-#   # => Fixnum
-class CloudFlock::Target::Servers::Platform
-  # Public: Gets/sets whether the target platform will be managed.
-  attr_accessor :managed
-  # Public: Gets/sets whether the target platform will use Rack Connect.
-  attr_accessor :rack_connect
-
-  # Public: Initialize the Platform object.
+module CloudFlock; module Target; module Servers
+  # Public: Map host specifications to Image and Flavor IDs in Rackspace Cloud.
   #
-  # cpe - CPE object from which to generate platform object.
+  # Examples
   #
-  # Raises ArgumentError if anything but a CPE object was given.
-  # Raises KeyError if the CPE object doesn't have a vendor or version defined.
-  def initialize(cpe)
-    raise ArgumentError unless cpe.kind_of?(CPE)
-    raise KeyError if cpe.vendor.nil? or cpe.version.nil?
-
-    @cpe = cpe
-    @distro = cpe.vendor
-    @product = cpe.product
-    @version = cpe.version
-    @managed = false
-    @rack_connect = false
-
-    build_maps
-  end
-
-  # Public: Generate a String of the platform's name/version suitable for
-  # display
+  #   platform = Platform.new(cpe, fog)
+  #   platform.get_appropriate_flavors
+  #   # => [<Fog::Compute::RackspaceV2::Flavor>, ...]
   #
-  # Returns a String describing the Platform
-  def to_s
-    "#{@distro.capitalize} #{@product.gsub(/_/, ' ').capitalize} #{@version}"
-  end
-
-  # Public: Return the Image ID to be used based on whether the account is
-  # managed, and the platform used
-  #
-  # Returns the Image ID corresponding to the Platform object as a String
-  def image
-    [:MANAGED_MAP, :UNMANAGED_MAP].each do |map|
-      unless self.class.const_defined?(map)
-        raise MapUndefined, "Const #{map} is undefined; maps appear unbuilt"
+  #   platform.to_s
+  #   # => "CentOS Linux 6.4"
+  class Platform
+    # Public: Initialize the Platform object.
+    #
+    # cpe - CPE object to use in determining image ID.
+    # fog - Fog Compute object used to obtain flavor and image maps.
+    #       (default: nil)
+    #
+    # Raises ArgumentError if anything but a CPE object is passed.
+    # Raises KeyError if the CPE object doesn't have vendor and version
+    # defined.
+    def initialize(cpe, fog = nil)
+      unless cpe.is_a?(CPE)
+        raise(ArgumentError, Errstr::NOT_CPE)
+      end
+      if cpe.vendor.nil? or cpe.version.nil?
+        raise(KeyError, Errstr::CPE_INCOMPLETE)
       end
-    end
-
-    map = @managed ? self.class::MANAGED_MAP : self.class::UNMANAGED_MAP
-    distro = @distro.downcase.to_sym
 
-    unless map[distro].nil?
-      return map[distro][@version] unless map[distro][@version].nil?
-      return map[distro]["*"] unless map[distro]["*"].nil?
+      @cpe = cpe
+      build_maps(fog)
     end
 
-    nil
-  end
-
-  # Public: Iterate through TARGET_LIST until a suitably large flavor_id is
-  # found, then return the appropriate index.  If no entry can be found, raise
-  # ValueError.
-  #
-  # symbol - A Symbol referencing the search target in TARGET_LIST.
-  # value  - A Fixnum containing the amount of memory or disk space required.
-  #
-  # Returns a Fixnum referencing the TARGET_LIST index.
-  # Raises ValueError if no symbol describes an appropriate target.
-  def get_target_by_symbol(symbol, value)
-    unless self.class.const_defined?(:FLAVOR_LIST)
-      raise MapUndefined, "FLAVOR_LIST is undefined; maps appear unbuild."
+    # Public: Generate a String describing the platform's name/version.
+    #
+    # Returns a String.
+    def to_s
+      "#{@cpe.distro.capitalize} #{@cpe.product.gsub(/_/, ' ').capitalize} " \
+      "#{@cpe.version}"
     end
 
-    self.class::FLAVOR_LIST.each_with_index do |target, idx|
-      if target[symbol] > value
-        return idx
+    # Public: Given a resource map (e.g. generated by profiling a host), 
+    # determine which flavors would be appropriate as migration targets.
+    #
+    # resource_map - Hash containing memory and hard disk usage information
+    #                from the source host:
+    #                :ram - Hash containing memory usage information
+    #                       :mem_used  - Amount of memory used in MiB.
+    #                                    (default: 0)
+    #                       :swapping? - Whether the source host is swapping.
+    #                                    (default: 0)
+    #                :hdd - Hash containing disk usage information:
+    #                       :disk - Amount of disk used in GB. (default: 0)
+    #
+    # Returns a Hash containing appropriate flavor IDs mapped to flavor names.
+    def get_appropriate_flavors(resource_map = {})
+      resource_map[:ram] ||= {mem_used: 0, swapping?: false}
+      resource_map[:hdd] ||= {disk: 0}
+      ram  = resource_map[:ram] * (resource_map[:ram][:swapping?] ? 2 : 1)
+      disk = resource_map[:hdd][:disk] * 1.2
+
+      @flavor_map.select do |id|
+        @flavor_map[e][:ram] >= ram  && @flavor_map[e][:disk] >= disk
       end
     end
 
-    raise ValueError, "Unable to find a flavor matching #{symbol} #{value}"
-  end
-
-  # Internal: Build image and flavor maps
-  #
-  # Returns nothing
-  def build_maps
-    build_image_maps
-    build_flavor_maps
-  end
-
-  # Public: Give a recommendation for a Flavor ID and an Image ID which should
-  # suffice for a migration target host.
-  #
-  # args - Hash containing information on which to base the recommendation:
-  #        :memory  - Hash containing memory information:
-  #                   :total     - Total amount of memory allocated to the host
-  #                                profiled.
-  #                   :mem_used  - Amount of RAM in use at the time of
-  #                                profiling.
-  #                   :swapping? - Boolean denoting whether the host was
-  #                                swapping at the time of profiling.
-  #        :disk    - Fixnum containing the amount of disk which appears to be
-  #                   in use at the time of profiling.
-  #
-  # Returns a Hash containing the Flavor ID and a String containing the
-  # reasoning for the decision.
-  def build_recommendation(args)
-    recommendation = {}
-    target_mem = get_target_by_symbol(:mem, args[:memory][:mem_used])
-    target_mem += 1 if args[:memory][:swapping?]
-
-    target_disk = get_target_by_symbol(:hdd, args[:disk])
-
-    if target_mem >= target_disk
-      recommendation[:flavor] = target_mem
-      recommendation[:flavor_reason] = "RAM usage"
-    else
-      recommendation[:flavor] = target_disk
-      recommendation[:flavor_reason] = "Disk usage"
+    private
+
+    # Internal: Assemble a map of flavors and images available for
+    # provisioning.
+    #
+    # fog - Fog Compute object used to obtain image/flavor mappings.
+    #
+    # Sets @image_map and @flavor_map.
+    #
+    # Returns nothing.
+    def build_maps(fog)
+      if fog.respond_to?(:images) && fog.respond_to?(:flavors)
+        @image_map = fog.images.reduce({}) do |c,e|
+          name, ver = map_image_id(e)
+          c[name] ||= []
+          c[name] << ver
+          c
+        end
+        @flavor_map = fog.flavors.reduce({}) do |c,e|
+          c.merge({e.id => { ram: e.ram, disk: e.disk }})
+        end
+      else
+        @image_map  = {}
+        @flavor_map = {}
+      end
     end
 
-    recommendation
+    # Internal: Convert a Fog Image Object to a tuple for easy selection later.
+    #
+    # image - Fog Image object.
+    #
+    # Returns an Array containing a String and a Hash.
+    def map_image_id(image)
+      name, _ = image.name.split(/ \d/, 2)
+      other = image.name.gsub(Regexp.new("^#{name} ?"), '')
+      version, extra = other.split(/ +/, 2)
+      [name, { version => { :id => image.id, :extra => extra } } ]
+    end
   end
-end
+end; end; end