rndb 0.1.1

data/lib/rndb/thicket.rb

@@ -0,0 +1,116 @@
+# frozen_string_literal: true
+
+module RnDB
+  class Thicket
+    include Enumerable
+
+    # A sorted array of disjoint ranges, optionally initialised with a default range.
+    def initialize(range=nil)
+      @ids = []
+      @ids << Slice.new(range.min, range.max) unless range.nil?
+    end
+
+    # Append a range, throwing an error if it overlaps with an existing range,
+    # and keeping the resulting array sorted so we can iterate over values.
+    def <<(range)
+      slice = Slice.new(range.min, range.max)
+      raise "slices in thicket must be disjoint" unless @ids.all? { |id| (id & slice).nil? }
+      @ids << Slice.new(range.min, range.max)
+      @ids.sort!
+      self
+    end
+
+    # Intersect two Thickets, useful when performing queries.
+    def &(other)
+      retval = self.class.new
+      slices.each do |slice|
+        other.slices.each do |other_slice|
+          next unless (intersection = slice & other_slice)
+          retval << intersection
+        end
+      end
+      retval
+    end
+
+    # Merge two Thickets, which we need during migrations.
+    def |(other)
+      retval = self.class.new
+      slices.each { |slice| retval << slice }
+      other.slices.each { |slice| retval << slice }
+      retval
+    end
+
+    # Subdivide a Thicket with a probability range, also used for migrations.
+    def *(other)
+      slices.each_with_object(self.class.new) do |slice, thicket|
+        min = slice.min + (slice.count * other.min).round
+        max = slice.min + (slice.count * other.max).round - 1
+        thicket << (min..max) unless max < min
+      end
+    end
+
+    # Sum the counts of the Slices in the Thicket.
+    def count
+      @ids.map(&:count).sum
+    end
+
+    # Return the ID corresponding to the supplied index.
+    def [](index)
+      index += count while index.negative?
+      @ids.each do |slice|
+        return index + slice.min if index < slice.count
+        index -= slice.count
+      end
+      nil
+    end
+
+    # Return the index corresponding to the supplied ID.
+    def index(id)
+      start = 0
+      @ids.each do |slice|
+        return start + id - slice.min if slice.include?(id)
+        start += slice.count
+      end
+      nil
+    end
+
+    # Test whether the specified ID exists in this Thicket.
+    def include?(id)
+      @ids.any? { |slice| slice.include?(id) }
+    end
+
+    # Implemented to be consistent with #first, which we get by magic.
+    def last
+      self[-1] unless count.zero?
+    end
+
+    # Iterate over each slice in the Thicket in turn.
+    def each(&block)
+      @ids.each { |slice| slice.each(&block) }
+    end
+
+    # Return a Thicket that contains a sampling of IDs.
+    def sample(limit=1, prng=Random)
+      ids = Set.new
+      limit = [limit, count].min
+      ids << self[prng.rand(count)] while ids.count < limit
+      ids.sort.each_with_object(self.class.new) do |id, thicket|
+        thicket << Slice.new(id, id)
+      end
+    end
+
+    # We display the internal slices.
+    def to_s
+      slices.to_s
+    end
+
+    alias min first
+    alias max last
+
+    protected
+
+    def slices
+      @ids
+    end
+  end
+end

data/rndb.gemspec

