cipherstash-pg 1.0.0.beta.1-x86_64-linux → 1.0.0.beta.3-x86_64-linux

data/sample/issue-119.rb

@@ -1,94 +1,46 @@
-# -*- ruby -*-
-
-require 'pg'
-
-# This is another example of how to use COPY FROM, this time as a
-# minimal test case used to try to figure out what was going on in
-# an issue submitted from a user:
-#
-#  https://bitbucket.org/ged/ruby-pg/issue/119
-#
-
-conn		   = PG.connect( dbname: 'test' )
-table_name	   = 'issue_119'
-field_list	   = %w[name body_weight brain_weight]
-method		   = 0
-options		   = { truncate: true }
-sql_parameters = ''
-
-conn.set_error_verbosity( PG::PQERRORS_VERBOSE )
-conn.exec( "DROP TABLE IF EXISTS #{table_name}" )
-conn.exec( "CREATE TABLE #{table_name} ( id SERIAL, name TEXT, body_weight REAL, brain_weight REAL )" )
-
-text = <<-END_DATA
-Mountain beaver	1.35	465
-Cow	465	423
-Grey wolf	36.33	119.5
-Goat	27.66	115
-Guinea pig	1.04	5.5
-Dipliodocus	11700	50
-Asian elephant	2547	4603
-Donkey	187.1	419
-Horse	521	655
-Potar monkey	10	115
-Cat	3.3	25.6
-Giraffe	529	680
-Gorilla	207	406
-Human	62	1320
-African elephant	6654	5712
-Triceratops	9400	70
-Rhesus monkey	6.8	179
-Kangaroo	35	56
-Golden hamster	0.12	1
-Mouse	0.023	0.4
-Rabbit	2.5	12.1
-Sheep	55.5	175
-Jaguar	100	157
-Chimpanzee	52.16	440
-Brachiosaurus	87000	154.5
-Mole	0.122	3
-Pig	192	18
-END_DATA
-
-#ActiveRecord::Base.connection_pool.with_connection do |conn|
-	conn.transaction do
-		rc = conn #.raw_connection
-		rc.exec "TRUNCATE TABLE #{table_name};" if options[:truncate]
-		sql = "COPY #{table_name} (#{field_list.join(',')}) FROM STDIN #{sql_parameters} "
-		p sql
-		rc.exec(sql)
-		errmsg = nil # scope this outside of the rescue below so it's visible later
-		begin
-			if method == 1
-				rc.put_copy_data text + "\\.\n"
-			else
-				text.each_line { |line| rc.put_copy_data(line) }
-			end
-		rescue Errno => err
-			errmsg = "%s while reading copy data: %s" % [err.class.name, err.message]
-			puts "an error occurred"
-		end
-
-		if errmsg
-			rc.put_copy_end(errmsg)
-			puts "ERROR #{errmsg}"
-		else
-			rc.put_copy_end
-		end
-
-		while res = rc.get_result
-			st = res.res_status( res.result_status )
-			puts "Result of COPY is: %s" % [ st ]
-			if res.result_status != PG::PGRES_COPY_IN
-				puts res.error_message
-			end
-		end
-		puts "end"
-	end #transaction
-#end #connection
-
-conn.exec( "SELECT name, brain_weight FROM #{table_name}" ) do |res|
-	p res.values
+require("cipherstash-pg")
+conn = CipherStashPG.connect(:dbname => "test")
+table_name = "issue_119"
+field_list = ["name", "body_weight", "brain_weight"]
+method = 0
+options = { :truncate => true }
+sql_parameters = ""
+conn.set_error_verbosity(CipherStashPG::PQERRORS_VERBOSE)
+conn.exec("DROP TABLE IF EXISTS #{table_name}")
+conn.exec("CREATE TABLE #{table_name} ( id SERIAL, name TEXT, body_weight REAL, brain_weight REAL )")
+text = "Mountain beaver\t1.35\t465\nCow\t465\t423\nGrey wolf\t36.33\t119.5\nGoat\t27.66\t115\nGuinea pig\t1.04\t5.5\nDipliodocus\t11700\t50\nAsian elephant\t2547\t4603\nDonkey\t187.1\t419\nHorse\t521\t655\nPotar monkey\t10\t115\nCat\t3.3\t25.6\nGiraffe\t529\t680\nGorilla\t207\t406\nHuman\t62\t1320\nAfrican elephant\t6654\t5712\nTriceratops\t9400\t70\nRhesus monkey\t6.8\t179\nKangaroo\t35\t56\nGolden hamster\t0.12\t1\nMouse\t0.023\t0.4\nRabbit\t2.5\t12.1\nSheep\t55.5\t175\nJaguar\t100\t157\nChimpanzee\t52.16\t440\nBrachiosaurus\t87000\t154.5\nMole\t0.122\t3\nPig\t192\t18\n"
+conn.transaction do
+  rc = conn
+  rc.exec("TRUNCATE TABLE #{table_name};") if options[:truncate]
+  sql = "COPY #{table_name} (#{field_list.join(",")}) FROM STDIN #{sql_parameters} "
+  p(sql)
+  rc.exec(sql)
+  errmsg = nil
+  begin
+    if (method == 1) then
+      rc.put_copy_data((text + "\\.\n"))
+    else
+      text.each_line { |line| rc.put_copy_data(line) }
+    end
+  rescue Errno => err
+    errmsg = ("%s while reading copy data: %s" % [err.class.name, err.message])
+    puts("an error occurred")
+  end
+  if errmsg then
+    rc.put_copy_end(errmsg)
+    puts("ERROR #{errmsg}")
+  else
+    rc.put_copy_end
+  end
+  while res = rc.get_result do
+    st = res.res_status(res.result_status)
+    puts(("Result of COPY is: %s" % [st]))
+    if (res.result_status != CipherStashPG::PGRES_COPY_IN) then
+      puts(res.error_message)
+    end
+  end
+  puts("end")
+end
+conn.exec("SELECT name, brain_weight FROM #{table_name}") do |res|
+  p(res.values)
 end
