Contents

module Clusterfuck
  class Cluster
    attr_reader :name, :network

    def initialize(name)
      @name = name
      @graph = {}
    end

    def self.create(name)
      @@clusters ||= {}
      @@clusters[name] = self.new(name)
    end

    def self.[](name)
      @@clusters[name]
    end

    def self.all
      @@clusters.values
    end

    def find_by_name(name)
      nodes.find { |node| node.name == name }
    end
    alias_method :[], :find_by_name

    def nodes
      @graph.keys
    end

    def connect(*nodes, subnet: SubnetFactory.next_subnet)
      # Create a complete graph (a graph where all nodes are connected to each
      # other) between all the nodes, because that's what a subnet is.
      nodes.permutation(2).each do |(a, b)|
        @graph[a] ||= []
        @graph[a] << b
      end

      nodes.each do |node|
        node.ips << subnet.next_ip
        node.cluster = self
      end

      subnet.last_ip
    end

    def ip_in_same_subnet(a, b)
      a.ips.find { |a_ip|
        b.ips.find { |b_ip|
          a_ip.include?(b_ip)
        }
      }
    end
    alias_method :neighbor?, :ip_in_same_subnet
  end

  class BGPCluster < Cluster
    def bgp_neighbors(node)
      @graph[node].select { |neighbor|
        neighbor.is_a?(BGPPeer)
      }
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
vagrant-clusterfuck-0.0.7	lib/clusterfuck/cluster.rb
vagrant-clusterfuck-0.0.6	lib/clusterfuck/cluster.rb
vagrant-clusterfuck-0.0.5	lib/clusterfuck/cluster.rb
vagrant-clusterfuck-0.0.4	lib/clusterfuck/cluster.rb
vagrant-clusterfuck-0.0.3	lib/clusterfuck/cluster.rb
vagrant-clusterfuck-0.0.2	lib/clusterfuck/cluster.rb