Enrichment Service

[nonrtric.git] / enrichment-coordinator-service / src / main / java / org / oransc / enrichment / SwaggerConfig.java

diff --git a/enrichment-coordinator-service/src/main/java/org/oransc/enrichment/SwaggerConfig.java b/enrichment-coordinator-service/src/main/java/org/oransc/enrichment/SwaggerConfig.java
index 07dbefd..ee80183 100644 (file)
--- a/enrichment-coordinator-service/src/main/java/org/oransc/enrichment/SwaggerConfig.java
+++ b/enrichment-coordinator-service/src/main/java/org/oransc/enrichment/SwaggerConfig.java
@@ -23,7 +23,10 @@ package org.oransc.enrichment;
 import com.fasterxml.classmate.TypeResolver;
 import com.google.common.base.Predicates;
 
+import org.oransc.enrichment.clients.ProducerJobInfo;
 import org.oransc.enrichment.controllers.consumer.ConsumerEiJobInfo;
+import org.oransc.enrichment.controllers.producer.ProducerEiTypeInfo;
+import org.oransc.enrichment.controllers.producer.ProducerRegistrationInfo;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
@@ -65,6 +68,9 @@ public class SwaggerConfig extends WebMvcConfigurationSupport {
         return new Docket(DocumentationType.SWAGGER_2) //
             .apiInfo(apiInfo()) //
             .additionalModels(resolver.resolve(ConsumerEiJobInfo.class)) //
+            .additionalModels(resolver.resolve(ProducerRegistrationInfo.class)) //
+            .additionalModels(resolver.resolve(ProducerEiTypeInfo.class)) //
+            .additionalModels(resolver.resolve(ProducerJobInfo.class)) //
             .select() //
             .apis(RequestHandlerSelectors.any()) //
             .paths(PathSelectors.any()) //