pg_advisory_lock 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0f4f49429d7f217072b2927e3f33876f0841caf8d027725bd7981eed010e1a37
-  data.tar.gz: 4bd6514b91d8dad257d575b7147a41f8faa5006dd398a7d98d340a2612c0a4f8
+  metadata.gz: 8a63a1943d6c02f12078939e1c72cad482972a9493ec408a0984308b329e269b
+  data.tar.gz: fcf0e101ff148fa73651759f0dbe4fe32cbaac1be1b5e9a13f26a7816c2d3a15
 SHA512:
-  metadata.gz: 92a9005c04a0b898ec8489027bb065ecf713ec75cc83fe0e7d0c721740874463bd80ccad2919a5be780f7b0cc5d6626593b3e5b844e9eb76b1f33851b41a6f9a
-  data.tar.gz: 8199402c1fd8ab0b8c72df92698f94912a81bc91a4af64ba2edd9385d68d9dcf3e0439ba72ce23424be5e4db3738eade36ebd2127ee6a2372539452180e5c065
+  metadata.gz: 12b30ae854f94924bbced7bd831f4bbe89e682a50a0d01c299b7c3071c79e04e1d8cadadc5017574138d328104dc118f35f0fcee4e69889ea3ebb01bb0b8c8a8
+  data.tar.gz: 33ce47ca69b19de4985d8bf9d7517fecb5d9074bf3fe5ce53518b2b880ae6a7e1110f607a2160a6e8b8ee3928938603c75d53e957ee3f8e7dca1f5961b38682b

data/.github/workflows/tests.yml

@@ -0,0 +1,69 @@
+name: Tests
+on:
+  pull_request:
+  push:
+    branches:
+      - master
+
+jobs:
+  rubocop:
+    runs-on: ubuntu-latest
+    name: Rubocop lint
+    env:
+      RAILS_VERSION: '~> 6.0'
+    steps:
+      - uses: actions/checkout@v2
+      - uses: ruby/setup-ruby@v1
+        with:
+          ruby-version: 2.7
+          bundler-cache: true
+      - name: Run rubocop
+        run: |
+          gem install bundler
+          bundle install
+          bundle exec rake rubocop
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        ruby: [ '2.7', '3.0', '3.1' ]
+        rails: [ '~> 6.0', '~> 7.0' ]
+    name: Tests with Ruby ${{ matrix.ruby }} Activerecord ${{ matrix.rails }}
+    services:
+      # Label used to access the service container
+      postgres:
+        # Docker Hub image
+        image: postgres
+        # Provide the password for postgres
+        env:
+          POSTGRES_DB: postgres_db
+          POSTGRES_PORT: 5432
+          POSTGRES_USER: postgres_user
+          POSTGRES_PASSWORD: postgres_password
+        ports:
+          - 5432:5432
+        # Set health checks to wait until postgres has started
+        options: >-
+          --health-cmd pg_isready
+          --health-interval 10s
+          --health-timeout 5s
+          --health-retries 5
+    env:
+      RAILS_VERSION: ${{ matrix.rails }}
+      POSTGRES_HOST: localhost
+      POSTGRES_DB: postgres_db
+      POSTGRES_PORT: 5432
+      POSTGRES_USER: postgres_user
+      POSTGRES_PASSWORD: postgres_password
+    steps:
+      - uses: actions/checkout@v2
+      - uses: ruby/setup-ruby@v1
+        with:
+          ruby-version: ${{ matrix.ruby }}
+          bundler-cache: true
+      - name: Run tests
+        run: |
+          gem install bundler
+          bundle install
+          cp -v spec/config/database.ci.yml spec/config/database.yml
+          bundle exec rake spec

data/.rubocop.yml

@@ -1,6 +1,6 @@
 AllCops:
   DisplayCopNames: true
-  TargetRubyVersion: 2.5
+  TargetRubyVersion: 2.7
   Exclude:
     - vendor/**/*
     - tmp/**/*

data/README.md

@@ -1,5 +1,9 @@
 # PgAdvisoryLock
 
