lib/antfarm/node.rb in antfarm-0.3.0 vs lib/antfarm/node.rb in antfarm-0.4.0

@@ -24,10 +24,12 @@
 # * has many layer 2 interfaces
 # * has many layer 3 interfaces through layer 2 interfaces
 class Node < ActiveRecord::Base
   has_many :layer2_interfaces
   has_many :layer3_interfaces, :through => :layer2_interfaces
+  has_many :services
+  has_one  :operating_system
 
   before_save :clamp_certainty_factor
 
   validates_presence_of :certainty_factor
 
@@ -63,9 +65,13 @@
       return nil
     else
       logger.info("Node: found existing nodes of given device type.")
       return nodes
     end
+  end
+
+  def to_label
+    return self
   end
 
   #######
   private
   #######