pggraphql 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: fdf0b9b66504b4787e8ec3e6928426228b531d61
-  data.tar.gz: c1435f3aaa8d96b2e8d420494d378072a6cded71
+  metadata.gz: ce3ae8e16af5b6972f61cacef41f24d9b19f8672
+  data.tar.gz: bec017114b149de19a7cdca9a4f5e9d0663e7086
 SHA512:
-  metadata.gz: 3d3b0514e72ca3c1dbdc2f92902c1b67537192c0970d82e950b8137b7a115eb2dc03ae13b4159041a890a028849d9df303bd9b7128658ab6b4a61c789ba934e2
-  data.tar.gz: 7f9dc9e43010a5cf019ba8567077b2751e8de9d22dbe16beb4e21afd0f547cf3df8a881a9b9caff547a9ca651e8078abf42811804331ec9527f43b81fc929a8f
+  metadata.gz: 0aa31bbf4d066e4d2265f410e00436dcf773488c5723cacf339b8c13221e948736398d96647c869e0605e03091de3047537d4b30a8bb394de2a0de8559f56cc0
+  data.tar.gz: 448db0dd34a71332eec63993039d423ea752df5b2f05521f35b9065c82644da8f4d852939f84e3450f2febf2ff32584b0b4db127b084096e62c877b2992b2049

data/lib/pggraphql/version.rb

@@ -1,3 +1,3 @@
 module Pggraphql
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

data/lib/pggraphql.rb

@@ -54,11 +54,7 @@ module PgGraphQl
           wheres = []
 
           if ids && ids.to_s != "id"
-            if ids.is_a?(Array)
-              wheres << "id in (#{ids.join(',')})" unless ids.empty?
-            else
-              wheres << "id = #{ids}"
-            end
+            wheres << type.pk.call(ids) if type.pk.call(ids)
           end
 
           wheres << ("(" + type.filter + ")") if type.filter
@@ -102,7 +98,7 @@ module PgGraphQl
     end
 
     class Type
-      attr_accessor :name, :table, :filter, :links, :order_by, :fields, :subtypes
+      attr_accessor :name, :table, :filter, :links, :order_by, :fields, :subtypes, :pk
       attr_reader :schema, :mappings
       def initialize(schema, name)
         @schema = schema
@@ -114,6 +110,17 @@ module PgGraphQl
         @links = {}
         @subtypes = {}
         @mappings = {}
+        @pk = ->(ids) do
+          if ids.is_a?(Array)
+            if ids.empty?
+              nil
+            else
+              "id in (#{ids.join(',')})"
+            end
+          else
+            "id = #{ids}"
+          end
+        end
       end
       def map(field, column_expr)
         @mappings[field] = column_expr

data/test/test_pggraphql.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pggraphql
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Jan Zimmek