-
-

data/sample/losample.rb

@@ -1,69 +1,51 @@
-# -*- ruby -*-
-
-require 'pg'
-
-SAMPLE_WRITE_DATA = 'some sample data'
-SAMPLE_EXPORT_NAME = 'lowrite.txt'
-
-conn = PG.connect( :dbname => 'test', :host => 'localhost', :port => 5432 )
-puts "dbname: " + conn.db + "\thost: " + conn.host + "\tuser: " + conn.user
-
-# Start a transaction, as all large object functions require one.
-puts "Beginning transaction"
-conn.exec( 'BEGIN' )
-
-# Test importing from a file
-puts "Import test:"
-puts "  importing %s" % [ __FILE__ ]
-oid = conn.lo_import( __FILE__ )
-puts "  imported as large object %d" % [ oid ]
-
-# Read back 50 bytes of the imported data
-puts "Read test:"
-fd = conn.lo_open( oid, PG::INV_READ|PG::INV_WRITE )
-conn.lo_lseek( fd, 0, PG::SEEK_SET )
-buf = conn.lo_read( fd, 50 )
-puts "  read: %p" % [ buf ]
-puts "  read was ok!" if buf =~ /require 'pg'/
-
-# Append some test data onto the end of the object
-puts "Write test:"
-conn.lo_lseek( fd, 0, PG::SEEK_END )
+require("cipherstash-pg")
+SAMPLE_WRITE_DATA = "some sample data"
+SAMPLE_EXPORT_NAME = "lowrite.txt"
+conn = CipherStashPG.connect(:dbname => "test", :host => "localhost", :port => 5432)
+puts(((((("dbname: " + conn.db) + "\thost: ") + conn.host) + "\tuser: ") + conn.user))
+puts("Beginning transaction")
+conn.exec("BEGIN")
+puts("Import test:")
+puts(("  importing %s" % ["(string)"]))
+oid = conn.lo_import("(string)")
+puts(("  imported as large object %d" % [oid]))
+puts("Read test:")
+fd = conn.lo_open(oid, (CipherStashPG::INV_READ | CipherStashPG::INV_WRITE))
+conn.lo_lseek(fd, 0, CipherStashPG::SEEK_SET)
+buf = conn.lo_read(fd, 50)
+puts(("  read: %p" % [buf]))
+if buf =~ /require 'pg'/ then
+  puts("  read was ok!")
+end
+puts("Write test:")
+conn.lo_lseek(fd, 0, CipherStashPG::SEEK_END)
 buf = SAMPLE_WRITE_DATA.dup
 totalbytes = 0