+![tests](https://github.com/didww/pg_advisory_lock/actions/workflows/tests.yml/badge.svg)
+[![Gem Version](https://badge.fury.io/rb/pg_advisory_lock.svg)](https://badge.fury.io/rb/pg_advisory_lock)
+
+
 Postgresql Advisory Lock for ActiveRecord.
 Allows to use mutex in applications that uses same database.
 
@@ -42,7 +46,7 @@ PgAdvisoryLock::Base.select_values 'SELECT id from users WHERE parent_name = ?',
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. 
-Create `spec/config/database.yml` (look at `spec/config/database.travis.yml` for example).
+Create `spec/config/database.yml` (look at `spec/config/database.example.yml` for example).
 You need to create test database, so run `psql -c 'CREATE DATABASE pg_advisory_lock_test;'`.
 Then, run `rake spec` to run the tests. 
 You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/lib/pg_advisory_lock/base.rb

@@ -23,6 +23,8 @@ module PgAdvisoryLock
       # @param name [Symbol, String]
       # @param value [Integer, Array<(Integer, Integer)>]
       def register_lock(name, value)
+        raise ArgumentError, 'value must be integer or array of integers' unless int_or_array_of_ints?(value)
+
         _lock_names[name.to_sym] = value
       end
 
@@ -61,6 +63,16 @@ module PgAdvisoryLock
       def try_lock(name, transaction: true, shared: false, id: nil, &block)
         new(name, transaction: transaction, shared: shared, id: id, wait: false).lock(&block)
       end
+
+      private
+
+      def int_or_array_of_ints?(value)
+        return true if value.is_a?(Integer)
+
+        return true if value.is_a?(Array) && value.all? { |i| i.is_a?(Integer) }
+
+        false
+      end
     end
 
     # @param name [Symbol, String] - lock name (will be transformed to number).
@@ -84,8 +96,8 @@ module PgAdvisoryLock
     # @return yield
     def lock(&block)
       with_logger do
-        lock_number = name_to_number
-        advisory_lock(lock_number, &block)
+        lock_args = build_lock_args
+        advisory_lock(lock_args, &block)
       end
     end
 
@@ -99,62 +111,68 @@ module PgAdvisoryLock
       logger.tagged(self.class.to_s, name.inspect) { yield }
     end
 
-    def advisory_lock(lock_number, &block)
+    def advisory_lock(lock_args, &block)
       if transaction
-        transaction_lock(lock_number, &block)
+        transaction_lock(lock_args, &block)
       else
-        non_transaction_lock(lock_number, &block)
+        non_transaction_lock(lock_args, &block)
       end
     end
 
-    def transaction_lock(lock_number)
+    def transaction_lock(lock_args)
       raise ArgumentError, 'block required when not within transaction' if !block_given? && !sql_caller_class.transaction_open?
 
-      return perform_lock(lock_number) unless block_given?
+      return perform_lock(lock_args) unless block_given?
 
       sql_caller_class.transaction do
-        perform_lock(lock_number)
+        perform_lock(lock_args)
         yield
       end
     end
 
-    def non_transaction_lock(lock_number)
+    def non_transaction_lock(lock_args)
       raise ArgumentError, 'block required on transaction: false' unless block_given?
 
       begin
-        perform_lock(lock_number)
+        perform_lock(lock_args)
         yield
       ensure
-        perform_unlock(lock_number)
+        perform_unlock(lock_args)
       end
     end
 
-    def perform_lock(lock_number)
+    def perform_lock(lock_args)
       function_name = "pg#{'_try' unless wait}_advisory#{'_xact' if transaction}_lock#{'_shared' if shared}"
 
       if wait
-        sql_caller_class.execute("SELECT #{function_name}(#{lock_number})")
+        sql_caller_class.execute("SELECT #{function_name}(#{lock_args})")
       else
-        result = sql_caller_class.select_value("SELECT #{function_name}(#{lock_number})")
-        raise LockNotObtained, "#{self.class} can't obtain lock (#{name}, #{id})" unless result
+        result = sql_caller_class.select_value("SELECT #{function_name}(#{lock_args})")
+        raise LockNotObtained, "#{self.class} can't obtain lock (#{name}, #{id.inspect})" unless result
       end
     end
 
-    def perform_unlock(lock_number)
-      sql_caller_class.select_value("SELECT pg_advisory_unlock#{'_shared' if shared}(#{lock_number})")
+    def perform_unlock(lock_args)
+      sql_caller_class.select_value("SELECT pg_advisory_unlock#{'_shared' if shared}(#{lock_args})")
     end
 
     # Converts lock name to number, because pg advisory lock functions accept only bigint numbers.
     # @return [String] lock number or two numbers delimited by comma.
-    def name_to_number
-      lock_number = _lock_names.fetch(name) do
+    def build_lock_args
+      lock_args = _lock_names.fetch(name) do
         raise ArgumentError, "lock name #{name.inspect} is invalid, see #{self.class}::NAMES"
       end
-      lock_number = Array.wrap(lock_number)
-      lock_number.push(id) if id.present?
-      raise ArgumentError, "can't use lock name #{name.inspect} with id" if lock_number.size > 2
+      lock_args = Array.wrap(lock_args)
+      if id.present?
+        if id.is_a?(Integer)
+          lock_args.push(id)
+        else
+          lock_args.push("hashtext(#{sql_caller_class.connection.quote(id.to_s)})")
+        end
+      end
+      raise ArgumentError, "can't use lock name #{name.inspect} with id" if lock_args.size > 2
 
-      lock_number.join(', ')
+      lock_args.join(', ')
     end
 
     def sql_caller_class

data/lib/pg_advisory_lock/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module PgAdvisoryLock
-  VERSION = '0.3.0'
+  VERSION = '0.4.0'
 end

data/pg_advisory_lock.gemspec

@@ -29,5 +29,5 @@ Gem::Specification.new do |spec|
 
   spec.add_dependency 'activerecord'
   spec.add_dependency 'activesupport'
-  spec.add_dependency 'pg_sql_caller'
+  spec.add_dependency 'pg_sql_caller', '>= 0.2.2'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_advisory_lock
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Denis Talakevich
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-12-13 00:00:00.000000000 Z
+date: 2023-02-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -44,14 +44,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.2.2
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.2.2
 description: Postgresql Advisory Lock for ActiveRecord.
 email:
 - senid231@gmail.com
@@ -59,10 +59,10 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".github/workflows/tests.yml"
 - ".gitignore"
 - ".rspec"
 - ".rubocop.yml"
-- ".travis.yml"
 - CODE_OF_CONDUCT.md
 - Gemfile
 - LICENSE.txt

data/.travis.yml

@@ -1,6 +0,0 @@
----
-language: ruby
-cache: bundler
-rvm:
-  - 2.5.5
-before_install: gem install bundler -v 2.1.4