pgdog 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: fd3fd8e3f0eac4f807e975eed7dd9c619386b1b1133aad2798a2dc96bc61fe98
+  data.tar.gz: b861aec2cf7e56316da42e588ba434365ebe159779651d913d3b7a9038325ed6
+SHA512:
+  metadata.gz: d1238a7fdcdf690550f30ec7f819096fa57f93a48851303f06a1efcd8378dc8bd11762e3ca8fae77152d9f824e4b80b99abf7b61cae1ee580f9d99985f57bef9
+  data.tar.gz: 0522ec89011c0c3259c85370a7db82231c92ff2a4f108f19e8a21146f9cc494588c319f72770cbaf873c1140702ea6df889997f4bdce093780e3f1dda78743ea

data/lib/pgdog.rb

@@ -0,0 +1,61 @@
+class PgDog
+  # Get a connection from ActiveRecord.
+  def self.connection
+    return ActiveRecord::Base.connection
+  end
+
+  # Start a transaction and set the shard number
+  # manually using SET.
+  def self.with_shard(shard)
+    # Basic SQL injection protection
+    shard = shard.to_i
+
+    PgDog.check_transaction
+    ActiveRecord::Base.transaction do
+      self.connection.execute "SET \"pgdog.shard\" TO #{shard}"
+      yield
+    end
+  end
+
+  # Start a transaction and set the sharding key
+  # manually using SET.
+  def self.with_sharding_key(key)
+    # Basic SQL injection protection.
+    key.to_s.sub "'", "''"
+
+    PgDog.check_transaction
+    ActiveRecord::Base.transaction do
+      self.connection.execute "SET \"pgdog.sharding_key\" TO '#{key}'"
+      yield
+    end
+  end
+
+  # Get currently set shard, if any.
+  def self.shard
+    shard = self.connection.execute "SELECT current_setting('pgdog.shard', true)"
+    shard = shard[0]["current_setting"]
+
+    if shard.nil?
+      return nil
+    else
+      return shard.to_i
+    end
+  end
+
+  # Get currently set sharding key, if any.
+  def self.sharding_key
+    key = self.connection.execute "SELECT current_setting('pgdog.sharding_key', true)"
+    key[0]["current_setting"]
+  end
+
+  # Ensure a transaction isn't started already.
+  def self.check_transaction
+    if ActiveRecord::Base.connection.open_transactions != 0
+      raise PgDogError, "Transaction already started, can't set route"
+    end
+  end
+end
+
+# Error raised if a transaction is already started.
+class PgDogError < StandardError
+end

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification
+name: pgdog
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Lev Kokotov
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: active_record
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5.0'
+    - - "<="
+      - !ruby/object:Gem::Version
+        version: '9.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5.0'
+    - - "<="
+      - !ruby/object:Gem::Version
+        version: '9.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.13'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+description: Add routing hints to the application to enable direct-to-shard transaction
+  routing in ambiguous contexts.
+email: hi@pgdog.dev
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/pgdog.rb
+homepage: https://rubygems.org/gems/pgdog
+licenses:
+- MIT
+metadata: {}
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.7
+specification_version: 4
+summary: PgDog plugin for Ruby on Rails.
+test_files: []