-until buf.empty?
-	bytes = conn.lo_write( fd, buf )
-	buf.slice!( 0, bytes )
-	totalbytes += bytes
+until buf.empty? do
+  (bytes = conn.lo_write(fd, buf)
+  buf.slice!(0, bytes)
+  totalbytes = (totalbytes + bytes))
 end
-puts "  appended %d bytes" % [ totalbytes ]
-
-# Now export it
-puts "Export test:"
-File.unlink( SAMPLE_EXPORT_NAME ) if File.exist?( SAMPLE_EXPORT_NAME )
-conn.lo_export( oid, SAMPLE_EXPORT_NAME )
-puts "  success!" if File.exist?( SAMPLE_EXPORT_NAME )
-puts "  exported as %s (%d bytes)" % [ SAMPLE_EXPORT_NAME, File.size(SAMPLE_EXPORT_NAME) ]
-
-conn.exec( 'COMMIT' )
-puts "End of transaction."
-
-
-puts 'Testing read and delete from a new transaction:'
-puts '  starting a new transaction'
-conn.exec( 'BEGIN' )
-
-fd = conn.lo_open( oid, PG::INV_READ )
-puts '  reopened okay.'
-conn.lo_lseek( fd, 50, PG::SEEK_END )
-buf = conn.lo_read( fd, 50 )
-puts '  read okay.' if buf == SAMPLE_WRITE_DATA
-
-puts 'Closing and unlinking:'
-conn.lo_close( fd )
-puts '  closed.'
-conn.lo_unlink( oid )
-puts '  unlinked.'
-conn.exec( 'COMMIT' )
-puts 'Done.'
-
+puts(("  appended %d bytes" % [totalbytes]))
+puts("Export test:")
+File.unlink(SAMPLE_EXPORT_NAME) if File.exist?(SAMPLE_EXPORT_NAME)
+conn.lo_export(oid, SAMPLE_EXPORT_NAME)
+puts("  success!") if File.exist?(SAMPLE_EXPORT_NAME)
+puts(("  exported as %s (%d bytes)" % [SAMPLE_EXPORT_NAME, File.size(SAMPLE_EXPORT_NAME)]))
+conn.exec("COMMIT")
+puts("End of transaction.")
+puts("Testing read and delete from a new transaction:")
+puts("  starting a new transaction")
+conn.exec("BEGIN")
+fd = conn.lo_open(oid, CipherStashPG::INV_READ)
+puts("  reopened okay.")
+conn.lo_lseek(fd, 50, CipherStashPG::SEEK_END)
+buf = conn.lo_read(fd, 50)
+puts("  read okay.") if (buf == SAMPLE_WRITE_DATA)
+puts("Closing and unlinking:")
+conn.lo_close(fd)
+puts("  closed.")
+conn.lo_unlink(oid)
+puts("  unlinked.")
+conn.exec("COMMIT")
+puts("Done.")

data/sample/minimal-testcase.rb

@@ -1,17 +1,6 @@
-# -*- ruby -*-
-
-require 'pg'
-
-conn = PG.connect( :dbname => 'test' )
-$stderr.puts '---',
-	RUBY_DESCRIPTION,
-	PG.version_string( true ),
-	"Server version: #{conn.server_version}",
-	"Client version: #{PG.library_version}",
-	'---'
-
-result = conn.exec( "SELECT * from pg_stat_activity" )
-
-$stderr.puts %Q{Expected this to return: ["select * from pg_stat_activity"]}
-p result.field_values( 'current_query' )
-
+require("cipherstash-pg")
+conn = CipherStashPG.connect(:dbname => "test")
+$stderr.puts("---", RUBY_DESCRIPTION, CipherStashPG.version_string(true), "Server version: #{conn.server_version}", "Client version: #{CipherStashPG.library_version}", "---")
+result = conn.exec("SELECT * from pg_stat_activity")
+$stderr.puts("Expected this to return: [\"select * from pg_stat_activity\"]")
+p(result.field_values("current_query"))