@@ -0,0 +1,91 @@
+# Generated by juwelier
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Juwelier::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+# stub: rndb 0.1.1 ruby lib
+
+Gem::Specification.new do |s|
+  s.name = "rndb".freeze
+  s.version = "0.1.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0".freeze) if s.respond_to? :required_rubygems_version=
+  s.require_paths = ["lib".freeze]
+  s.authors = ["Lloyd Kranzky".freeze]
+  s.date = "2021-02-25"
+  s.description = "".freeze
+  s.email = "lloyd@kranzky.com".freeze
+  s.extra_rdoc_files = [
+    "LICENSE",
+    "README.md"
+  ]
+  s.files = [
+    ".document",
+    ".editorconfig",
+    ".github/workflows/ship.yml",
+    ".github/workflows/test.yml",
+    ".rdoc_options",
+    ".rgignore",
+    ".rspec",
+    ".rubocop.yml",
+    ".ruby-gemset",
+    ".ruby-version",
+    "Gemfile",
+    "Gemfile.lock",
+    "LICENSE",
+    "README.md",
+    "Rakefile",
+    "VERSION",
+    "lib/rndb.rb",
+    "lib/rndb/database.rb",
+    "lib/rndb/query.rb",
+    "lib/rndb/slice.rb",
+    "lib/rndb/table.rb",
+    "lib/rndb/thicket.rb",
+    "rndb.gemspec",
+    "spec/rn_db/database_spec.rb",
+    "spec/rn_db/query_spec.rb",
+    "spec/rn_db/slice_spec.rb",
+    "spec/rn_db/table_spec.rb",
+    "spec/rn_db/thicket_spec.rb",
+    "spec/spec_helper.rb"
+  ]
+  s.homepage = "https://github.com/kranzky/rndb".freeze
+  s.licenses = ["Unlicense".freeze]
+  s.required_ruby_version = Gem::Requirement.new(">= 2.1".freeze)
+  s.rubygems_version = "2.7.6.2".freeze
+  s.summary = "RnDB is an procedurally-generated mock database.".freeze
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 4
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<rdoc>.freeze, ["~> 6.3"])
+      s.add_runtime_dependency(%q<yard>.freeze, ["~> 0.9"])
+      s.add_development_dependency(%q<bundler>.freeze, ["~> 1.17"])
+      s.add_development_dependency(%q<gemfile_updater>.freeze, ["~> 0.1"])
+      s.add_development_dependency(%q<juwelier>.freeze, ["~> 2.4"])
+      s.add_development_dependency(%q<byebug>.freeze, ["~> 11.1"])
+      s.add_development_dependency(%q<crypt-isaac>.freeze, ["~> 1.2"])
+      s.add_development_dependency(%q<faker>.freeze, ["~> 2.16"])
+    else
+      s.add_dependency(%q<rdoc>.freeze, ["~> 6.3"])
+      s.add_dependency(%q<yard>.freeze, ["~> 0.9"])
+      s.add_dependency(%q<bundler>.freeze, ["~> 1.17"])
+      s.add_dependency(%q<gemfile_updater>.freeze, ["~> 0.1"])
+      s.add_dependency(%q<juwelier>.freeze, ["~> 2.4"])
+      s.add_dependency(%q<byebug>.freeze, ["~> 11.1"])
+      s.add_dependency(%q<crypt-isaac>.freeze, ["~> 1.2"])
+      s.add_dependency(%q<faker>.freeze, ["~> 2.16"])
+    end
+  else
+    s.add_dependency(%q<rdoc>.freeze, ["~> 6.3"])
+    s.add_dependency(%q<yard>.freeze, ["~> 0.9"])
+    s.add_dependency(%q<bundler>.freeze, ["~> 1.17"])
+    s.add_dependency(%q<gemfile_updater>.freeze, ["~> 0.1"])
+    s.add_dependency(%q<juwelier>.freeze, ["~> 2.4"])
+    s.add_dependency(%q<byebug>.freeze, ["~> 11.1"])
+    s.add_dependency(%q<crypt-isaac>.freeze, ["~> 1.2"])
+    s.add_dependency(%q<faker>.freeze, ["~> 2.16"])
+  end
+end
+

data/spec/rn_db/database_spec.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+describe RnDB::Database do
+  before(:all) do # rubocop:disable RSpec/BeforeAfterAll
+    described_class.new(137).add_table(Ball, 1_000_000)
+  end
+
+  it "can be counted" do
+    expect(Ball.count).to eq(1_000_000)
+  end
+
+  it "has a first element" do
+    expect(Ball.first.id).to eq(0)
+  end
+
+  it "has a last element" do
+    expect(Ball.last.id).to eq(999_999)
+  end
+
+  it "can find things" do
+    ball = Ball.find { |b| b.location =~ /island/i }
+    expect(ball.location).to match(/Island/)
+  end
+
+  it "can take random samples" do
+    ids = Ball.sample(10).pluck(:id)
+    expect(ids.sort.uniq.size).to eq(10)
+  end
+
+  context "when running a query" do
+    let(:query) do
+      Ball.where(colour: [:red, :blue], material: :wood)
+    end
+
+    it "can be counted" do
+      expect(query.count).to eq(240_000)
+    end
+
+    it "has a first element" do
+      expect(query.first.id).to eq(5400)
+    end
+
+    it "has a last element" do
+      expect(query.last.id).to eq(905_499)
+    end
+
+    it "can find things" do
+      ball = query.find { |b| !b.transparent }
+      expect(ball.transparent).to be(false)
+    end
+
+    it "can take random samples" do
+      ids = query.sample(10).pluck(:id)
+      expect(ids.sort.uniq.size).to eq(10)
+    end
+  end
+end

data/spec/rn_db/query_spec.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+describe RnDB::Query do
+  context 'when no tests have been written' do
+    it 'displays "not yet implemented"'
+  end
+end

data/spec/rn_db/slice_spec.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+describe RnDB::Slice do
+  it "has a count" do
+    slice = described_class.new(5, 13)
+    expect(slice.count).to eq(slice.to_a.length)
+  end
+
+  it "can find the intersection of two slices" do
+    left = described_class.new(5, 13)
+    right = described_class.new(10, 17)
+    expect(left & right).to eq(described_class.new(10, 13))
+  end
+
+  it "returns nil if the intersection is empty" do
+    left = described_class.new(5, 10)
+    right = described_class.new(12, 17)
+    expect(left & right).to be_nil
+  end
+end

