do_postgres 0.10.0-java → 0.10.1-java

data/ChangeLog.markdown

@@ -0,0 +1,22 @@
+## 0.10.1 (unreleased, in git)
+
+* Support for Ruby 1.8 and 1.9 on Windows.
+* Switch to Jeweler for Gem building tasks (this change may be temporary).
+* Switch to using Bacon for running specs: This should make specs friendlier to
+  new Ruby implementations that are not yet 100% MRI-compatible, and in turn,
+  prepared the road for our own IronRuby and MacRuby support.
+* Switch to the newly added rake-compiler `JavaExtensionTask` for compiling
+  JRuby extensions, instead of our (broken) home-grown solution.
+
+## 0.9.12 2009-05-17
+* Improvements
+  * Windows support
+
+## 0.9.11 2009-01-19
+* Improvements
+  * Ruby 1.9 support
+* Fixes
+  * Fix build issue on certain platforms introduces with 0.9.10
+
+## 0.9.9 2008-11-27
+* No changes since 0.9.8

data/LICENSE

@@ -1,4 +1,4 @@
-Copyright (c) 2007, 2008, 2009 Yehuda Katz, Dirkjan Bussink
+Copyright (c) 2007 - 2010 Yehuda Katz, Dirkjan Bussink
 
 Permission is hereby granted, free of charge, to any person obtaining
 a copy of this software and associated documentation files (the

data/README.markdown

@@ -1,4 +1,107 @@
-do_postgres
-===========
+# do_postgres
 
-A PostgreSQL driver for DataObjects
+* <http://dataobjects.info>
+
+## Description
+
+A PostgreSQL driver for DataObjects.
+
+## Features/Problems
+
+This driver implements the DataObjects API for the PostgreSQL relational database.
+
+## Synopsis
+
+An example of usage:
+
+    # default user (postgres, postgres), default port (5432)
+    DataObjects::Connection.new("postgres://host/database")
+    # specified user, specified port
+    DataObjects::Connection.new("postgres://user:pass@host:8888/database")
+
+    @connection = DataObjects::Connection.new("postgres://localhost/employees")
+    @reader = @connection.create_command('SELECT * FROM users').execute_reader
+    @reader.next!
+
+In the future, the `Connection` constructor will be able to be passed either a
+DataObjects-style URL or JDBC style URL, when using do\_postgres on JRuby.
+However, this feature is not currently working reliably and is a known issue.
+
+## Requirements
+
+This driver is provided for the following platforms:
+ * Ruby MRI (1.8.6/7), 1.9: tested on Linux, Mac OS X and Windows platforms.
+ * JRuby 1.3.1 + (1.4+ recommended).
+ * Rubinius (experimental).
+
+Additionally you should have the following prerequisites:
+ * `data_objects` gem
+ * `do_jdbc` gem (shared library), if running on JRuby.
+
+## Install
+
+To install the gem:
+
+    gem install do_postgres
+
+To compile and install from source:
+
+* Install rake-compiler: `gem install rake-compiler`.
+
+* For MRI/Rubinius extensions:
+  * Install the `gcc` compiler. On OS X, you should install XCode tools. On
+    Ubuntu, run `apt-get install build-essential`.
+  * Install Ruby and PostgreSQL client.
+  * Install the Ruby and PostgreSQL development headers.
+    * On Debian-Linux distributions, you can install the following packages
+      with `apt`: `ruby-dev` `libpostgresql-dev`.
+  * If you want to cross-compile for Windows:
+    * Install MinGW:
+      * On Debian-Linux distributions, you can install the following package
+        with `apt`: `mingw32`.
+      * On OS X, this can install the following package with MacPorts: `i386-mingw32-gcc`.
+    * Run `rake-compiler cross-ruby`.
+    * Run `rake-compiler update-config`.
+
+ * For JRuby extensions:
+   * Install the Java Development Kit (provided if you are
+     on a recent version of Mac OS X) from <http://java.sun.com>.
+   * Install a recent version of JRuby. Ensure `jruby` is in your `PATH` and/or
+     you have configured the `JRUBY_HOME` environment variable to point to your
+     JRuby installation.
+   * Install `data_objects` and `do_jdbc` with `jruby -S rake install`.
+
+ * Then, install this driver with `(jruby -S) rake install`.
+
+For more information, see the PostgreSQL driver wiki page:
+<http://wiki.github.com/datamapper/do/postgresql>.
+
+## Developers
+
+Follow the above installation instructions. Additionally, you'll need:
+  * `bacon` gem for running specs.
+  * `YARD` gem for generating documentation.
+
+See the DataObjects wiki for more comprehensive information on installing and
+contributing to the JRuby-variant of this driver:
+<http://wiki.github.com/datamapper/do/jruby>.
+
+To run specs:
+
+    rake spec
+
+To run specs without compiling extensions first:
+
+    rake spec_no_compile
+
+To run individual specs:
+
+    rake spec TEST=spec/connection_spec.rb
+
+(Note that the `rake` task uses a `TEST` parameter, not `SPEC`. This is because
+the `Rake::TestTask` is used for executing the Bacon specs).
+
+## License
+
+This code is licensed under an **MIT (X11) License**. Please see the
+accompanying `LICENSE` file.

data/Rakefile

@@ -1,16 +1,63 @@
+require 'pathname'
 require 'rubygems'
 require 'rake'
 require 'rake/clean'
 
-require 'pathname'
-require 'lib/do_postgres/version'
+ROOT = Pathname(__FILE__).dirname.expand_path
+
+require ROOT + 'lib/do_postgres/version'
+
+JRUBY    = RUBY_PLATFORM =~ /java/
+IRONRUBY = defined?(RUBY_ENGINE) && RUBY_ENGINE == 'ironruby'
+WINDOWS  = Gem.win_platform? || (JRUBY && ENV_JAVA['os.name'] =~ /windows/i)
+SUDO     = WINDOWS ? '' : ('sudo' unless ENV['SUDOLESS'])
+BINARY_VERSION = '8.3.9'
+
+CLEAN.include(%w[ {tmp,pkg}/ **/*.{o,so,bundle,jar,log,a,gem,dSYM,obj,pdb,exp,DS_Store,rbc,db} ext/do_postgres/Makefile ext-java/target ])
+
+begin
+  gem 'jeweler', '~> 1.4'
+  require 'jeweler'
+
+  Jeweler::Tasks.new do |gem|
+    gem.name        = 'do_postgres'
+    gem.version     = DataObjects::Postgres::VERSION
+    gem.summary     = 'DataObjects PostgreSQL Driver'
+    gem.description = 'Implements the DataObjects API for PostgreSQL'
+    gem.platform    = Gem::Platform::RUBY
+    gem.files       = Dir['lib/**/*.rb', 'spec/**/*.rb', 'tasks/**/*.rake',
+                          'ext/**/*.{rb,c,h}', 'LICENSE', 'Rakefile',
+                          '*.{markdown,rdoc,txt,yml}']
+    gem.extra_rdoc_files = FileList['README*', 'ChangeLog*', 'LICENSE']
+    gem.test_files  = FileList['spec/**/*.rb']
+
+    # rake-compiler should generate gemspecs for other platforms (e.g. 'java')
+    # and modify dependencies and extensions appropriately
+    gem.extensions << 'ext/do_postgres/extconf.rb'
+
+    gem.add_dependency 'data_objects', DataObjects::Postgres::VERSION
+
+    gem.add_development_dependency 'bacon',         '~>1.1'
+    gem.add_development_dependency 'rake-compiler', '~>0.7'
+
+    gem.has_rdoc = false
+    gem.rubyforge_project = 'dorb'
+    gem.authors     = [ 'Dirkjan Bussink' ]
+    gem.email       = 'd.bussink@gmail.com'
+  end
 
-ROOT    = Pathname(__FILE__).dirname.expand_path
-JRUBY   = RUBY_PLATFORM =~ /java/
-WINDOWS = Gem.win_platform?
-SUDO    = WINDOWS ? '' : ('sudo' unless ENV['SUDOLESS'])
-BINARY_VERSION = '5.0.77'
+  if JRUBY
+    Rake::Task['build'].clear_actions   if Rake::Task.task_defined?('build')
+    Rake::Task['install'].clear_actions if Rake::Task.task_defined?('install')
+    task :build => [ :java, :gem ]
+    task :install do
+      sh "#{Config::CONFIG['RUBY_INSTALL_NAME']} -S gem install pkg/do_postgres-#{DataObjects::Postgres::VERSION}-java.gem"
+    end
+  end
 
-Dir['tasks/*.rake'].sort.each { |f| import f }
+  Jeweler::GemcutterTasks.new
 
-CLEAN.include(%w[ {tmp,pkg}/ **/*.{o,so,bundle,jar,log,a,gem,dSYM,obj,pdb,exp,DS_Store,rbc,db} ext/do_postgres_ext/Makefile ext-java/target ])
+  FileList['tasks/**/*.rake'].each { |task| import task }
+rescue LoadError
+  puts 'Jeweler (or a dependency) not available. Install it with: gem install jeweler'
+end

data/ext/do_postgres/compat.h

@@ -0,0 +1,55 @@
+#ifndef RUBY_COMPAT_H
+#define RUBY_COMPAT_H
+
+/*
+ * Rules for better ruby C extensions:
+ *
+ * Never use the R<TYPE> macros directly, always use R<TYPE>_<FIELD>
+ *
+ * Never compare with RBASIC(obj)->klass, always use
+ *   rb_obj_is_instance_of()
+ *
+ * Never use RHASH(obj)->tbl or RHASH_TBL().
+ *
+ */
+
+
+// Array
+#ifndef RARRAY_PTR
+#define RARRAY_PTR(obj) RARRAY(obj)->ptr
+#endif
+
+#ifndef RARRAY_LEN
+#define RARRAY_LEN(obj) RARRAY(obj)->len
+#endif
+
+// String
+#ifndef RSTRING_PTR
+#define RSTRING_PTR(obj) RSTRING(obj)->ptr
+#endif
+
+#ifndef RSTRING_LEN
+#define RSTRING_LEN(obj) RSTRING(obj)->len
+#endif
+
+#ifndef rb_str_ptr
+#define rb_str_ptr(str) RSTRING_PTR(str)
+#endif
+
+#ifndef rb_str_ptr_readonly
+#define rb_str_ptr_readonly(str) RSTRING_PTR(str)
+#endif
+
+#ifndef rb_str_flush
+#define rb_str_flush(str)
+#endif
+
+#ifndef rb_str_update
+#define rb_str_update(str)
+#endif
+
+#ifndef rb_str_len
+#define rb_str_len(str) RSTRING_LEN(str)
+#endif
+
+#endif

data/ext/do_postgres/do_postgres.c

@@ -0,0 +1,1060 @@
+#include <libpq-fe.h>
+#include <postgres.h>
+#include <mb/pg_wchar.h>
+#include <catalog/pg_type.h>
+#include <utils/errcodes.h>
+
+/* Undefine constants Postgres also defines */
+#undef PACKAGE_BUGREPORT
+#undef PACKAGE_NAME
+#undef PACKAGE_STRING
+#undef PACKAGE_TARNAME
+#undef PACKAGE_VERSION
+
+#ifdef _WIN32
+/* On Windows this stuff is also defined by Postgres, but we don't
+   want to use Postgres' version actually */
+#undef fsync
+#undef ftruncate
+#undef fseeko
+#undef ftello
+#undef stat
+#undef vsnprintf
+#undef snprintf
+#undef sprintf
+#undef printf
+#define cCommand_execute cCommand_execute_sync
+#define do_int64 signed __int64
+#else
+#define cCommand_execute cCommand_execute_async
+#define do_int64 signed long long int
+#endif
+
+#include <ruby.h>
+#include <string.h>
+#include <math.h>
+#include <ctype.h>
+#include <time.h>
+#include "error.h"
+#include "compat.h"
+
+#define ID_CONST_GET rb_intern("const_get")
+#define ID_PATH rb_intern("path")
+#define ID_NEW rb_intern("new")
+#define ID_ESCAPE rb_intern("escape_sql")
+
+#define CONST_GET(scope, constant) (rb_funcall(scope, ID_CONST_GET, 1, rb_str_new2(constant)))
+#define POSTGRES_CLASS(klass, parent) (rb_define_class_under(mPostgres, klass, parent))
+
+#ifdef HAVE_RUBY_ENCODING_H
+#include <ruby/encoding.h>
+
+#define DO_STR_NEW2(str, encoding) \
+  ({ \
+    VALUE _string = rb_str_new2((const char *)str); \
+    if(encoding != -1) { \
+      rb_enc_associate_index(_string, encoding); \
+    } \
+    _string; \
+  })
+
+#define DO_STR_NEW(str, len, encoding) \
+  ({ \
+    VALUE _string = rb_str_new((const char *)str, (long)len); \
+    if(encoding != -1) { \
+      rb_enc_associate_index(_string, encoding); \
+    } \
+    _string; \
+  })
+
+#else
+
+#define DO_STR_NEW2(str, doc) \
+  rb_str_new2((const char *)str)
+
+#define DO_STR_NEW(str, len, doc) \
+  rb_str_new((const char *)str, (long)len)
+#endif
+
+
+// To store rb_intern values
+static ID ID_NEW_DATE;
+static ID ID_LOGGER;
+static ID ID_DEBUG;
+static ID ID_LEVEL;
+static ID ID_TO_S;
+static ID ID_RATIONAL;
+
+static VALUE mExtlib;
+static VALUE mDO;
+static VALUE mEncoding;
+static VALUE cDO_Quoting;
+static VALUE cDO_Connection;
+static VALUE cDO_Command;
+static VALUE cDO_Result;
+static VALUE cDO_Reader;
+
+static VALUE rb_cDate;
+static VALUE rb_cDateTime;
+static VALUE rb_cBigDecimal;
+static VALUE rb_cByteArray;
+
+static VALUE mPostgres;
+static VALUE cConnection;
+static VALUE cCommand;
+static VALUE cResult;
+static VALUE cReader;
+
+static VALUE eArgumentError;
+static VALUE eConnectionError;
+static VALUE eDataError;
+
+static void data_objects_debug(VALUE string, struct timeval* start) {
+  struct timeval stop;
+  char *message;
+
+  const char *query = rb_str_ptr_readonly(string);
+  size_t length     = rb_str_len(string);
+  char total_time[32];
+  do_int64 duration = 0;
+
+  VALUE logger = rb_funcall(mPostgres, ID_LOGGER, 0);
+  int log_level = NUM2INT(rb_funcall(logger, ID_LEVEL, 0));
+
+  if (0 == log_level) {
+    gettimeofday(&stop, NULL);
+
+    duration = (stop.tv_sec - start->tv_sec) * 1000000 + stop.tv_usec - start->tv_usec;
+
+    snprintf(total_time, 32, "%.6f", duration / 1000000.0);
+    message = (char *)calloc(length + strlen(total_time) + 4, sizeof(char));
+    snprintf(message, length + strlen(total_time) + 4, "(%s) %s", total_time, query);
+    rb_funcall(logger, ID_DEBUG, 1, rb_str_new(message, length + strlen(total_time) + 3));
+  }
+}
+
+static const char * get_uri_option(VALUE query_hash, const char * key) {
+  VALUE query_value;
+  const char * value = NULL;
+
+  if(!rb_obj_is_kind_of(query_hash, rb_cHash)) { return NULL; }
+
+  query_value = rb_hash_aref(query_hash, rb_str_new2(key));
+
+  if (Qnil != query_value) {
+    value = StringValuePtr(query_value);
+  }
+
+  return value;
+}
+
+/* ====== Time/Date Parsing Helper Functions ====== */
+static void reduce( do_int64 *numerator, do_int64 *denominator ) {
+  do_int64 a, b, c;
+  a = *numerator;
+  b = *denominator;
+  while ( a != 0 ) {
+    c = a; a = b % a; b = c;
+  }
+  *numerator = *numerator / b;
+  *denominator = *denominator / b;
+}
+
+// Generate the date integer which Date.civil_to_jd returns
+static int jd_from_date(int year, int month, int day) {
+  int a, b;
+  if ( month <= 2 ) {
+    year -= 1;
+    month += 12;
+  }
+  a = year / 100;
+  b = 2 - a + (a / 4);
+  return (int) (floor(365.25 * (year + 4716)) + floor(30.6001 * (month + 1)) + day + b - 1524);
+}
+
+static VALUE parse_date(const char *date) {
+  int year, month, day;
+  int jd, ajd;
+  VALUE rational;
+
+  sscanf(date, "%4d-%2d-%2d", &year, &month, &day);
+
+  jd = jd_from_date(year, month, day);
+
+  // Math from Date.jd_to_ajd
+  ajd = jd * 2 - 1;
+  rational = rb_funcall(rb_mKernel, ID_RATIONAL, 2, INT2NUM(ajd), INT2NUM(2));
+
+  return rb_funcall(rb_cDate, ID_NEW_DATE, 3, rational, INT2NUM(0), INT2NUM(2299161));
+}
+
+// Creates a Rational for use as a Timezone offset to be passed to DateTime.new!
+static VALUE seconds_to_offset(do_int64 num) {
+  do_int64 den = 86400;
+  reduce(&num, &den);
+  return rb_funcall(rb_mKernel, ID_RATIONAL, 2, rb_ll2inum(num), rb_ll2inum(den));
+}
+
+static VALUE timezone_to_offset(int hour_offset, int minute_offset) {
+  do_int64 seconds = 0;
+
+  seconds += hour_offset * 3600;
+  seconds += minute_offset * 60;
+
+  return seconds_to_offset(seconds);
+}
+
+static VALUE parse_date_time(const char *date) {
+  VALUE ajd, offset;
+
+  int year, month, day, hour, min, sec, usec, hour_offset, minute_offset;
+  int jd;
+  do_int64 num, den;
+
+  long int gmt_offset;
+  int is_dst;
+
+  time_t rawtime;
+  struct tm * timeinfo;
+
+  int tokens_read, max_tokens;
+
+  if (0 != strchr(date, '.')) {
+    // This is a datetime with sub-second precision
+    tokens_read = sscanf(date, "%4d-%2d-%2d %2d:%2d:%2d.%d%3d:%2d", &year, &month, &day, &hour, &min, &sec, &usec, &hour_offset, &minute_offset);
+    max_tokens = 9;
+  } else {
+    // This is a datetime second precision
+    tokens_read = sscanf(date, "%4d-%2d-%2d %2d:%2d:%2d%3d:%2d", &year, &month, &day, &hour, &min, &sec, &hour_offset, &minute_offset);
+    max_tokens = 8;
+  }
+
+  if (max_tokens == tokens_read) {
+    // We read the Date, Time, and Timezone info
+    minute_offset *= hour_offset < 0 ? -1 : 1;
+  } else if ((max_tokens - 1) == tokens_read) {
+    // We read the Date and Time, but no Minute Offset
+    minute_offset = 0;
+  } else if (tokens_read == 3 || tokens_read >= (max_tokens - 3)) {
+    if (tokens_read == 3) {
+      hour = 0;
+      min = 0;
+      hour_offset = 0;
+      minute_offset = 0;
+      sec = 0;
+    }
+    // We read the Date and Time, default to the current locale's offset
+
+    // Get localtime
+    time(&rawtime);
+    timeinfo = localtime(&rawtime);
+
+    is_dst = timeinfo->tm_isdst * 3600;
+
+    // Reset to GM Time
+    timeinfo = gmtime(&rawtime);
+
+    gmt_offset = mktime(timeinfo) - rawtime;
+
+    if ( is_dst > 0 )
+      gmt_offset -= is_dst;
+
+    hour_offset = -((int)gmt_offset / 3600);
+    minute_offset = -((int)gmt_offset % 3600 / 60);
+
+  } else {
+    // Something went terribly wrong
+    rb_raise(eDataError, "Couldn't parse date: %s", date);
+  }
+
+  jd = jd_from_date(year, month, day);
+
+  // Generate ajd with fractional days for the time
+  // Extracted from Date#jd_to_ajd, Date#day_fraction_to_time, and Rational#+ and #-
+  num = (hour * 1440) + (min * 24);
+
+  // Modify the numerator so when we apply the timezone everything works out
+  num -= (hour_offset * 1440) + (minute_offset * 24);
+
+  den = (24 * 1440);
+  reduce(&num, &den);
+
+  num = (num * 86400) + (sec * den);
+  den = den * 86400;
+  reduce(&num, &den);
+
+  num = (jd * den) + num;
+
+  num = num * 2;
+  num = num - den;
+  den = den * 2;
+
+  reduce(&num, &den);
+
+  ajd = rb_funcall(rb_mKernel, ID_RATIONAL, 2, rb_ull2inum(num), rb_ull2inum(den));
+  offset = timezone_to_offset(hour_offset, minute_offset);
+
+  return rb_funcall(rb_cDateTime, ID_NEW_DATE, 3, ajd, offset, INT2NUM(2299161));
+}
+
+static VALUE parse_time(const char *date) {
+
+  int year, month, day, hour, min, sec, usec, tokens;
+  char subsec[7];
+
+  if (0 != strchr(date, '.')) {
+    // right padding usec with 0. e.g. '012' will become 12000 microsecond, since Time#local use microsecond
+    sscanf(date, "%4d-%2d-%2d %2d:%2d:%2d.%s", &year, &month, &day, &hour, &min, &sec, subsec);
+    usec   = atoi(subsec);
+    usec  *= (int) pow(10, (6 - strlen(subsec)));
+  } else {
+    tokens = sscanf(date, "%4d-%2d-%2d %2d:%2d:%2d", &year, &month, &day, &hour, &min, &sec);
+    if (tokens == 3) {
+      hour = 0;
+      min  = 0;
+      sec  = 0;
+    }
+    usec = 0;
+  }
+
+  return rb_funcall(rb_cTime, rb_intern("local"), 7, INT2NUM(year), INT2NUM(month), INT2NUM(day), INT2NUM(hour), INT2NUM(min), INT2NUM(sec), INT2NUM(usec));
+}
+
+/* ===== Typecasting Functions ===== */
+
+static VALUE infer_ruby_type(Oid type) {
+  switch(type) {
+    case BITOID:
+    case VARBITOID:
+    case INT2OID:
+    case INT4OID:
+    case INT8OID:
+      return rb_cInteger;
+    case FLOAT4OID:
+    case FLOAT8OID:
+      return rb_cFloat;
+    case NUMERICOID:
+    case CASHOID:
+      return rb_cBigDecimal;
+    case BOOLOID:
+      return rb_cTrueClass;
+    case TIMESTAMPTZOID:
+    case TIMESTAMPOID:
+      return rb_cDateTime;
+    case DATEOID:
+      return rb_cDate;
+    case BYTEAOID:
+      return rb_cByteArray;
+    default:
+      return rb_cString;
+  }
+}
+
+static VALUE typecast(const char *value, long length, const VALUE type, int encoding) {
+
+  if (type == rb_cInteger) {
+    return rb_cstr2inum(value, 10);
+  } else if (type == rb_cString) {
+    return DO_STR_NEW(value, length, encoding);
+  } else if (type == rb_cFloat) {
+    return rb_float_new(rb_cstr_to_dbl(value, Qfalse));
+  } else if (type == rb_cBigDecimal) {
+    return rb_funcall(rb_cBigDecimal, ID_NEW, 1, rb_str_new(value, length));
+  } else if (type == rb_cDate) {
+    return parse_date(value);
+  } else if (type == rb_cDateTime) {
+    return parse_date_time(value);
+  } else if (type == rb_cTime) {
+    return parse_time(value);
+  } else if (type == rb_cTrueClass) {
+    return *value == 't' ? Qtrue : Qfalse;
+  } else if (type == rb_cByteArray) {
+    size_t new_length = 0;
+    char* unescaped = (char *)PQunescapeBytea((unsigned char*)value, &new_length);
+    VALUE byte_array = rb_funcall(rb_cByteArray, ID_NEW, 1, rb_str_new(unescaped, new_length));
+    PQfreemem(unescaped);
+    return byte_array;
+  } else if (type == rb_cClass) {
+    return rb_funcall(mDO, rb_intern("full_const_get"), 1, rb_str_new(value, length));
+  } else if (type == rb_cObject) {
+    return rb_marshal_load(rb_str_new(value, length));
+  } else if (type == rb_cNilClass) {
+    return Qnil;
+  } else {
+    return DO_STR_NEW(value, length, encoding);
+  }
+
+}
+
+static void raise_error(VALUE self, PGresult *result, VALUE query) {
+  VALUE exception;
+  char *message;
+  char *sqlstate;
+  int postgres_errno;
+
+  message  = PQresultErrorMessage(result);
+  sqlstate = PQresultErrorField(result, PG_DIAG_SQLSTATE);
+  postgres_errno = MAKE_SQLSTATE(sqlstate[0], sqlstate[1], sqlstate[2], sqlstate[3], sqlstate[4]);
+  PQclear(result);
+
+  const char *exception_type = "SQLError";
+
+  struct errcodes *errs;
+
+  for (errs = errors; errs->error_name; errs++) {
+    if(errs->error_no == postgres_errno) {
+      exception_type = errs->exception;
+      break;
+    }
+  }
+
+  VALUE uri = rb_funcall(rb_iv_get(self, "@connection"), rb_intern("to_s"), 0);
+
+  exception = rb_funcall(CONST_GET(mDO, exception_type), ID_NEW, 5,
+                         rb_str_new2(message),
+                         INT2NUM(postgres_errno),
+                         rb_str_new2(sqlstate),
+                         query,
+                         uri);
+  rb_exc_raise(exception);
+}
+
+
+/* ====== Public API ======= */
+static VALUE cConnection_dispose(VALUE self) {
+  VALUE connection_container = rb_iv_get(self, "@connection");
+
+  PGconn *db;
+
+  if (Qnil == connection_container)
+    return Qfalse;
+
+  db = DATA_PTR(connection_container);
+
+  if (NULL == db)
+    return Qfalse;
+
+  PQfinish(db);
+  rb_iv_set(self, "@connection", Qnil);
+
+  return Qtrue;
+}
+
+static VALUE cCommand_set_types(int argc, VALUE *argv, VALUE self) {
+  VALUE type_strings = rb_ary_new();
+  VALUE array = rb_ary_new();
+
+  int i, j;
+
+  for ( i = 0; i < argc; i++) {
+    rb_ary_push(array, argv[i]);
+  }
+
+  for (i = 0; i < RARRAY_LEN(array); i++) {
+    VALUE entry = rb_ary_entry(array, i);
+    if(TYPE(entry) == T_CLASS) {
+      rb_ary_push(type_strings, entry);
+    } else if (TYPE(entry) == T_ARRAY) {
+      for (j = 0; j < RARRAY_LEN(entry); j++) {
+        VALUE sub_entry = rb_ary_entry(entry, j);
+        if(TYPE(sub_entry) == T_CLASS) {
+          rb_ary_push(type_strings, sub_entry);
+        } else {
+          rb_raise(eArgumentError, "Invalid type given");
+        }
+      }
+    } else {
+      rb_raise(eArgumentError, "Invalid type given");
+    }
+  }
+
+  rb_iv_set(self, "@field_types", type_strings);
+
+  return array;
+}
+
+static VALUE build_query_from_args(VALUE klass, int count, VALUE *args[]) {
+  VALUE query = rb_iv_get(klass, "@text");
+
+  int i;
+  VALUE array = rb_ary_new();
+  for ( i = 0; i < count; i++) {
+    rb_ary_push(array, (VALUE)args[i]);
+  }
+  query = rb_funcall(klass, ID_ESCAPE, 1, array);
+
+  return query;
+}
+
+static VALUE cConnection_quote_string(VALUE self, VALUE string) {
+  PGconn *db = DATA_PTR(rb_iv_get(self, "@connection"));
+
+  const char *source = rb_str_ptr_readonly(string);
+  size_t source_len  = rb_str_len(string);
+
+  char *escaped;
+  size_t quoted_length = 0;
+  VALUE result;
+
+  // Allocate space for the escaped version of 'string'
+  // http://www.postgresql.org/docs/8.3/static/libpq-exec.html#LIBPQ-EXEC-ESCAPE-STRING
+  escaped = (char *)calloc(source_len * 2 + 3, sizeof(char));
+
+  // Escape 'source' using the current charset in use on the conection 'db'
+  quoted_length = PQescapeStringConn(db, escaped + 1, source, source_len, NULL);
+
+  // Wrap the escaped string in single-quotes, this is DO's convention
+  escaped[quoted_length + 1] = escaped[0] = '\'';
+
+  result = DO_STR_NEW(escaped, quoted_length + 2, FIX2INT(rb_iv_get(self, "@encoding_id")));
+
+  free(escaped);
+  return result;
+}
+
+static VALUE cConnection_quote_byte_array(VALUE self, VALUE string) {
+  PGconn *db = DATA_PTR(rb_iv_get(self, "@connection"));
+
+  const unsigned char *source = (unsigned char*) rb_str_ptr_readonly(string);
+  size_t source_len     = rb_str_len(string);
+
+  unsigned char *escaped;
+  unsigned char *escaped_quotes;
+  size_t quoted_length = 0;
+  VALUE result;
+
+  // Allocate space for the escaped version of 'string'
+  // http://www.postgresql.org/docs/8.3/static/libpq-exec.html#LIBPQ-EXEC-ESCAPE-STRING
+  escaped = PQescapeByteaConn(db, source, source_len, &quoted_length);
+  escaped_quotes = (unsigned char *)calloc(quoted_length + 1, sizeof(unsigned char));
+  memcpy(escaped_quotes + 1, escaped, quoted_length);
+
+  // Wrap the escaped string in single-quotes, this is DO's convention (replace trailing \0)
+  escaped_quotes[quoted_length] = escaped_quotes[0] = '\'';
+
+  result = rb_str_new((const char *)escaped_quotes, quoted_length + 1);
+  PQfreemem(escaped);
+  free(escaped_quotes);
+  return result;
+}
+
+static void full_connect(VALUE self, PGconn *db);
+
+#ifdef _WIN32
+static PGresult* cCommand_execute_sync(VALUE self, PGconn *db, VALUE query) {
+  PGresult *response;
+  struct timeval start;
+  char* str = StringValuePtr(query);
+
+  while ((response = PQgetResult(db)) != NULL) {
+    PQclear(response);
+  }
+
+  gettimeofday(&start, NULL);
+
+  response = PQexec(db, str);
+
+  data_objects_debug(query, &start);
+
+  if (response == NULL) {
+    if(PQstatus(db) != CONNECTION_OK) {
+      PQreset(db);
+      if (PQstatus(db) == CONNECTION_OK) {
+        response = PQexec(db, str);
+      } else {
+        VALUE connection = rb_iv_get(self, "@connection");
+        full_connect(connection, db);
+        response = PQexec(db, str);
+      }
+    }
+
+    if(response == NULL) {
+      rb_raise(eConnectionError, PQerrorMessage(db));
+    }
+  }
+
+  return response;
+}
+#else
+static PGresult* cCommand_execute_async(VALUE self, PGconn *db, VALUE query) {
+  int socket_fd;
+  int retval;
+  fd_set rset;
+  PGresult *response;
+  struct timeval start;
+  char* str = StringValuePtr(query);
+
+  while ((response = PQgetResult(db)) != NULL) {
+    PQclear(response);
+  }
+
+  retval = PQsendQuery(db, str);
+
+  if (!retval) {
+    if(PQstatus(db) != CONNECTION_OK) {
+      PQreset(db);
+      if (PQstatus(db) == CONNECTION_OK) {
+        retval = PQsendQuery(db, str);
+      } else {
+        VALUE connection = rb_iv_get(self, "@connection");
+        full_connect(connection, db);
+        retval = PQsendQuery(db, str);
+      }
+    }
+
+    if(!retval) {
+      rb_raise(eConnectionError, "%s", PQerrorMessage(db));
+    }
+  }
+
+  gettimeofday(&start, NULL);
+  socket_fd = PQsocket(db);
+
+  data_objects_debug(query, &start);
+
+  for(;;) {
+      FD_ZERO(&rset);
+      FD_SET(socket_fd, &rset);
+      retval = rb_thread_select(socket_fd + 1, &rset, NULL, NULL, NULL);
+      if (retval < 0) {
+          rb_sys_fail(0);
+      }
+
+      if (retval == 0) {
+          continue;
+      }
+
+      if (PQconsumeInput(db) == 0) {
+          rb_raise(eConnectionError, "%s", PQerrorMessage(db));
+      }
+
+      if (PQisBusy(db) == 0) {
+          break;
+      }
+  }
+
+  return PQgetResult(db);
+}
+#endif
+
+static VALUE cConnection_initialize(VALUE self, VALUE uri) {
+  VALUE r_host, r_user, r_password, r_path, r_query, r_port;
+
+  PGconn *db = NULL;
+
+  rb_iv_set(self, "@using_socket", Qfalse);
+
+  r_host = rb_funcall(uri, rb_intern("host"), 0);
+  if (Qnil != r_host) {
+    rb_iv_set(self, "@host", r_host);
+  }
+
+  r_user = rb_funcall(uri, rb_intern("user"), 0);
+  if (Qnil != r_user) {
+    rb_iv_set(self, "@user", r_user);
+  }
+
+  r_password = rb_funcall(uri, rb_intern("password"), 0);
+  if (Qnil != r_password) {
+    rb_iv_set(self, "@password", r_password);
+  }
+
+  r_path = rb_funcall(uri, rb_intern("path"), 0);
+  if (Qnil != r_path) {
+    rb_iv_set(self, "@path", r_path);
+  }
+
+  r_port = rb_funcall(uri, rb_intern("port"), 0);
+  if (Qnil != r_port) {
+    r_port = rb_funcall(r_port, rb_intern("to_s"), 0);
+    rb_iv_set(self, "@port", r_port);
+  }
+
+  // Pull the querystring off the URI
+  r_query = rb_funcall(uri, rb_intern("query"), 0);
+  rb_iv_set(self, "@query", r_query);
+
+  const char* encoding = get_uri_option(r_query, "encoding");
+  if (!encoding) { encoding = get_uri_option(r_query, "charset"); }
+  if (!encoding) { encoding = "UTF-8"; }
+
+  rb_iv_set(self, "@encoding", rb_str_new2(encoding));
+
+  full_connect(self, db);
+
+  rb_iv_set(self, "@uri", uri);
+
+  return Qtrue;
+}
+
+static void full_connect(VALUE self, PGconn *db) {
+
+  PGresult *result = NULL;
+  VALUE r_host, r_user, r_password, r_path, r_port, r_query, r_options;
+  char *host = NULL, *user = NULL, *password = NULL, *path = NULL, *database = NULL;
+  const char *port = "5432";
+  VALUE encoding = Qnil;
+  const char *search_path = NULL;
+  char *search_path_query = NULL;
+  const char *backslash_off = "SET backslash_quote = off";
+  const char *standard_strings_on = "SET standard_conforming_strings = on";
+  const char *warning_messages = "SET client_min_messages = warning";
+
+  if((r_host = rb_iv_get(self, "@host")) != Qnil) {
+    host     = StringValuePtr(r_host);
+  }
+
+  if((r_user = rb_iv_get(self, "@user")) != Qnil) {
+    user     = StringValuePtr(r_user);
+  }
+
+  if((r_password = rb_iv_get(self, "@password")) != Qnil) {
+    password = StringValuePtr(r_password);
+  }
+
+  if((r_port = rb_iv_get(self, "@port")) != Qnil) {
+    port = StringValuePtr(r_port);
+  }
+
+  if((r_path = rb_iv_get(self, "@path")) != Qnil) {
+    path = StringValuePtr(r_path);
+    database = strtok(path, "/");
+  }
+
+  if (NULL == database || 0 == strlen(database)) {
+    rb_raise(eConnectionError, "Database must be specified");
+  }
+
+  r_query        = rb_iv_get(self, "@query");
+
+  search_path = get_uri_option(r_query, "search_path");
+
+  db = PQsetdbLogin(
+    host,
+    port,
+    NULL,
+    NULL,
+    database,
+    user,
+    password
+  );
+
+  if ( PQstatus(db) == CONNECTION_BAD ) {
+    rb_raise(eConnectionError, "%s", PQerrorMessage(db));
+  }
+
+  if (search_path != NULL) {
+    search_path_query = (char *)calloc(256, sizeof(char));
+    snprintf(search_path_query, 256, "set search_path to %s;", search_path);
+    r_query = rb_str_new2(search_path_query);
+    result = cCommand_execute(self, db, r_query);
+
+    if (PQresultStatus(result) != PGRES_COMMAND_OK) {
+      free((void *)search_path_query);
+      raise_error(self, result, r_query);
+    }
+
+    free((void *)search_path_query);
+  }
+
+  r_options = rb_str_new2(backslash_off);
+  result = cCommand_execute(self, db, r_options);
+
+  if (PQresultStatus(result) != PGRES_COMMAND_OK) {
+    rb_warn("%s", PQresultErrorMessage(result));
+  }
+
+  r_options = rb_str_new2(standard_strings_on);
+  result = cCommand_execute(self, db, r_options);
+
+  if (PQresultStatus(result) != PGRES_COMMAND_OK) {
+    rb_warn("%s", PQresultErrorMessage(result));
+  }
+
+  r_options = rb_str_new2(warning_messages);
+  result = cCommand_execute(self, db, r_options);
+
+  if (PQresultStatus(result) != PGRES_COMMAND_OK) {
+    rb_warn("%s", PQresultErrorMessage(result));
+  }
+
+  encoding = rb_iv_get(self, "@encoding");
+
+#ifdef HAVE_PQSETCLIENTENCODING
+  VALUE pg_encoding = rb_hash_aref(CONST_GET(mEncoding, "MAP"), encoding);
+  if(pg_encoding != Qnil) {
+    if(PQsetClientEncoding(db, rb_str_ptr_readonly(pg_encoding))) {
+      rb_raise(eConnectionError, "Couldn't set encoding: %s", rb_str_ptr_readonly(encoding));
+    } else {
+#ifdef HAVE_RUBY_ENCODING_H
+      rb_iv_set(self, "@encoding_id", INT2FIX(rb_enc_find_index(rb_str_ptr_readonly(encoding))));
+#endif
+      rb_iv_set(self, "@pg_encoding", pg_encoding);
+    }
+  } else {
+    rb_warn("Encoding %s is not a known Ruby encoding for PostgreSQL\n", rb_str_ptr_readonly(encoding));
+    rb_iv_set(self, "@encoding", rb_str_new2("UTF-8"));
+#ifdef HAVE_RUBY_ENCODING_H
+    rb_iv_set(self, "@encoding_id", INT2FIX(rb_enc_find_index("UTF-8")));
+#endif
+    rb_iv_set(self, "@pg_encoding", rb_str_new2("UTF8"));
+  }
+#endif
+  rb_iv_set(self, "@connection", Data_Wrap_Struct(rb_cObject, 0, 0, db));
+}
+
+static VALUE cConnection_character_set(VALUE self) {
+  return rb_iv_get(self, "@encoding");
+}
+
+static VALUE cCommand_execute_non_query(int argc, VALUE *argv[], VALUE self) {
+  VALUE connection = rb_iv_get(self, "@connection");
+  VALUE postgres_connection = rb_iv_get(connection, "@connection");
+  if (Qnil == postgres_connection) {
+    rb_raise(eConnectionError, "This connection has already been closed.");
+  }
+
+  PGconn *db = DATA_PTR(postgres_connection);
+  PGresult *response;
+  int status;
+
+  VALUE affected_rows = Qnil;
+  VALUE insert_id = Qnil;
+
+  VALUE query = build_query_from_args(self, argc, argv);
+
+  response = cCommand_execute(self, db, query);
+
+  status = PQresultStatus(response);
+
+  if ( status == PGRES_TUPLES_OK ) {
+    insert_id = INT2NUM(atoi(PQgetvalue(response, 0, 0)));
+    affected_rows = INT2NUM(atoi(PQcmdTuples(response)));
+  }
+  else if ( status == PGRES_COMMAND_OK ) {
+    insert_id = Qnil;
+    affected_rows = INT2NUM(atoi(PQcmdTuples(response)));
+  }
+  else {
+    raise_error(self, response, query);
+  }
+
+  PQclear(response);
+
+  return rb_funcall(cResult, ID_NEW, 3, self, affected_rows, insert_id);
+}
+
+static VALUE cCommand_execute_reader(int argc, VALUE *argv[], VALUE self) {
+  VALUE reader, query;
+  VALUE field_names, field_types;
+
+  int i;
+  int field_count;
+  int infer_types = 0;
+
+  VALUE connection = rb_iv_get(self, "@connection");
+  VALUE postgres_connection = rb_iv_get(connection, "@connection");
+  if (Qnil == postgres_connection) {
+    rb_raise(eConnectionError, "This connection has already been closed.");
+  }
+
+  PGconn *db = DATA_PTR(postgres_connection);
+  PGresult *response;
+
+  query = build_query_from_args(self, argc, argv);
+
+  response = cCommand_execute(self, db, query);
+
+  if ( PQresultStatus(response) != PGRES_TUPLES_OK ) {
+    raise_error(self, response, query);
+  }
+
+  field_count = PQnfields(response);
+
+  reader = rb_funcall(cReader, ID_NEW, 0);
+  rb_iv_set(reader, "@connection", connection);
+  rb_iv_set(reader, "@reader", Data_Wrap_Struct(rb_cObject, 0, 0, response));
+  rb_iv_set(reader, "@field_count", INT2NUM(field_count));
+  rb_iv_set(reader, "@row_count", INT2NUM(PQntuples(response)));
+
+  field_names = rb_ary_new();
+  field_types = rb_iv_get(self, "@field_types");
+
+  if ( field_types == Qnil || 0 == RARRAY_LEN(field_types) ) {
+    field_types = rb_ary_new();
+    infer_types = 1;
+  } else if (RARRAY_LEN(field_types) != field_count) {
+    // Whoops...  wrong number of types passed to set_types.  Close the reader and raise
+    // and error
+    rb_funcall(reader, rb_intern("close"), 0);
+    rb_raise(eArgumentError, "Field-count mismatch. Expected %ld fields, but the query yielded %d", RARRAY_LEN(field_types), field_count);
+  }
+
+  for ( i = 0; i < field_count; i++ ) {
+    rb_ary_push(field_names, rb_str_new2(PQfname(response, i)));
+    if ( infer_types == 1 ) {
+      rb_ary_push(field_types, infer_ruby_type(PQftype(response, i)));
+    }
+  }
+
+  rb_iv_set(reader, "@position", INT2NUM(0));
+  rb_iv_set(reader, "@fields", field_names);
+  rb_iv_set(reader, "@field_types", field_types);
+
+  return reader;
+}
+
+static VALUE cReader_close(VALUE self) {
+  VALUE reader_container = rb_iv_get(self, "@reader");
+
+  PGresult *reader;
+
+  if (Qnil == reader_container)
+    return Qfalse;
+
+  reader = DATA_PTR(reader_container);
+
+  if (NULL == reader)
+    return Qfalse;
+
+  PQclear(reader);
+  rb_iv_set(self, "@reader", Qnil);
+  return Qtrue;
+}
+
+static VALUE cReader_next(VALUE self) {
+  PGresult *reader = DATA_PTR(rb_iv_get(self, "@reader"));
+
+  int field_count;
+  int row_count;
+  int i;
+  int position;
+
+  VALUE array = rb_ary_new();
+  VALUE field_types, field_type;
+  VALUE value;
+
+  row_count = NUM2INT(rb_iv_get(self, "@row_count"));
+  field_count = NUM2INT(rb_iv_get(self, "@field_count"));
+  field_types = rb_iv_get(self, "@field_types");
+  position = NUM2INT(rb_iv_get(self, "@position"));
+
+  if ( position > (row_count - 1) ) {
+    rb_iv_set(self, "@values", Qnil);
+    return Qfalse;
+  }
+
+  int enc = -1;
+#ifdef HAVE_RUBY_ENCODING_H
+  VALUE encoding_id = rb_iv_get(rb_iv_get(self, "@connection"), "@encoding_id");
+  if (encoding_id != Qnil) {
+    enc = FIX2INT(encoding_id);
+  }
+#endif
+
+  for ( i = 0; i < field_count; i++ ) {
+    field_type = rb_ary_entry(field_types, i);
+
+    // Always return nil if the value returned from Postgres is null
+    if (!PQgetisnull(reader, position, i)) {
+      value = typecast(PQgetvalue(reader, position, i), PQgetlength(reader, position, i), field_type, enc);
+    } else {
+      value = Qnil;
+    }
+
+    rb_ary_push(array, value);
+  }
+
+  rb_iv_set(self, "@values", array);
+  rb_iv_set(self, "@position", INT2NUM(position+1));
+
+  return Qtrue;
+}
+
+static VALUE cReader_values(VALUE self) {
+
+  VALUE values = rb_iv_get(self, "@values");
+  if(values == Qnil) {
+    rb_raise(eDataError, "Reader not initialized");
+    return Qnil;
+  } else {
+    return values;
+  }
+}
+
+static VALUE cReader_fields(VALUE self) {
+  return rb_iv_get(self, "@fields");
+}
+
+static VALUE cReader_field_count(VALUE self) {
+  return rb_iv_get(self, "@field_count");
+}
+
+void Init_do_postgres() {
+  rb_require("date");
+  rb_require("bigdecimal");
+
+  // Get references classes needed for Date/Time parsing
+  rb_cDate = CONST_GET(rb_mKernel, "Date");
+  rb_cDateTime = CONST_GET(rb_mKernel, "DateTime");
+  rb_cBigDecimal = CONST_GET(rb_mKernel, "BigDecimal");
+
+  rb_funcall(rb_mKernel, rb_intern("require"), 1, rb_str_new2("data_objects"));
+
+#ifdef RUBY_LESS_THAN_186
+  ID_NEW_DATE = rb_intern("new0");
+#else
+  ID_NEW_DATE = rb_intern("new!");
+#endif
+  ID_LOGGER = rb_intern("logger");
+  ID_DEBUG = rb_intern("debug");
+  ID_LEVEL = rb_intern("level");
+  ID_TO_S = rb_intern("to_s");
+  ID_RATIONAL = rb_intern("Rational");
+
+  // Get references to the Extlib module
+  mExtlib = CONST_GET(rb_mKernel, "Extlib");
+  rb_cByteArray = CONST_GET(mExtlib, "ByteArray");
+
+  // Get references to the DataObjects module and its classes
+  mDO = CONST_GET(rb_mKernel, "DataObjects");
+  cDO_Quoting = CONST_GET(mDO, "Quoting");
+  cDO_Connection = CONST_GET(mDO, "Connection");
+  cDO_Command = CONST_GET(mDO, "Command");
+  cDO_Result = CONST_GET(mDO, "Result");
+  cDO_Reader = CONST_GET(mDO, "Reader");
+
+  eArgumentError = CONST_GET(rb_mKernel, "ArgumentError");
+  mPostgres = rb_define_module_under(mDO, "Postgres");
+  eConnectionError = CONST_GET(mDO, "ConnectionError");
+  eDataError = CONST_GET(mDO, "DataError");
+  mEncoding = rb_define_module_under(mPostgres, "Encoding");
+
+  cConnection = POSTGRES_CLASS("Connection", cDO_Connection);
+  rb_define_method(cConnection, "initialize", cConnection_initialize, 1);
+  rb_define_method(cConnection, "dispose", cConnection_dispose, 0);
+  rb_define_method(cConnection, "character_set", cConnection_character_set , 0);
+  rb_define_method(cConnection, "quote_string", cConnection_quote_string, 1);
+  rb_define_method(cConnection, "quote_byte_array", cConnection_quote_byte_array, 1);
+
+  cCommand = POSTGRES_CLASS("Command", cDO_Command);
+  rb_define_method(cCommand, "set_types", cCommand_set_types, -1);
+  rb_define_method(cCommand, "execute_non_query", cCommand_execute_non_query, -1);
+  rb_define_method(cCommand, "execute_reader", cCommand_execute_reader, -1);
+
+  cResult = POSTGRES_CLASS("Result", cDO_Result);
+
+  cReader = POSTGRES_CLASS("Reader", cDO_Reader);
+  rb_define_method(cReader, "close", cReader_close, 0);
+  rb_define_method(cReader, "next!", cReader_next, 0);
+  rb_define_method(cReader, "values", cReader_values, 0);
+  rb_define_method(cReader, "fields", cReader_fields, 0);
+  rb_define_method(cReader, "field_count", cReader_field_count, 0);
+
+  struct errcodes *errs;
+
+  for (errs = errors; errs->error_name; errs++) {
+    rb_const_set(mPostgres, rb_intern(errs->error_name), INT2NUM(errs->error_no));
+  }
+
+}