Problem: accessing original class behind GraphQLObjectType is impossible

Solution: pass the original class to GraphQLObjectTypeWrapper
and add a getter

Author: yrashk

gradle.properties

@@ -1,2 +1,2 @@
-version = 0.13.0
+version = 0.13.1
 group = com.graphql-java

src/main/java/graphql/annotations/GraphQLAnnotations.java

@@ -167,28 +167,13 @@ private Class<?> getDeclaringClass(Method method) {
                                                                          NoSuchMethodException {
         GraphQLObjectType.Builder builder = getObjectBuilder(object);
 
-        return new GraphQLObjectTypeWrapper(builder.build());
+        return new GraphQLObjectTypeWrapper(object, builder.build());
     }
 
     public static GraphQLObjectType object(Class<?> object) throws IllegalAccessException, InstantiationException, NoSuchMethodException {
         return getInstance().getObject(object);
     }
 
-    public static class GraphQLObjectTypeWrapper extends GraphQLObjectType {
-
-        public GraphQLObjectTypeWrapper(GraphQLObjectType objectType) {
-            super(objectType.getName(), objectType.getDescription(), objectType.getFieldDefinitions(),
-                    objectType.getInterfaces());
-        }
-
-        @Override
-        public boolean equals(Object obj) {
-            return obj instanceof GraphQLObjectType &&
-                   ((GraphQLObjectType) obj).getName().contentEquals(getName()) &&
-                   ((GraphQLObjectType) obj).getFieldDefinitions().equals(getFieldDefinitions());
-        }
-    }
-
     public static class GraphQLFieldDefinitionWrapper extends GraphQLFieldDefinition {
 
         public GraphQLFieldDefinitionWrapper(GraphQLFieldDefinition fieldDefinition) {

src/main/java/graphql/annotations/GraphQLObjectTypeWrapper.java

@@ -0,0 +1,23 @@
+package graphql.annotations;
+
+import graphql.schema.GraphQLObjectType;
+import lombok.Getter;
+
+public class GraphQLObjectTypeWrapper extends GraphQLObjectType {
+
+    @Getter
+    private final Class<?> objectClass;
+
+    public GraphQLObjectTypeWrapper(Class<?> objectClass, GraphQLObjectType objectType) {
+        super(objectType.getName(), objectType.getDescription(), objectType.getFieldDefinitions(),
+                objectType.getInterfaces());
+        this.objectClass = objectClass;
+    }
+
+    @Override
+    public boolean equals(Object obj) {
+        return obj instanceof GraphQLObjectType &&
+               ((GraphQLObjectType) obj).getName().contentEquals(getName()) &&
+               ((GraphQLObjectType) obj).getFieldDefinitions().equals(getFieldDefinitions());
+    }
+}

src/test/java/graphql/annotations/GraphQLObjectTest.java

@@ -117,6 +117,13 @@ public void metainformation() {
         assertEquals(object.getDescription(), "TestObject object");
     }
 
+    @Test @SneakyThrows
+    public void objectClass() {
+        GraphQLObjectType object = GraphQLAnnotations.object(TestObject.class);
+        assertTrue(object instanceof GraphQLObjectTypeWrapper);
+        assertEquals(((GraphQLObjectTypeWrapper) object).getObjectClass(), TestObject.class);
+    }
+
     @Test @SneakyThrows
     public void fields() {
         GraphQLObjectType object = GraphQLAnnotations.object(TestObject.class);