diff options
author | Seth Chisamore <schisamo@opscode.com> | 2012-10-30 10:39:35 -0400 |
---|---|---|
committer | Seth Chisamore <schisamo@opscode.com> | 2012-10-30 10:39:35 -0400 |
commit | 24dc69a9a97e82a6e4207de68d6dcc664178249b (patch) | |
tree | 19bb289c9f88b4bbab066bc56b95d6d222fd5c35 /lib/chef/resource_collection | |
parent | 9348c1c9c80ee757354d624b7dc1b78ebc7605c4 (diff) | |
download | chef-24dc69a9a97e82a6e4207de68d6dcc664178249b.tar.gz |
[OC-3564] move core Chef to the repo root \o/ \m/
The opscode/chef repository now only contains the core Chef library code
used by chef-client, knife and chef-solo!
Diffstat (limited to 'lib/chef/resource_collection')
-rw-r--r-- | lib/chef/resource_collection/stepable_iterator.rb | 124 |
1 files changed, 124 insertions, 0 deletions
diff --git a/lib/chef/resource_collection/stepable_iterator.rb b/lib/chef/resource_collection/stepable_iterator.rb new file mode 100644 index 0000000000..ec1e244758 --- /dev/null +++ b/lib/chef/resource_collection/stepable_iterator.rb @@ -0,0 +1,124 @@ +# Author:: Daniel DeLeo (<dan@kallistec.com>) +# Copyright:: Copyright (c) 2009 Daniel DeLeo +# License:: Apache License, Version 2.0 +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# + +class Chef + class ResourceCollection + class StepableIterator + + def self.for_collection(new_collection) + instance = new(new_collection) + instance + end + + attr_accessor :collection + attr_reader :position + + def initialize(collection=[]) + @position = 0 + @paused = false + @collection = collection + end + + def size + collection.size + end + + def each(&block) + reset_iteration(block) + @iterator_type = :element + iterate + end + + def each_index(&block) + reset_iteration(block) + @iterator_type = :index + iterate + end + + def each_with_index(&block) + reset_iteration(block) + @iterator_type = :element_with_index + iterate + end + + def paused? + @paused + end + + def pause + @paused = true + end + + def resume + @paused = false + iterate + end + + def rewind + @position = 0 + end + + def skip_back(skips=1) + @position -= skips + end + + def skip_forward(skips=1) + @position += skips + end + + def step + return nil if @position == size + call_iterator_block + @position += 1 + end + + def iterate_on(iteration_type, &block) + @iterator_type = iteration_type + @iterator_block = block + end + + private + + def reset_iteration(iterator_block) + @iterator_block = iterator_block + @position = 0 + @paused = false + end + + def iterate + while @position < size && !paused? + step + end + collection + end + + def call_iterator_block + case @iterator_type + when :element + @iterator_block.call(collection[@position]) + when :index + @iterator_block.call(@position) + when :element_with_index + @iterator_block.call(collection[@position], @position) + else + raise "42error: someone forgot to set @iterator_type, wtf?" + end + end + + end + end +end |