db-postgres 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 61a45c4212bcf4b8c3fdf1a244292581da1a1df513450d51fcd322ea625ccd94
-  data.tar.gz: 18310b502a0cae91906948fa5ce8c5c07c07e81590612e902588abba7fdd37e1
+  metadata.gz: c30d669b19ff5cf7b41eb45ff896363060472f2de9231c0c3a25236f23560c82
+  data.tar.gz: a3026d300addc7817bf9dbae7124f1c04140a77d8d9a6bd8854ed3741ae7af53
 SHA512:
-  metadata.gz: '039afaa56fac564fbb12e66d164ab0d0700eeb6c22d6bcbb7a56ecbd209da1b065ab16c28c5b14cac513ff0af15d7ece1f549cf4fe055db62e8f742fac4406a2'
-  data.tar.gz: 94e9caed1d36af0a15b9efb71f96a1c151c171fe062bdd789e23c0eab023a2aadd759dbbda996809d93de7fa32323e4fdc67fa52592b7a4e6a1eb784033234bb
+  metadata.gz: 7b0314e22e03bb2579ead32274df8015064d9799f751e02f38e5b9aee1160c92eb20ba490f502a0e39672934d7099a2445584f19cda5c42f08524648c1bda799
+  data.tar.gz: b951e9c36caed76f39f98342cae5eda8383c658eba45d2cab17cfe8aa0a0b6a73a7bbb7d3475746514b09229e68696d4690b475b89e7035d6eb662045a6e7613

data/lib/db/postgres/connection.rb

@@ -76,31 +76,33 @@ module DB
 				return buffer
 			end
 			
+			def id_column(name = 'id', primary_key: true)
+				buffer = String.new
+				
+				append_identifier(name, buffer)
+				
+				buffer << " BIGSERIAL"
+				
+				if primary_key
+					buffer << " PRIMARY KEY"
+				end
+				
+				return buffer
+			end
+			
 			def status
 				@native.status
 			end
 			
 			def send_query(statement)
+				@native.discard_results
+				
 				@native.send_query(statement)
 			end
 			
 			def next_result
 				@native.next_result
 			end
-			
-			def call(statement, streaming: false)
-				@native.send_query(statement)
-				
-				@native.single_row_mode! if streaming
-				
-				last_result = nil
-				
-				while result = @native.next_result
-					last_result = result
-				end
-				
-				return last_result
-			end
 		end
 	end
 end

data/lib/db/postgres/error.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+# Copyright, 2020, by Samuel G. D. Williams. <http://www.codeotaku.com>
+# 
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+module DB
+	module Postgres
+		class Error < StandardError
+		end
+	end
+end

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

@@ -20,6 +20,7 @@
 
 require_relative 'result'
 require_relative 'field'
+require_relative '../error'
 
 module DB
 	module Postgres
@@ -196,6 +197,16 @@ module DB
 				
 				def next_result(types: @types)
 					if result = self.get_result
+						status = Native.result_status(result)
+						
+						if status == :fatal_error
+							message = Native.result_error_message(result)
+							
+							Native.clear(result)
+							
+							raise Error, "Could not get next result: #{message}"
+						end
+						
 						return Result.new(self, types, result)
 					end
 				end
@@ -208,7 +219,7 @@ module DB
 						
 						case status
 						when :copy_in
-							self.put_copy_end("discard results")
+							self.put_copy_end("Discard results")
 						when :copy_out
 							self.flush_copy_out
 						end
@@ -244,7 +255,7 @@ module DB
 						
 						if status == -1
 							message = Native.error_message(self)
-							raise Error.new(message)
+							raise Error, message
 						elsif status == 0
 							@io.wait_writable
 						else
@@ -261,7 +272,7 @@ module DB
 						
 						if status == -2
 							message = Native.error_message(self)
-							raise Error.new(message)
+							raise Error, message
 						elsif status == -1
 							break
 						elsif status == 0
@@ -289,7 +300,7 @@ module DB
 				def check!(result)
 					if result == 0
 						message = Native.error_message(self)
-						raise Error.new(message)
+						raise Error, message
 					end
 				end
 			end

data/lib/db/postgres/version.rb

@@ -20,6 +20,6 @@
 
 module DB
 	module Postgres
-		VERSION = "0.1.1"
+		VERSION = "0.1.2"
 	end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: db-postgres
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Samuel Williams
@@ -117,6 +117,7 @@ files:
 - lib/db/postgres.rb
 - lib/db/postgres/adapter.rb
 - lib/db/postgres/connection.rb
+- lib/db/postgres/error.rb
 - lib/db/postgres/native.rb
 - lib/db/postgres/native/connection.rb
 - lib/db/postgres/native/field.rb