data/sample/notify_wait.rb

@@ -1,72 +1,26 @@
-# -*- ruby -*-
-#
-# Test script, demonstrating a non-poll notification for a table event.
-#
-
 BEGIN {
-        require 'pathname'
-        basedir = Pathname.new( __FILE__ ).expand_path.dirname.parent
-        libdir = basedir + 'lib'
-        $LOAD_PATH.unshift( libdir.to_s ) unless $LOAD_PATH.include?( libdir.to_s )
+  require("pathname")
+  basedir = Pathname.new("(string)").expand_path.dirname.parent
+  libdir = (basedir + "lib")
+  $LOAD_PATH.unshift(libdir.to_s) unless $LOAD_PATH.include?(libdir.to_s)
 }
-
-require 'pg'
-
-TRIGGER_TABLE = %{
-	CREATE TABLE IF NOT EXISTS test ( message text );
-}
-
-TRIGGER_FUNCTION = %{
-CREATE OR REPLACE FUNCTION notify_test()
-RETURNS TRIGGER
-LANGUAGE plpgsql
-AS $$
-    BEGIN
-        NOTIFY woo;
-        RETURN NULL;
-    END
-$$
-}
-
-DROP_TRIGGER = %{
-DROP TRIGGER IF EXISTS notify_trigger ON test
-}
-
-
-TRIGGER = %{
-CREATE TRIGGER notify_trigger
-AFTER UPDATE OR INSERT OR DELETE
-ON test
-FOR EACH STATEMENT
-EXECUTE PROCEDURE notify_test();
-}
-
-conn = PG.connect( :dbname => 'test' )
-
-conn.exec( TRIGGER_TABLE )
-conn.exec( TRIGGER_FUNCTION )
-conn.exec( DROP_TRIGGER )
-conn.exec( TRIGGER )
-
-conn.exec( 'LISTEN woo' )  # register interest in the 'woo' event
-
+require("cipherstash-pg")
+TRIGGER_TABLE = "\n\tCREATE TABLE IF NOT EXISTS test ( message text );\n"
+TRIGGER_FUNCTION = "\nCREATE OR REPLACE FUNCTION notify_test()\nRETURNS TRIGGER\nLANGUAGE plpgsql\nAS $$\n    BEGIN\n        NOTIFY woo;\n        RETURN NULL;\n    END\n$$\n"
+DROP_TRIGGER = "\nDROP TRIGGER IF EXISTS notify_trigger ON test\n"
+TRIGGER = "\nCREATE TRIGGER notify_trigger\nAFTER UPDATE OR INSERT OR DELETE\nON test\nFOR EACH STATEMENT\nEXECUTE PROCEDURE notify_test();\n"
+conn = CipherStashPG.connect(:dbname => "test")
+conn.exec(TRIGGER_TABLE)
+conn.exec(TRIGGER_FUNCTION)
+conn.exec(DROP_TRIGGER)
+conn.exec(TRIGGER)
+conn.exec("LISTEN woo")
 notifications = []
-
-puts "Now switch to a different term and run:",
-     '',
-     %{  psql test -c "insert into test values ('A message.')"},
-	 ''
-
-puts "Waiting up to 30 seconds for for an event!"
-conn.wait_for_notify( 30 ) do |notify, pid|
-	notifications << [ pid, notify ]
-end
-
-if notifications.empty?
-	puts "Awww, I didn't see any events."
+puts("Now switch to a different term and run:", "", "  psql test -c \"insert into test values ('A message.')\"", "")
+puts("Waiting up to 30 seconds for for an event!")
+conn.wait_for_notify(30) { |notify, pid| (notifications << [pid, notify]) }
+if notifications.empty? then
+  puts("Awww, I didn't see any events.")
 else
-	puts "I got one from pid %d: %s" % notifications.first
+  puts(("I got one from pid %d: %s" % notifications.first))
 end
-
-
-