strongman 1.0.0 → 1.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 329a7cf887ef3a484d31e6c1f943d86857d045cf33daa6862e8c4720366d2273
-  data.tar.gz: 1f4f60aebb814692098f07d5f1b20d520ebd7d6ad897dfec1b541f60b6f4efbc
+  metadata.gz: 07d29577fed62c5f82058d27afc01a59454115031a4b421fd932f8dc03689655
+  data.tar.gz: c85cb8d1d63b3889096e856160a31082dd10a8342ae8c9c3a5d89252b8c1cf78
 SHA512:
-  metadata.gz: 1ca56b74b4bec3f25cfcbd51b5aea8d7eb75725d03ee51adb3e43dd00c01925eeb219b4071c204a61fc7ecad08882410bc466fc80a26bc8de90d70445f8993f2
-  data.tar.gz: 2a986b767a76db4e1a5a087a0e05d9d9f8b632de1c6fd0288e622d2ebd25d926aa897d55ac510e45bf918649e9214611a38cbe895641eea270b1d5dcfcc49372
+  metadata.gz: 5760f359bfa016ec0eb414264418fbefbaa49381f85e8dd35fc33df4d31f35bab751d454b341bfb6684ebd943a75100d826c02551e6fb53697c74cced425c4c1
+  data.tar.gz: 79fe0d462a995d9d58d7f152519730921d80c4a816aa5054149665a2b263c7361cee8523fc0db231819fe6f4eca8eb07e4ca9247860df57e5875544f86d92324

data/lib/strongman/version.rb

@@ -1,4 +1,4 @@
 class Strongman
   # @!visibility private
-  VERSION = "1.0.0".freeze
+  VERSION = "1.0.5".freeze
 end

data/lib/wimp.rb

@@ -0,0 +1,180 @@
+require 'concurrent'
+
+class DelayedResult
+  def initialize(&resolver)
+    @resolver = resolver
+  end
+
+  def then(&block)
+    DelayedResult.new do
+      block.(value!)
+    end
+  end
+
+  def self.zip(*results, &block)
+    DelayedResult.new do
+      results = results.map(&:value!)
+      block.(*results)
+    end
+  end
+
+  def value!
+    @value ||= @resolver.().yield_self do |val|
+      if val&.is_a?(DelayedResult)
+        val.value!
+      else
+        val
+      end
+    end
+  end
+
+  def value
+    value!
+  end
+end
+
+class Wimp
+  class NoCache
+    def compute_if_absent(_key)
+      yield
+    end
+  end
+
+  class Batch
+    attr_accessor :name
+    attr_accessor :fulfilled
+
+    def initialize(loader_block, name: nil, max_batch_size: Float::INFINITY)
+      @name = name
+      @queue = Concurrent::Array.new
+      @lock = Concurrent::ReadWriteLock.new
+      @loader_block = loader_block
+      @max_batch_size = max_batch_size
+      @fulfilled = false
+      @results = nil
+    end
+
+    def queue(key)
+      @queue << key
+
+      DelayedResult.new do
+        results = if @fulfilled
+                    @lock.with_read_lock do
+                      @results
+                    end
+                  else
+                    @lock.with_write_lock do
+                      if @fulfilled
+                        @results
+                      else
+                        @fulfilled = true
+                        r = @loader_block.(@queue)
+                        @results = if r.is_a?(DelayedResult)
+                          normalize_results(r.value!)
+                        else
+                          normalize_results(r)
+                        end
+                      end
+                    end
+                  end
+
+        unless results.key?(key)
+          raise StandardError, "Batch loader didn't resolve a key: #{key}. Resolved keys: #{results.keys}"
+        end
+
+        results[key]
+      end
+    end
+
+    def fulfilled?
+      @fulfilled
+    end
+
+    private
+
+    def normalize_results(results)
+      unless results.is_a?(Array) || results.is_a?(Hash)
+        raise TypeError, "Batch loader must return an Array or Hash, but returned: #{results.class.name}"
+      end
+
+      if @queue.size != results.size
+        raise StandardError, "Batch loader must be instantiated with function that returns Array or Hash " \
+          "of the same size as provided to it Array of keys" \
+          "\n\nProvided keys:\n#{@queue}" \
+          "\n\nReturned values:\n#{results}"
+      end
+
+      if results.is_a?(Array)
+        Hash[@queue.zip(results)]
+      elsif results.is_a?(Hash)
+        results
+      end
+    end
+  end
+
+  attr_accessor :cache
+
+  def initialize(**options, &block)
+    unless block_given?
+      raise TypeError, "Dataloader must be constructed with a block which accepts " \
+        "Array and returns either Array or Hash of the same size (or Promise)"
+    end
+
+    @name = options.delete(:name)
+    @cache = if options.has_key?(:cache)
+               options.delete(:cache) || NoCache.new
+             else
+               Concurrent::Map.new
+             end
+    @max_batch_size = options.fetch(:max_batch_size, Float::INFINITY)
+
+    @interceptor = options.delete(:interceptor) || -> (n) {
+      -> (ids) {
+        n.call(ids)
+      }
+    }
+
+    @loader_block = @interceptor.call(block)
+  end
+
+  def load(key)
+    if key.nil?
+      raise TypeError, "#load must be called with a key, but got: nil"
+    end
+
+    result = retrieve_from_cache(key) do
+      batch.queue(key)
+    end
+
+    if result.is_a?(DelayedResult)
+      result
+    else
+      DelayedResult.new { result }
+    end
+  end
+
+  def load_many(keys)
+    unless keys.is_a?(Array)
+      raise TypeError, "#load_many must be called with an Array, but got: #{keys.class.name}"
+    end
+
+    delayed_results = keys.map(&method(:load))
+    DelayedResult.new do
+      delayed_results.map(&:value!)
+    end
+  end
+
+  def batch
+    if @batch.nil? || @batch.fulfilled?
+      @batch = Batch.new(@loader_block, name: @name, max_batch_size: @max_batch_size)
+    else
+      @batch
+    end
+  end
+
+  def retrieve_from_cache(key)
+    @cache.compute_if_absent(key) do
+      yield
+    end
+  end
+end

metadata

@@ -1,22 +1,22 @@
 --- !ruby/object:Gem::Specification
 name: strongman
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.5
 platform: ruby
 authors:
 - Caleb Land
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-07-19 00:00:00.000000000 Z
+date: 2020-09-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: concurrent-ruby
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.1'
+  name: concurrent-ruby
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -36,6 +36,7 @@ files:
 - README.md
 - lib/strongman.rb
 - lib/strongman/version.rb
+- lib/wimp.rb
 homepage: https://github.com/caleb/strongman
 licenses:
 - MIT
@@ -55,7 +56,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.2
+rubygems_version: 3.0.6
 signing_key:
 specification_version: 4
 summary: Batch data loading, works great with graphql