ahamid-postgres-pr 0.6.1

data/lib/postgres-pr/postgres-compat.rb

@@ -0,0 +1,146 @@
+# This is a compatibility layer for using the pure Ruby postgres-pr instead of
+# the C interface of postgres.
+
+require 'postgres-pr/connection'
+
+class PGconn
+  class << self
+    alias connect new
+
+    def quote_ident(name)
+      %("#{name}")
+    end
+  end
+
+  def initialize(host, port, options, tty, database, user, auth)
+    uri =
+    if host.nil?
+      nil
+    elsif host[0] != ?/
+      "tcp://#{ host }:#{ port }"
+    else
+      "unix:#{ host }/.s.PGSQL.#{ port }"
+    end
+    @host = host
+    @db = database
+    @user = user
+    @conn = PostgresPR::Connection.new(database, user, auth, uri)
+  end
+
+
+  def close
+    @conn.close
+  end
+
+  attr_reader :host, :db, :user
+
+  def query(sql)
+    PGresult.new(@conn.query(sql))
+  end
+
+  alias exec query
+
+  def transaction_status
+    @conn.transaction_status
+  end
+
+  def self.escape(str)
+    str.gsub("'","''").gsub("\\", "\\\\\\\\")
+  end
+
+end
+
+class PGresult
+  include Enumerable
+
+  EMPTY_QUERY = 0
+  COMMAND_OK = 1
+  TUPLES_OK = 2
+  COPY_OUT = 3
+  COPY_IN = 4
+  BAD_RESPONSE = 5
+  NONFATAL_ERROR = 6
+  FATAL_ERROR = 7
+
+  def each(&block)
+    @result.each(&block)
+  end
+
+  def [](index)
+    @result[index]
+  end
+ 
+  def initialize(res)
+    @res = res
+    @fields = @res.fields.map {|f| f.name}
+    @result = @res.rows
+  end
+
+  # TODO: status, getlength, cmdstatus
+
+  attr_reader :result, :fields
+
+  def num_tuples
+    @result.size
+  end
+
+  def num_fields
+    @fields.size
+  end
+
+  def fieldname(index)
+    @fields[index]
+  end
+
+  def fieldnum(name)
+    @fields.index(name)
+  end
+
+  def type(index)
+    # TODO: correct?
+    @res.fields[index].type_oid
+  end
+
+  def size(index)
+    raise
+    # TODO: correct?
+    @res.fields[index].typlen
+  end
+
+  def getvalue(tup_num, field_num)
+    @result[tup_num][field_num]
+  end
+
+  def status
+    if num_tuples > 0
+      TUPLES_OK
+    else
+      COMMAND_OK
+    end
+  end
+
+  def cmdstatus
+    @res.cmd_tag || ''
+  end
+
+  # free the result set
+  def clear
+    @res = @fields = @result = nil
+  end
+
+  # Returns the number of rows affected by the SQL command
+  def cmdtuples
+    case @res.cmd_tag
+    when nil 
+      return nil
+    when /^INSERT\s+(\d+)\s+(\d+)$/, /^(DELETE|UPDATE|MOVE|FETCH)\s+(\d+)$/
+      $2.to_i
+    else
+      nil
+    end
+  end
+
+end
+
+class PGError < Exception
+end

data/lib/postgres-pr/typeconv/TC_conv.rb

@@ -0,0 +1,18 @@
+require 'test/unit'
+require 'conv'
+require 'array'
+require 'bytea'
+
+class TC_Conversion < Test::Unit::TestCase
+  def test_decode_array
+    assert_equal ["abcdef ", "hallo", ["1", "2"]], decode_array("{   abcdef ,    hallo, {   1, 2} }")
+    assert_equal [""], decode_array("{ }") # TODO: Correct?
+    assert_equal [], decode_array("{}")
+    assert_equal ["hallo", ""], decode_array("{hallo,}")
+  end
+
+  def test_bytea
+  end
+
+  include Postgres::Conversion
+end

data/lib/postgres-pr/typeconv/array.rb

@@ -0,0 +1,46 @@
+require 'strscan'
+
+module Postgres::Conversion
+
+  def decode_array(str, delim=',', &conv_proc)
+    delim = Regexp.escape(delim)
+    buf = StringScanner.new(str)
+    return parse_arr(buf, delim, &conv_proc)
+  ensure
+    raise ConversionError, "end of string expected (#{buf.rest})" unless buf.empty?
+  end
+
+  private
+
+  def parse_arr(buf, delim, &conv_proc)
+    # skip whitespace
+    buf.skip(/\s*/)       
+
+    raise ConversionError, "'{' expected" unless buf.get_byte == '{'
+
+    elems = []
+    unless buf.scan(/\}/) # array is not empty
+      loop do
+        # skip whitespace
+        buf.skip(/\s+/)   
+
+        elems <<
+        if buf.check(/\{/)
+          parse_arr(buf, delim, &conv_proc)
+        else
+          e = buf.scan(/("((\\.)|[^"])*"|\\.|[^\}#{ delim }])*/) || raise(ConversionError)
+          if conv_proc then conv_proc.call(e) else e end
+        end
+
+        break if buf.scan(/\}/)
+        break unless buf.scan(/#{ delim }/)
+      end
+    end
+
+    # skip whitespace
+    buf.skip(/\s*/)
+
+    elems
+  end
+
+end

data/lib/postgres-pr/typeconv/bytea.rb

@@ -0,0 +1,26 @@
+module Postgres::Conversion
+
+  #
+  # Encodes a string as bytea value.
+  #
+  # for encoding rules see:
+  #   http://www.postgresql.org/docs/7.4/static/datatype-binary.html
+  #
+
+  def encode_bytea(str)
+    str.gsub(/[\000-\037\047\134\177-\377]/) {|b| "\\#{ b[0].to_s(8).rjust(3, '0') }" }
+  end
+
+  #
+  # Decodes a bytea encoded string.
+  #
+  # for decoding rules see:
+  #   http://www.postgresql.org/docs/7.4/static/datatype-binary.html
+  #
+  def decode_bytea(str)
+    str.gsub(/\\(\\|'|[0-3][0-7][0-7])/) {|s|
+      if s.size == 2 then s[1,1] else s[1,3].oct.chr end
+    }
+  end
+
+end

data/lib/postgres-pr/typeconv/conv.rb

@@ -0,0 +1,5 @@
+module Postgres
+  module Conversion
+    class ConversionError < Exception; end
+  end
+end

data/lib/postgres-pr/version.rb

@@ -0,0 +1,3 @@
+module PostgresPR
+  Version = "0.6.1"
+end

data/lib/postgres.rb

@@ -0,0 +1,8 @@
+# This is a compatibility layer for using the pure Ruby postgres-pr instead of
+# the C interface of postgres.
+
+begin
+  require 'postgres.so'
+rescue LoadError
+  require 'postgres-pr/postgres-compat'
+end

data/test/TC_message.rb

@@ -0,0 +1,103 @@
+require 'test/unit'
+require 'stringio'
+
+class Module
+  def attr_accessor(*attrs)
+    @@attrs = [] unless defined?(@@attrs)
+    @@attrs += attrs
+
+    x = @@attrs.map {|a| "self.#{a} == o.#{a}"}.join(" && ")
+    class_eval %{
+      def ==(o)
+        #{ x }
+      end
+    }
+
+    @@attrs.each do |a|
+      class_eval %{
+        def #{a}() @#{a} end
+        def #{a}=(v) @#{a}=v end
+      }
+    end
+  end
+end
+
+$LOAD_PATH.unshift '../lib'
+require 'postgres-pr/message'
+include PostgresPR
+
+class Buffer
+  alias old_content content
+  def content
+    self
+  end
+end
+
+class Message
+  attr_accessor :buffer
+
+  class << self
+    alias old_create create
+    def create(buffer)
+      obj = old_create(buffer)
+      obj.buffer = buffer
+      obj
+    end
+  end
+
+  alias old_dump dump
+
+  def dump(body_size=0, &block)
+    buf = old_dump(body_size, &block)
+    self.buffer = buf
+    buf.old_content
+  end
+end
+
+class StartupMessage
+  alias old_dump dump
+  def dump
+    buf = old_dump
+    self.buffer = buf
+    buf.old_content
+  end
+end
+
+class StringIO
+  alias readbytes read
+end
+
+class TC_Message < Test::Unit::TestCase
+
+  CASES = [ 
+    #[AuthentificationOk], 
+    #[ErrorResponse],
+    [ParameterStatus, "key", "value"],
+    [BackendKeyData, 234234234, 213434],
+    [ReadyForQuery, ?T],
+    # TODO: RowDescription
+    [DataRow, ["a", "bbbbbb", "ccc", nil, nil, "ddddd", "e" * 10_000]],
+    [DataRow, []],
+    [CommandComplete, "INSERT"],
+    [StartupMessage, 196608, {"user" => "mneumann", "database" => "mneumann"}],
+    [Parse, "INSERT INTO blah values (?, ?)", ""],
+    [Query, "SELECT * FROM test\nWHERE a='test'"]
+  ]
+
+  def test_pack_unpack_feature
+    assert_equal ['a', 'b'], "a\000b\000".unpack('Z*Z*')
+  end
+
+  def test_marshal_unmarshal
+    CASES.each do |klass, *params|
+      msg = klass.new(*params)
+      new_msg = Message.read(StringIO.new(msg.dump), klass == StartupMessage)
+      assert_equal(msg, new_msg)
+
+      msg1, msg2 = klass.new(*params), klass.new(*params)
+      msg1.dump
+      msg2.dump; msg2.parse(msg2.buffer)
+      assert_equal(msg1, msg2)
+    end
+  end
+end

metadata

@@ -0,0 +1,72 @@
+--- !ruby/object:Gem::Specification 
+name: ahamid-postgres-pr
+version: !ruby/object:Gem::Version 
+  version: 0.6.1
+platform: ruby
+authors: 
+- Michael Neumann
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-10-15 00:00:00 -04:00
+default_executable: 
+dependencies: []
+
+description: 
+email: mneumann@ntecs.de
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/binary_reader.rb
+- lib/binary_writer.rb
+- lib/buffer.rb
+- lib/byteorder.rb
+- lib/postgres-pr/connection.rb
+- lib/postgres-pr/message.rb
+- lib/postgres-pr/postgres-compat.rb
+- lib/postgres-pr/typeconv/array.rb
+- lib/postgres-pr/typeconv/bytea.rb
+- lib/postgres-pr/typeconv/conv.rb
+- lib/postgres-pr/typeconv/TC_conv.rb
+- lib/postgres-pr/version.rb
+- lib/postgres.rb
+- test/TC_message.rb
+- examples/client.rb
+- examples/og/test.rb
+- examples/server.rb
+- examples/test_connection.rb
+has_rdoc: true
+homepage: http://github.com/ahamid/postgres-pr
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: 
+- PostgreSQL >= 7.4
+rubyforge_project: postgres-pr
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: A pure Ruby interface to the PostgreSQL (>= 7.4) database
+test_files: []
+