pg_conn 0.21.0 → 0.22.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6f293ad25581975cbd57dc5973d072043d6b5cf13093a21bd354720b39c91c12
-  data.tar.gz: 1ad472d136a9b626717f90058cb220eaaf4c87bced2f8802c8cab68b69bf187a
+  metadata.gz: 8d8571dfbe07ffc9e846741807d87d45373553d987e353693aa4599010804b3e
+  data.tar.gz: 4d0bd8faced962ac9c0837b3a98f1a6f0ee4fbc0c1c1aa4a12c448472e10376a
 SHA512:
-  metadata.gz: 7720aef7f128e0b8e3c47fdd2acb752b1d44af9625653eae0552a2ac619e3c29ae2cad25789f0f0806fe4576104a64210cf5a5d7e7c7fa66817b8538a556bd35
-  data.tar.gz: 61425eaeabb6b6e0872f6ed49064f8650361ecd8be684b7db2322a6a346265e253489e49be37688c9ef74171733d2c8f19a12d38105a188ba263cf5ea48c9593
+  metadata.gz: 3b3ff2ae7e1fa63f697616e16eff4065e5118ef5fe9cb446837cc30b028df5a385b56848e23546c7718f5f704e622cd35175acbd56aacd6fd56774dec9c34627
+  data.tar.gz: 4c9428ef6abc6162c068f154575eb1d8c7b63c1aa072af4350c7bacb2c6ea63e0ffea80a470f9ca8b72c635e83ceda7345b54bc6a24d5adc450bf0863fb0df1b

data/lib/pg_conn/rdbms_methods.rb

@@ -21,7 +21,7 @@ module PgConn
     end
 
     # Create a new database
-    def create(database, owner: ENV['USER'], template: "template1") 
+    def create(database, owner: ENV['USER'], template: "template1")
       owner_clause = owner ? "owner = \"#{owner}\"" : nil
       template_clause = template ? "template = \"#{template}\"" : nil
       stmt = ["create database \"#{database}\"", owner_clause, template_clause].compact.join(" ")
@@ -80,7 +80,7 @@ module PgConn
       local = !database.nil?
       begin
         conn = local ? PgConn.new(database) : self.conn
-        schemas = 
+        schemas =
             conn
               .values("select nspname from pg_namespace where nspowner != 10 or nspname = 'public'")
               .select { |schema| !exclude.include?(schema) }
@@ -90,7 +90,7 @@ module PgConn
         # FIXME FIXME FIXME SECURITY Why grant 'create' to public?
         conn.exec %(
             create schema public authorization postgres;
-            grant usage, create on schema public to public 
+            grant usage, create on schema public to public
         ) if public
       ensure
         conn&.terminate if local

data/lib/pg_conn/role_methods.rb

@@ -18,7 +18,7 @@ module PgConn
           else
             nil
           end
-      can_login_clause = 
+      can_login_clause =
           case can_login
             when true; "rolcanlogin"
             when false; "not rolcanlogin"
@@ -62,7 +62,7 @@ module PgConn
     # counldn't be deleted
     #
     # Note that cascade only works if connected to the database where the
-    # privileges exist. 
+    # privileges exist.
     #
     # TODO The :silent option is used in tests - fix it somehow!
     def drop(*rolenames, cascade: false, fail: true, silent: false)
@@ -86,7 +86,7 @@ module PgConn
       superuser_clause = superuser.nil? ? nil : "rolsuper = #{superuser}"
       can_login_clause = can_login.nil? ? nil : "rolcanlogin = #{can_login}"
       query = [
-          "select rolname from pg_roles where true", 
+          "select rolname from pg_roles where true",
           database_clause, superuser_clause, can_login_clause
       ].compact.join(" and ")
       conn.values(query)

data/lib/pg_conn/schema_methods.rb

