pg_json 0.1.1 → 0.1.2

data/bin/pg_json

@@ -4,27 +4,55 @@ require "sequel"
 
 DB = Sequel.connect "postgres://#{ARGV.slice!(0)}", :logger => nil, :max_connections => 1
 
-def unquoted(type)
+def simple(type, &block)
+  
+  block = lambda do
+    "data::text"
+  end unless block
+  
   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);
+  return (case when data is null then 'null' else #{block.call} 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)
+simple :integer
+simple :float
+simple :text do
+  "'"'||data::text||'"'"
 end
+simple :date do
+  "'"'||to_char(data, 'YYYY-MM-DD')::text||'"'"
+end
+simple :time do
+  "'"'||to_char(data, 'HH24:MI:SS')::text||'"'"
+end
+
+# 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
@@ -80,7 +108,6 @@ $$ language plpgsql;
   EOF
 end
 
-while entry = ARGV.slice!(0)
-  typeFun, typeName = entry.split(":")
-  send(typeFun.to_sym, typeName.to_sym)
+while composite = ARGV.slice!(0)
+  send :composite, composite.to_sym
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pg_json
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
   prerelease: 
 platform: ruby
 authors: