added description for $ref

urmuzov · urmuzov · commit 239e96d90715 · 2017-05-23T21:38:04.000+05:00
diff --git a/endpoints-framework/src/main/java/com/google/api/server/spi/config/model/SchemaRepository.java b/endpoints-framework/src/main/java/com/google/api/server/spi/config/model/SchemaRepository.java
@@ -195,10 +195,6 @@ private Schema createBeanSchema(
     Schema.Builder builder = Schema.builder()
         .setName(Types.getSimpleName(type, config.getSerializationConfig()))
         .setType("object");
-    final Description description = type.getRawType().getAnnotation(Description.class);
-    if (description != null) {
-      builder.setDescription(description.value());
-    }
     ResourceSchema schema = resourceSchemaProvider.getResourceSchema(type, config);
     for (Entry<String, ResourcePropertySchema> entry : schema.getProperties().entrySet()) {
       String propertyName = entry.getKey();
diff --git a/endpoints-framework/src/main/java/com/google/api/server/spi/discovery/DiscoveryGenerator.java b/endpoints-framework/src/main/java/com/google/api/server/spi/discovery/DiscoveryGenerator.java
@@ -243,7 +243,9 @@ private JsonSchema convertToDiscoverySchema(Schema schema) {
 
   private JsonSchema convertToDiscoverySchema(Field f) {
     if (f.schemaReference() != null) {
-      return new JsonSchema().set$ref(f.schemaReference().get().name());
+      return new JsonSchema()
+          .setDescription(f.description())
+          .set$ref(f.schemaReference().get().name());
     }
     JsonSchema fieldSchema = new JsonSchema()
         .setType(f.type().getDiscoveryType())
