pg_conn 0.55.0 → 0.56.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c7a478850b588c635e78fd50cac46e52665ece96fc538eb467eb2bf88fe6d6d7
-  data.tar.gz: 8882d70f0289f8b3cdec8c9e3135513788eb09416a0ceb5ad763849f71324d95
+  metadata.gz: 26dbcd85db043fdf28c698606f7f35450d77bf0cbafb63324d62297724a4698f
+  data.tar.gz: 59444453e3ca2b4c6244d356696264f7caef6f20038549edc33ad085e9339f08
 SHA512:
-  metadata.gz: 5c3709874689cbfadf194fde8e4dd1d956ab0456ab205fd70064b680aa5dcebc161e11e5f3441237f502bf2c4df7cafa5a9a9b6608df42cd21ddc75848062c64
-  data.tar.gz: b791c4afaefddc8286abf836abdbb6e92657bb5139bc377be032f174a6d2888dc774c8f8fd3c8643c6ca9f98f920d3f747d3163380c2fa85569a373ac30e45f0
+  metadata.gz: 5fddf126010d3cd13cefbbb8bf1a94402cee13987c967e17e346b32db9c02044c1e035ddd0f69d0c2a1fd91cdfedf8e0cbbf8da556c1c1588c6bb8bb5e28092c
+  data.tar.gz: aa082a3881474398e3c17c19bd219801f65596470b69d7046217d43093a511c06ef436e5f13189a602d97ca3ab3657d68582c4a53338f38b5f05810f126921d7

data/lib/pg_conn/version.rb

@@ -1,3 +1,3 @@
 module PgConn
-  VERSION = "0.55.0"
+  VERSION = "0.56.1"
 end

data/lib/pg_conn.rb

@@ -1,5 +1,6 @@
 require "pg"
 require 'ostruct'
+require 'open3'
 require 'json'
 
 require_relative "pg_conn/version"
@@ -206,8 +207,8 @@ module PgConn
     def logger() = @logger
 
     # Controls error messages. It can be assigned true, false, nil, or a Proc
-    # object that recieves the message. True causes the message to be printed
-    # to standard error, false ignores it, and nil resets the state to the
+    # object that recieves the message. True causes the message to be ignored,
+    # false prints it to standard error, and nil resets the state to the
     # default given when the connection was initialized. #silent? returns true
     # if #silent is false or a Proc object and should be used instead #silent
     # to check the state because #silent returns truish when output is
@@ -1066,6 +1067,34 @@ module PgConn
       end
     end
 
+    # Execute SQL or pgsql file. The file must be pure SQL or pgsql statements,
+    # not psql meta commands. Any output from the file is ignored
+    def sqlexec(file, **opts)
+      source = IO.read(file) or error "Can't read #{file}"
+      exec(source, **opts)
+    end
+
+    # Execute file using psql(1) so that special psql meta commands (eg. \gset)
+    # can be used. Return the output from the SQL file. The file is executed
+    # with ON_ERROR_STOP on
+    def psqlexec(file, fail: true, silent: self.silent)
+      begin
+        output, errors, status = Open3.capture3 %(
+            psql -U #{username} -d #{database} --no-psqlrc -c '\\set ON_ERROR_STOP on' -f #{file}
+        )
+        status.success? or raise PG::Error.new(errors)
+        output
+      rescue PG::Error => ex
+        if @error.nil?
+          @error = ex
+          @err = nil
+        end
+        $stderr.puts errors if !silent
+        raise if fail
+        nil
+      end
+    end
+
     # Execute block with the given set of global or local options and reset
     # them afterwards
     #
@@ -1542,11 +1571,12 @@ module PgConn
             @err = nil
           end
 
-          # Process message before resetting silent state
+          # Process message
           message_processor(@error.message, arg)
-          self.silent = saved_silent
 
           raise
+        ensure
+          self.silent = saved_silent
         end
 
       # For dump of SQL statements

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pg_conn
 version: !ruby/object:Gem::Version
-  version: 0.55.0
+  version: 0.56.1
 platform: ruby
 authors:
 - Claus Rasmussen