lib/consul/power.rb in consul-0.11.1 vs lib/consul/power.rb in consul-0.11.2

@@ -72,18 +72,22 @@
         names.each do |name|
           define_power(name, &block)
         end
       end
 
-      def context_count_name(name)
-        "#{name}_context_count"
-      end
-
       def power_ids_name(name)
         "#{name.to_s.singularize}_ids"
       end
 
-      attr_accessor :current
+      THREAD_KEY = :'Power.current'
+
+      def current
+        Thread.current[THREAD_KEY]
+      end
+
+      def current=(power)
+        Thread.current[THREAD_KEY] = power
+      end
 
       def with_power(inner_power, &block)
         unless inner_power.is_a?(self) || inner_power.nil?
           inner_power = new(inner_power)
         end