@@ -49,8 +49,8 @@ module PgConn
     # Empty all tables in the given schema
     def clean!(schema, exclude: [])
       conn.session.triggers(false) {
-        self.list_tables(schema, exclude: exclude).each { |table| 
-          conn.exec "delete from #{schema}.#{table}" 
+        self.list_tables(schema, exclude: exclude).each { |table|
+          conn.exec "delete from #{schema}.#{table}"
         }
       }
     end
@@ -125,8 +125,8 @@ module PgConn
     end
 
     # Return name of the table's sequence (if any)
-    def sequence(schema, table) 
-      conn.value "select pg_get_serial_sequence('#{schema}.#{table}', 'id')" 
+    def sequence(schema, table)
+      conn.value "select pg_get_serial_sequence('#{schema}.#{table}', 'id')"
     end
 
     # Get the current serial value for the table. Returns nil if the serial has
@@ -168,7 +168,7 @@ module PgConn
       kind_sql_list = "'" + (kind.nil? ? %w(r f v m) : Array(kind).flatten).join("', '") + "'"
       %(
           select  1
-          from    pg_class 
+          from    pg_class
           where   relnamespace::regnamespace::text = '#{schema}'
           and     relname = '#{relation}'
           and     relkind in (#{kind_sql_list})
@@ -183,7 +183,7 @@ module PgConn
       exclude_expr = exclude.empty? ? "true = true" : "not relname in (#{exclude_list})"
       %(
           select  relname
-          from    pg_class 
+          from    pg_class
           where   relnamespace::regnamespace::text = '#{schema}'
           and     #{kind_expr}
           and     #{exclude_expr}
@@ -211,7 +211,7 @@ module PgConn
             join    pg_attribute a on a.attrelid = c.oid
             where   relnamespace::regnamespace::text = '#{schema}'
             and     a.attnum > 0
-          ), 
+          ),
           relation_clause
       ].compact.join(" and ")
     end

data/lib/pg_conn/version.rb

@@ -1,3 +1,3 @@
 module PgConn
-  VERSION = "0.21.0"
+  VERSION = "0.22.0"
 end

data/lib/pg_conn.rb

@@ -515,8 +515,9 @@ module PgConn
 
     # Returns a hash from the first field to a tuple of the remaining fields.
     # If there is only one remaining field then that value is used instead of a
-    # tuple of that value. The optional +key+ argument sets the mapping field
-    def map(query, key = nil) # TODO Swap arguments
+    # tuple. The optional +key+ argument sets the mapping field and the
+    # +symbol+ option convert key to Symbol objects when true
+    def map(query, key = nil, symbol: false) # TODO Swap arguments
       r = pg_exec(query)
       begin
         key = (key || r.fname(0)).to_s
@@ -528,13 +529,18 @@ module PgConn
       h = {}
       r.each_row { |row|
         key_value = row.delete_at(key_index)
+        key_value = key_value.to_sym if symbol
         !h.key?(key_value) or raise Error, "Duplicate key: #{key_value}"
         h[key_value] = (one ? row.first : row)
       }
       h
     end
 
-    def multimap(query, key = nil)
+    # Like #map but values of duplicate keys are concatenated. It acts as a
+    # group-by on the key and array_agg on the remaining values. The value is
+    # an array of tuples if the query has more than one value field and an
+    # array of values if there is only one value field
+    def multimap(query, key = nil, symbol: false)
       r = pg_exec(query)
       begin
         key = (key || r.fname(0)).to_s
@@ -546,6 +552,7 @@ module PgConn
       h = {}
       r.each_row { |row|
         key_value = row.delete_at(key_index)
+        key_value = key_value.to_sym if symbol
         (h[key_value] ||= []) << (one ? row.first : row)
       }
       h

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_conn
 version: !ruby/object:Gem::Version
-  version: 0.21.0
+  version: 0.22.0
 platform: ruby
 authors:
 - Claus Rasmussen
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-08-07 00:00:00.000000000 Z
+date: 2024-08-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg