pg_json 0.1

data/bin/pg_json

@@ -0,0 +1,84 @@
+#!/usr/bin/env ruby
+
+require "sequel"
+
+DB = Sequel.connect "postgres://#{ARGV.slice!(0)}", :logger => nil, :max_connections => 1
+
+def unquoted(type)
+  puts <<-EOF
+create or replace function to_json(data #{type}) returns text as $$
+begin
+  return (case when data is null then 'null' else data::text end);
+end;
+$$ language plpgsql;
+  EOF
+  array(type)
+end
+
+def quoted(type)
+  puts <<-EOF
+create or replace function to_json(data #{type}) returns text as $$
+begin
+  return (case when data is null then 'null' else '"'||data::text||'"' end);
+end;
+$$ language plpgsql;
+  EOF
+  array(type)
+end
+
+def composite(type)
+  sql = <<-SQL
+    select t1.name from (
+      select column_name as name, ordinal_position from information_schema.columns where table_name = '#{type}'
+      union
+      select attribute_name as name, ordinal_position from information_schema.attributes where udt_name = '#{type}'
+    ) t1
+    order by t1.ordinal_position asc
+  SQL
+  
+  fields = DB[sql].all.map do |row|
+    "\"" + row[:name] + "\":'||to_json(data.#{row[:name]})"
+  end.join("||',")
+  
+  puts <<-EOF
+drop function if exists to_json(#{type});
+create or replace function to_json(data #{type}) returns text as $$
+begin
+  if data is null then
+    return 'null';
+  else
+    return '{#{fields}||'}';
+  end if;
+end;
+$$ language plpgsql;
+  EOF
+  
+  array(type)
+end
+
+def array(type)
+  puts <<-EOF
+create or replace function to_json(data #{type}[]) returns text as $$
+declare
+  res text;
+begin
+  res = '[';
+  if data is not null then
+    for i in 1..array_upper(data,1) loop
+      res = res || to_json(data[i]);
+      if i < array_upper(data,1) then
+        res = res || ',';
+      end if;
+    end loop;
+  end if;
+  res = res || ']';
+  return res;
+end;
+$$ language plpgsql;
+  EOF
+end
+
+while entry = ARGV.slice!(0)
+  typeFun, typeName = entry.split(":")
+  send(typeFun.to_sym, typeName.to_sym)
+end

data/lib/pg_json/sequel/pg_json.rb

@@ -0,0 +1,41 @@
+class Hash
+  def recursively_symbolize_keys!
+    self.symbolize_keys!
+    self.values.each do |v|
+      if v.is_a? Hash
+        v.recursively_symbolize_keys!
+      elsif v.is_a? Array
+        v.recursively_symbolize_keys!
+      end
+    end
+    self
+  end
+end
+
+class Array
+  def recursively_symbolize_keys!
+    self.each do |item|
+      if item.is_a? Hash
+        item.recursively_symbolize_keys!
+      elsif item.is_a? Array
+        item.recursively_symbolize_keys!
+      end
+    end
+  end
+end
+
+module Sequel
+  module Postgres
+    class Dataset
+      def json
+        if self.first
+          JSON.parse(self.first[:to_json]).tap do |h|
+            h.recursively_symbolize_keys!
+          end
+        else
+          nil
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,47 @@
+--- !ruby/object:Gem::Specification
+name: pg_json
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- Jan Zimmek
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-04-17 00:00:00.000000000 Z
+dependencies: []
+description: bring basic json support to postgresql
+email: jan.zimmek@web.de
+executables:
+- pg_json
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/pg_json/sequel/pg_json.rb
+- bin/pg_json
+homepage: 
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: bring basic json support to postgresql
+test_files: []