pggraphql 0.0.8 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4b016ad8dffbdb2b014ec63cb6b4f2f730a2a514
-  data.tar.gz: 2f99620927aa7043a7130156596dcfd9f1f2d2d1
+  metadata.gz: c0b33e3078467161a88af120f37b4de4fc10e127
+  data.tar.gz: eb9f61d5aa769cab705f3d1194227278fbbe585f
 SHA512:
-  metadata.gz: d7637666111ab189bcd497da5837d5d06e3baf89c0716839481294c160d6a0ca13531de7c2e4ae71ad7e54f12dd55cbf55578736ffe1fddabefff4743bd11703
-  data.tar.gz: dbbdc73800151b6ac3bf6e3535245d412c38ca7f1843fe8c27eabbe14bdbc57b687229c1edac9bab220d5fa03ae5290e9b1f489dd548695da9898be136e5a901
+  metadata.gz: bf40ede2fc8a855cf762dfe9052ba58565696cdbf539a446ad7554c2517389feb1292b089aca5bba7454b6c8eec4fec05e8cb15a2ead57fc69751e2692c55bc7
+  data.tar.gz: d970827a05f6af6e537390062ea00fe5443f4e64cf5ce802e2fd06106a5a0227836cb3fe50e4dde05602b7d91d8e5ebd1fe10cf8caa37350cf54de97c0d51fc4

data/lib/pggraphql/version.rb

@@ -1,3 +1,3 @@
 module Pggraphql
-  VERSION = "0.0.8"
+  VERSION = "0.0.9"
 end

data/lib/pggraphql.rb

@@ -59,7 +59,7 @@ module PgGraphQl
 
           raise "missing :id for root type #{type.name.inspect}" if !ids && level == 1 && !type.null_pk
 
-          wheres << type.pk.call(ids) if ids && type.pk.call(ids)
+          wheres << type.pk.call(ids, level) if ids && type.pk.call(ids, level)
 
           wheres << ("(" + type.filter + ")") if type.filter
 
@@ -115,7 +115,7 @@ module PgGraphQl
         @subtypes = {}
         @mappings = {}
         @null_pk = false
-        @pk = ->(ids) do
+        @pk = ->(ids, level) do
           id_column = @mappings[:id] || "id"
           if ids.is_a?(Array)
             "#{id_column} in (" + ids.map{|id| id.is_a?(String) ? "'#{id}'" : id.to_s}.join(',') + ")"

data/test/test_pggraphql.rb

@@ -97,7 +97,7 @@ module PgGraphQl
     def test_simple_pk_custom
       res = to_sql({user: {id: "1", email: "email"}}) do |s|
         s.root :user
-        s.type :user, fields: [:email], pk: ->(id){ "access_token = '#{id}'" }
+        s.type :user, fields: [:email], pk: ->(id, level){ "access_token = '#{id}'" }
       end
 
       assert_equal token(<<-SQL
@@ -110,6 +110,22 @@ module PgGraphQl
       ), token(res)
     end
 
+    def test_simple_pk_with_level
+      res = to_sql({user: {id: "99", email: "email"}}) do |s|
+        s.root :user
+        s.type :user, fields: [:email], pk: ->(id, level){ "level#{level} = '#{id}'" }
+      end
+
+      assert_equal token(<<-SQL
+        select 'user'::text as key,
+          (select to_json(x.*)
+            from (select id,
+                  email
+                from users where level1 = '99' limit 1) x) as value
+      SQL
+      ), token(res)
+    end
+
     def test_simple_pk_type_handling
       res = to_sql({user: {id: ["1"], email: "email"}}) do |s|
         s.root :user

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pggraphql
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.9
 platform: ruby
 authors:
 - Jan Zimmek
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-03-18 00:00:00.000000000 Z
+date: 2015-03-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json