db-postgres 0.2.4 → 0.5.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9914774aac9e6cf060fdba11e1ce395b7311ee55c514b3720a94b8252aaf82f6
-  data.tar.gz: fa42571a59bc7c9f12103359a4e396607536a6542a5ab37bd6a736b882450792
+  metadata.gz: cf6b2560d3bc8cc9b355b8e316eac4cf5c85fc9e134ee78aa3a36284e773ef4a
+  data.tar.gz: e69437b200e76fe8e16e96d73cc674f8f632ac468bcc328badbce3fa91046194
 SHA512:
-  metadata.gz: 9766e98f2f42277a5ffe6df30777f8f328f8d12146cce6dddfcc0e9f50a8c1359e4b2d4766377775eb996046219e188b14e7f728609da7544d338363dacc478f
-  data.tar.gz: d0a9718c070dede2a6aa61e060518c8a43d0448345a0e0c79a20ff1486f89a303cd0ef120ac9137ab13575044b3811574904056c3342ce0073ab6b532ea3754f
+  metadata.gz: 971db1d0d40d8817e6169c1d090f98fd1ddb6b4e2dcb0c5b7ee7cd05ddcbb065d54fe1ec243635cbc53a83b4b0f2c0e18328193b1b20481fb1adc1b370944a3c
+  data.tar.gz: de60cce07b7cadc009344fe4cb370978b741c2f99c2485adc24f88147c4b59171f96a133490c511b9bccd43a104536f971a3f59fe43f02000a8f858322a8bb82

data/lib/db/postgres/connection.rb

@@ -65,20 +65,37 @@ module DB
 			end
 			
 			def append_identifier(value, buffer = String.new)
-				buffer << @native.escape_identifier(value)
+				case value
+				when Array
+					first = true
+					value.each do |part|
+						buffer << '.' unless first
+						first = false
+						
+						buffer << @native.escape_identifier(part)
+					end
+				else
+					buffer << @native.escape_identifier(value)
+				end
 				
 				return buffer
 			end
 			
-			def id_column(name = 'id', primary_key: true)
+			def key_column(name = 'id', primary: true, null: false)
 				buffer = String.new
 				
 				append_identifier(name, buffer)
 				
-				buffer << " BIGSERIAL"
+				if primary
+					buffer << " BIGSERIAL"
+				else
+					buffer << " BIGINT"
+				end
 				
-				if primary_key
+				if primary
 					buffer << " PRIMARY KEY"
+				elsif !null
+					buffer << " NOT NULL"
 				end
 				
 				return buffer

data/lib/db/postgres/native.rb

@@ -18,14 +18,38 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 
-require 'ffi'
+require 'ffi/module'
+require 'ffi/module/config_tool'
 
 module DB
 	module Postgres
 		module Native
-			extend FFI::Library
+			extend FFI::Module::Library
+			extend FFI::Module::Loader
+			extend FFI::Module::ConfigTool
 			
-			ffi_lib 'pq'
+			ffi_load('pq') ||
+				ffi_load_using_config_tool(%w{pg_config --libdir}, names: ['pq']) ||
+				ffi_load_failure(<<~EOF)
+					Unable to load libpq!
+					
+					## Ubuntu
+					
+						sudo apt-get install postgresql postgresql-contrib
+					
+					## Arch Linux
+					
+						sudo pacman -S postgresql
+					
+					## MacPorts
+					
+						sudo port install postgresql11
+					
+					## Homebrew
+					
+						brew install postgresql
+					
+				EOF
 		end
 	end
 end

data/lib/db/postgres/native/connection.rb

@@ -37,23 +37,23 @@ module DB
 				attr :array
 			end
 			
-			attach_function :connect_start_params, :PQconnectStartParams, [:pointer, :pointer, :int], :pointer
+			ffi_attach_function :PQconnectStartParams, [:pointer, :pointer, :int], :pointer, as: :connect_start_params
 			
-			enum :polling_status, [
+			ffi_define_enumeration :polling_status, [
 				:failed,
 				:wait_readable,
 				:wait_writable,
 				:ok,
 			]
 			
-			attach_function :connect_poll, :PQconnectPoll, [:pointer], :polling_status
+			ffi_attach_function :PQconnectPoll, [:pointer], :polling_status, as: :connect_poll
 			
 			# Close the connection and release underlying resources.
-			attach_function :finish, :PQfinish, [:pointer], :void
+			ffi_attach_function :PQfinish, [:pointer], :void, as: :finish
 			
-			attach_function :error_message, :PQerrorMessage, [:pointer], :string
+			ffi_attach_function :PQerrorMessage, [:pointer], :string, as: :error_message
 			
-			enum :status, [
+			ffi_define_enumeration :status, [
 				# Normal mode:
 				:ok,
 				:bad,
@@ -70,31 +70,31 @@ module DB
 				:consume, # Wait for any pending message and consume them.
 			]
 			
-			attach_function :status, :PQstatus, [:pointer], :status
+			ffi_attach_function :PQstatus, [:pointer], :status, as: :status
 			
-			attach_function :socket, :PQsocket, [:pointer], :int
+			ffi_attach_function :PQsocket, [:pointer], :int, as: :socket
 			
-			attach_function :set_nonblocking, :PQsetnonblocking, [:pointer, :int], :int
-			attach_function :flush, :PQflush, [:pointer], :int
+			ffi_attach_function :PQsetnonblocking, [:pointer, :int], :int, as: :set_nonblocking
+			ffi_attach_function :PQflush, [:pointer], :int, as: :flush
 			
 			# Submits a command to the server without waiting for the result(s). 1 is returned if the command was successfully dispatched and 0 if not (in which case, use PQerrorMessage to get more information about the failure).
-			attach_function :send_query, :PQsendQuery, [:pointer, :string], :int
+			ffi_attach_function :PQsendQuery, [:pointer, :string], :int, as: :send_query
 			
 			# int PQsendQueryParams(PGconn *conn, const char *command, int nParams, const Oid *paramTypes, const char * const *paramValues, const int *paramLengths, const int *paramFormats, int resultFormat);
-			attach_function :send_query_params, :PQsendQueryParams, [:pointer, :string, :int, :pointer, :pointer, :pointer, :pointer, :int], :int
+			ffi_attach_function :PQsendQueryParams, [:pointer, :string, :int, :pointer, :pointer, :pointer, :pointer, :int], :int, as: :send_query_params
 			
-			attach_function :set_single_row_mode, :PQsetSingleRowMode, [:pointer], :int
+			ffi_attach_function :PQsetSingleRowMode, [:pointer], :int, as: :set_single_row_mode
 			
-			attach_function :get_result, :PQgetResult, [:pointer], :pointer
+			ffi_attach_function :PQgetResult, [:pointer], :pointer, as: :get_result
 			
 			# If input is available from the server, consume it:
-			attach_function :consume_input, :PQconsumeInput, [:pointer], :int
+			ffi_attach_function :PQconsumeInput, [:pointer], :int, as: :consume_input
 			
 			# Returns 1 if a command is busy, that is, PQgetResult would block waiting for input. A 0 return indicates that PQgetResult can be called with assurance of not blocking.
-			attach_function :is_busy, :PQisBusy, [:pointer], :int
+			ffi_attach_function :PQisBusy, [:pointer], :int, as: :is_busy
 			
-			attach_function :escape_literal, :PQescapeLiteral, [:pointer, :string, :size_t], :pointer
-			attach_function :escape_identifier, :PQescapeIdentifier, [:pointer, :string, :size_t], :pointer
+			ffi_attach_function :PQescapeLiteral, [:pointer, :string, :size_t], :pointer, as: :escape_literal
+			ffi_attach_function :PQescapeIdentifier, [:pointer, :string, :size_t], :pointer, as: :escape_identifier
 			
 			module IO
 				def self.new(fd, mode)
@@ -104,11 +104,16 @@ module DB
 			
 			class Connection < FFI::Pointer
 				def self.connect(wrapper: IO, types: DEFAULT_TYPES, **options)
-					# Prefer 'database' key for 'dbname' parameter:
+					# Postgres expects "dbname" as the key name:
 					if database = options.delete(:database)
 						options[:dbname] = database
 					end
 					
+					# Postgres expects "user" as the key name:
+					if username = options.delete(:username)
+						options[:user] = username
+					end
+					
 					keys = Strings.new(options.keys)
 					values = Strings.new(options.values)
 					

data/lib/db/postgres/native/result.rb

@@ -23,7 +23,7 @@ require_relative '../native'
 module DB
 	module Postgres
 		module Native
-			enum :query_status, [
+			ffi_define_enumeration :query_status, [
 				:empty_query, # empty query string was executed
 				:command_ok, # a query command that doesn't return anything was executed properly by the backend
 				:tuples_ok, #  a query command that returns tuples was executed properly by the backend, PGresult contains the result tuples
@@ -36,22 +36,22 @@ module DB
 				:single_tuple, # single tuple from larger resultset
 			]
 			
-			attach_function :result_status, :PQresultStatus, [:pointer], :query_status
-			attach_function :result_error_message, :PQresultErrorMessage, [:pointer], :string
+			ffi_attach_function :PQresultStatus, [:pointer], :query_status, as: :result_status
+			ffi_attach_function :PQresultErrorMessage, [:pointer], :string, as: :result_error_message
 			
-			attach_function :row_count, :PQntuples, [:pointer], :int
-			attach_function :field_count, :PQnfields, [:pointer], :int
-			attach_function :field_name, :PQfname, [:pointer, :int], :string
-			attach_function :field_type, :PQftype, [:pointer, :int], :int
+			ffi_attach_function :PQntuples, [:pointer], :int, as: :row_count
+			ffi_attach_function :PQnfields, [:pointer], :int, as: :field_count
+			ffi_attach_function :PQfname, [:pointer, :int], :string, as: :field_name
+			ffi_attach_function :PQftype, [:pointer, :int], :int, as: :field_type
 			
-			attach_function :get_value, :PQgetvalue, [:pointer, :int, :int], :string
-			attach_function :get_is_null, :PQgetisnull, [:pointer, :int, :int], :int
+			ffi_attach_function :PQgetvalue, [:pointer, :int, :int], :string, as: :get_value
+			ffi_attach_function :PQgetisnull, [:pointer, :int, :int], :int, as: :get_is_null
 			
-			attach_function :clear, :PQclear, [:pointer], :void
+			ffi_attach_function :PQclear, [:pointer], :void, as: :clear
 			
-			attach_function :put_copy_end, :PQputCopyEnd, [:pointer, :string], :int
-			attach_function :get_copy_data, :PQgetCopyData, [:pointer, :pointer, :int], :int
-			attach_function :free_memory, :PQfreemem, [:pointer], :void
+			ffi_attach_function :PQputCopyEnd, [:pointer, :string], :int, as: :put_copy_end
+			ffi_attach_function :PQgetCopyData, [:pointer, :pointer, :int], :int, as: :get_copy_data
+			ffi_attach_function :PQfreemem, [:pointer], :void, as: :free_memory
 			
 			class Result < FFI::Pointer
 				def initialize(connection, types = {}, address)
@@ -99,17 +99,21 @@ module DB
 					Native.clear(self)
 				end
 				
-				def to_a
-					rows = []
+				def map(&block)
+					results = []
 					
 					self.each do |row|
-						rows << row
+						results << yield row
 					end
 					
-					return rows
+					return results
+				end
+				
+				def to_a
 				end
 				
 			protected
+				
 				def get_value(row, field)
 					if Native.get_is_null(self, row, field) == 0
 						Native.get_value(self, row, field)

data/lib/db/postgres/version.rb

@@ -20,6 +20,6 @@
 
 module DB
 	module Postgres
-		VERSION = "0.2.4"
+		VERSION = "0.5.2"
 	end
 end

metadata

@@ -1,29 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: db-postgres
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.5.2
 platform: ruby
 authors:
 - Samuel Williams
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-12-25 00:00:00.000000000 Z
+date: 2021-04-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: ffi
+  name: ffi-module
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.3.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.3.0
 - !ruby/object:Gem::Dependency
   name: async-io
   requirement: !ruby/object:Gem::Requirement
@@ -158,7 +158,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.2
+rubygems_version: 3.2.3
 signing_key:
 specification_version: 4
 summary: Ruby FFI bindings for libpq C interface.