Add support for pool tags.

Author: ioquatix

async-pool.gemspec

@@ -26,6 +26,6 @@ Gem::Specification.new do |spec|
 	spec.required_ruby_version = ">= 3.1"
 
 	spec.add_dependency "async", ">= 1.25"
-	spec.add_dependency "metrics"
+	spec.add_dependency "metrics", "~> 0.11"
 	spec.add_dependency "traces"
 end

lib/async/pool/controller.rb

@@ -29,7 +29,7 @@ def self.wrap(**options, &block)
 			# @parameter limit [Integer | Nil] The maximum number of resources that this pool can have at any given time. If nil, the pool can have an unlimited number of resources.
 			# @parameter concurrency [Integer] The maximum number of concurrent tasks that can be creating a new resource.
 			# @parameter policy [Policy] The pool policy.
-			def initialize(constructor, limit: nil, concurrency: (limit || 1), policy: nil)
+			def initialize(constructor, limit: nil, concurrency: (limit || 1), policy: nil, tags: nil)
 				@constructor = constructor
 				@limit = limit
 
@@ -39,6 +39,8 @@ def initialize(constructor, limit: nil, concurrency: (limit || 1), policy: nil)
 				@policy = policy
 				@gardener = nil
 
+				@tags = Metrics::Tags.normalize(tags)
+				
 				# All available resources:
 				@resources = {}
 
@@ -96,6 +98,9 @@ def concurrency= value
 			# @attribute [Hash(Resource, Integer)] all allocated resources, and their associated usage.
 			attr :resources
 
+			# @attribute [Array(String)] The name of the pool.
+			attr_accessor :tags
+			
 			# The number of resources in the pool.
 			def size
 				@resources.size
@@ -388,6 +393,7 @@ def create_resource(...)
 						concurrency: @guard.limit,
 						size: @resources.size,
 						limit: @limit,
+						tags: @tags,
 					}
 
 					Traces.trace('async.pool.create', attributes: attributes) {super}
@@ -396,6 +402,7 @@ def create_resource(...)
 				def drain(...)
 					attributes = {
 						size: @resources.size,
+						tags: @tags,
 					}
 
 					Traces.trace('async.pool.drain', attributes: attributes) {super}
@@ -408,20 +415,20 @@ def drain(...)
 				RETIRE_COUNT = Metrics.metric('async.pool.retire', :counter, description: 'Number of times a resource was retired.')
 
 				def acquire(...)
-					ACQUIRE_COUNT.emit(1)
+					ACQUIRE_COUNT.emit(1, tags: @tags)
 
 					super
 				end
 
 				def release(...)
 					super.tap do
-						RELEASE_COUNT.emit(1)
+						RELEASE_COUNT.emit(1, tags: @tags)
 					end
 				end
 
 				def retire(...)
 					super.tap do
-						RETIRE_COUNT.emit(1)
+						RETIRE_COUNT.emit(1, tags: @tags)
 					end
 				end
 			end

test/async/pool/controller.rb

@@ -46,6 +46,14 @@
 		end
 	end
 
+	with "tags" do
+		let(:pool) {subject.new(Async::Pool::Resource, tags: {a: 1, b: 2})}
+		
+		it "can assign tags to the pool" do
+			expect(pool.tags).to be == ["a:1", "b:2"]
+		end
+	end
+	
 	with 'a limited pool' do
 		let(:pool) {subject.new(Async::Pool::Resource, limit: 1)}