Browse Source

Add a machine_tags attribute to MachineTag::Set.

Douglas Thrift 10 years ago
parent
commit
222d52e487
1 changed files with 4 additions and 0 deletions
  1. 4 0
      lib/machine_tag/set.rb

+ 4 - 0
lib/machine_tag/set.rb

@@ -27,7 +27,10 @@ module MachineTag
   # Set of tags which can be machine tags.
   #
   class Set < ::Set
+    attr_reader :machine_tags
+
     def initialize(enum = nil, &block)
+      @machine_tags = ::Set.new
       @tags_by_namespace = {}
       @tags_by_namespace_and_predicate = {}
       super
@@ -38,6 +41,7 @@ module MachineTag
       super(tag)
 
       if tag.machine_tag?
+        @machine_tags << tag
         @tags_by_namespace[tag.namespace] ||= ::Set.new
         @tags_by_namespace[tag.namespace] << tag
         @tags_by_namespace_and_predicate[tag.namespace_and_predicate] ||= ::Set.new