Sha256: 175614b1259b4cbee1da71ac2b7cd9c53d3bc720fb48f564037ba8c3073ad031

Contents?: true

Size: 1.45 KB

Versions: 16

Compression:

Stored size: 1.45 KB

Contents

# encoding: utf-8
# Copyright 2014 Aerospike, Inc.
#
# Portions may be licensed to Aerospike, Inc. under one or more contributor
# license agreements.
#
# 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.

require 'thread'
require 'timeout'

module Aerospike

  private

  class Pool #:nodoc:

    attr_accessor :create_block, :cleanup_block

    def initialize(max_size = 256, &block)
      @create_block = block
      @cleanup_block = nil

      @pool = Queue.new
      @max_size = max_size
    end

    def offer(obj)
      if @pool.length < @max_size
        @pool << obj
      elsif @cleanup_block
        @cleanup_block.call(obj)
      end
    end
    alias_method :<<, :offer

    def poll(create_new=true)
      res = nil
      begin
        res = @pool.pop(true) # non_blocking
        return res
      rescue
        return @create_block.call if @create_block && create_new
      end
    end

    def empty?
      @pool.length == 0
    end

    def length
      @pool.length
    end

  end

end

Version data entries

16 entries across 16 versions & 1 rubygems

Version Path
aerospike-2.1.1 lib/aerospike/utils/pool.rb
aerospike-2.1.0 lib/aerospike/utils/pool.rb
aerospike-2.0.0 lib/aerospike/utils/pool.rb
aerospike-1.0.12 lib/aerospike/utils/pool.rb
aerospike-1.0.11 lib/aerospike/utils/pool.rb
aerospike-1.0.10 lib/aerospike/utils/pool.rb
aerospike-1.0.9 lib/aerospike/utils/pool.rb
aerospike-1.0.8 lib/aerospike/utils/pool.rb
aerospike-1.0.7 lib/aerospike/utils/pool.rb
aerospike-1.0.6 lib/aerospike/utils/pool.rb
aerospike-1.0.5 lib/aerospike/utils/pool.rb
aerospike-1.0.4 lib/aerospike/utils/pool.rb
aerospike-1.0.3 lib/aerospike/utils/pool.rb
aerospike-1.0.2 lib/aerospike/utils/pool.rb
aerospike-1.0.1 lib/aerospike/utils/pool.rb
aerospike-1.0.0 lib/aerospike/utils/pool.rb