data/spec/rn_db/table_spec.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+describe RnDB::Table do
+  context 'when no tests have been written' do
+    it 'displays "not yet implemented"'
+  end
+end

data/spec/rn_db/thicket_spec.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+describe RnDB::Thicket do
+  it "is empty by default" do
+    thicket = described_class.new
+    expect(thicket.count).to be_zero
+  end
+
+  it "can be initialized with a value" do
+    thicket = described_class.new(2..21)
+    expect(thicket.count).to eq((2..21).size)
+  end
+
+  it "can contain multiple disjoint slices" do
+    thicket = described_class.new
+    thicket << (11..16)
+    thicket << (4..9)
+    expect(thicket.to_a).to eq((4..9).to_a | (11..16).to_a)
+  end
+
+  it "can contain large values" do
+    thicket = described_class.new(7..1e15)
+    expect((thicket.min..thicket.max)).to eq((7..1e15.to_i))
+  end
+
+  it "can return the id by index" do
+    thicket = described_class.new
+    thicket << (11..16)
+    thicket << (4..9)
+    expect(thicket[7]).to eq(12)
+  end
+
+  it "returns a nil value when the index is out of range" do
+    thicket = described_class.new
+    thicket << (11..16)
+    thicket << (4..9)
+    expect(thicket[16]).to be_nil
+  end
+
+  it "can get the index of an id" do
+    thicket = described_class.new
+    thicket << (11..16)
+    thicket << (4..9)
+    expect(thicket.index(12)).to eq(7)
+  end
+
+  it "returns a nil index when the id does not exist" do
+    thicket = described_class.new
+    thicket << (11..16)
+    thicket << (4..9)
+    expect(thicket.index(10)).to be_nil
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require 'simplecov'
+require 'coveralls'
+SimpleCov.start do
+  add_filter 'spec'
+end
+Coveralls.wear!
+
+require 'faker'
+require_relative '../lib/rndb'
+
+class Ball < RnDB::Table
+  column :colour, {
+    red: 0.3,
+    green: 0.1,
+    brown: 0.01,
+    blue: 0.5,
+    orange: 0.09
+  }
+  column :transparent, {
+    true => 0.1,
+    false => 0.9
+  }
+  column :weight, {
+    light: 0.3,
+    medium: 0.6,
+    heavy: 0.1
+  }, lambda { |value|
+    lookup = {
+      light: (0.1..3.0),
+      medium: (3.0..6.0),
+      heavy: (6.0..9.9)
+    }
+    self.rand(lookup[value])
+  }
+  column :material, {
+    leather: 0.2,
+    steel: 0.4,
+    wood: 0.3,
+    fluff: 0.1
+  }
+  column :name, -> { Faker::Games::Pokemon.name }
+  column :location, -> { Faker::Games::Pokemon.location }
+  column :move, -> { Faker::Games::Pokemon.move }
+end

metadata

@@ -0,0 +1,186 @@
+--- !ruby/object:Gem::Specification
+name: rndb
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- Lloyd Kranzky
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2021-02-25 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rdoc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.3'
+- !ruby/object:Gem::Dependency
+  name: yard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.17'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.17'
+- !ruby/object:Gem::Dependency
+  name: gemfile_updater
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+- !ruby/object:Gem::Dependency
+  name: juwelier
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.4'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.4'
+- !ruby/object:Gem::Dependency
+  name: byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '11.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '11.1'
+- !ruby/object:Gem::Dependency
+  name: crypt-isaac
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+- !ruby/object:Gem::Dependency
+  name: faker
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.16'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.16'
+description: ''
+email: lloyd@kranzky.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- LICENSE
+- README.md
+files:
+- ".document"
+- ".editorconfig"
+- ".github/workflows/ship.yml"
+- ".github/workflows/test.yml"
+- ".rdoc_options"
+- ".rgignore"
+- ".rspec"
+- ".rubocop.yml"
+- ".ruby-gemset"
+- ".ruby-version"
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README.md
+- Rakefile
+- VERSION
+- lib/rndb.rb
+- lib/rndb/database.rb
+- lib/rndb/query.rb
+- lib/rndb/slice.rb
+- lib/rndb/table.rb
+- lib/rndb/thicket.rb
+- rndb.gemspec
+- spec/rn_db/database_spec.rb
+- spec/rn_db/query_spec.rb
+- spec/rn_db/slice_spec.rb
+- spec/rn_db/table_spec.rb
+- spec/rn_db/thicket_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/kranzky/rndb
+licenses:
+- Unlicense
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.1'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project:
+rubygems_version: 2.7.6.2
+signing_key:
+specification_version: 4
+summary: RnDB is an procedurally-generated mock database.
+test_files: []