fres - swagger/codegen#6488 - notNullJacksonAnnotation option

src/main/java/io/swagger/codegen/v3/generators/features/NotNullAnnotationFeatures.java

@@ -0,0 +1,11 @@
+package io.swagger.codegen.v3.generators.features;
+
+public interface NotNullAnnotationFeatures {
+
+ // Language supports generating not Null Jackson Annotation
+ String NOT_NULL_JACKSON_ANNOTATION = "notNullJacksonAnnotation";
+
+ void setNotNullJacksonAnnotation(boolean notNullJacksonAnnotation);
+
+ boolean isNotNullJacksonAnnotation();
+}

src/main/java/io/swagger/codegen/v3/generators/java/AbstractJavaCodegen.java

@@ -9,6 +9,7 @@
 import io.swagger.codegen.v3.CodegenParameter;
 import io.swagger.codegen.v3.CodegenProperty;
 import io.swagger.codegen.v3.generators.DefaultCodegenConfig;
+import io.swagger.codegen.v3.generators.features.NotNullAnnotationFeatures;
 import io.swagger.codegen.v3.generators.handlebars.java.JavaHelper;
 import io.swagger.v3.oas.models.OpenAPI;
 import io.swagger.v3.oas.models.Operation;
@@ -40,6 +41,7 @@
 
 import static io.swagger.codegen.v3.CodegenConstants.HAS_ENUMS_EXT_NAME;
 import static io.swagger.codegen.v3.CodegenConstants.IS_ENUM_EXT_NAME;
+import static io.swagger.codegen.v3.generators.features.NotNullAnnotationFeatures.NOT_NULL_JACKSON_ANNOTATION;
 import static io.swagger.codegen.v3.generators.handlebars.ExtensionHelper.getBooleanValue;
 
 public abstract class AbstractJavaCodegen extends DefaultCodegenConfig {
@@ -81,6 +83,7 @@ public abstract class AbstractJavaCodegen extends DefaultCodegenConfig {
  protected String apiDocPath = "docs/";
  protected String modelDocPath = "docs/";
  protected boolean supportJava6= false;
+ private NotNullAnnotationFeatures notNullOption;
 
  public AbstractJavaCodegen() {
  super();
@@ -150,7 +153,9 @@ public AbstractJavaCodegen() {
  cliOptions.add(new CliOption(CodegenConstants.HIDE_GENERATION_TIMESTAMP, CodegenConstants.HIDE_GENERATION_TIMESTAMP_DESC));
  cliOptions.add(CliOption.newBoolean(WITH_XML, "whether to include support for application/xml content type and include XML annotations in the model (works with libraries that provide support for JSON and XML)"));
  cliOptions.add(CliOption.newBoolean(CodegenConstants.USE_OAS2, CodegenConstants.USE_OAS2_DESC));
-
+ if(this instanceof NotNullAnnotationFeatures){
+ cliOptions.add(CliOption.newBoolean(NOT_NULL_JACKSON_ANNOTATION, "adds @JsonInclude(JsonInclude.Include.NON_NULL) annotation to model classes"));
+ }
  CliOption dateLibrary = new CliOption(DATE_LIBRARY, "Option. Date library to use");
  Map<String, String> dateOptions = new HashMap<String, String>();
  dateOptions.put("java8", "Java 8 native JSR310 (preferred for jdk 1.8+) - note: this also sets \"" + JAVA8_MODE + "\" to true");
@@ -328,6 +333,17 @@ public void processOpts() {
  this.setFullJavaUtil(Boolean.valueOf(additionalProperties.get(FULL_JAVA_UTIL).toString()));
  }
 
+ if (this instanceof NotNullAnnotationFeatures) {
+ notNullOption = (NotNullAnnotationFeatures)this;
+ if (additionalProperties.containsKey(NOT_NULL_JACKSON_ANNOTATION)) {
+ notNullOption.setNotNullJacksonAnnotation(convertPropertyToBoolean(NOT_NULL_JACKSON_ANNOTATION));
+ writePropertyBack(NOT_NULL_JACKSON_ANNOTATION, notNullOption.isNotNullJacksonAnnotation());
+ if (notNullOption.isNotNullJacksonAnnotation()) {
+ importMapping.put("JsonInclude", "com.fasterxml.jackson.annotation.JsonInclude");
+ }
+ }
+ }
+
  if (fullJavaUtil) {
  javaUtilPrefix = "java.util.";
  }
@@ -912,6 +928,16 @@ public CodegenModel fromModel(String name, Schema schema, Map<String, Schema> al
  final CodegenModel parentCodegenModel = super.fromModel(codegenModel.parent, parentModel, allSchemas);
  codegenModel = AbstractJavaCodegen.reconcileInlineEnums(codegenModel, parentCodegenModel);
  }
+ if (this instanceof NotNullAnnotationFeatures) {
+ if (this instanceof NotNullAnnotationFeatures) {
+ notNullOption = (NotNullAnnotationFeatures)this;
+ if (additionalProperties.containsKey(NOT_NULL_JACKSON_ANNOTATION)) {
+ if (notNullOption.isNotNullJacksonAnnotation()) {
+ codegenModel.imports.add("JsonInclude");
+ }
+ }
+ }
+ }
  return codegenModel;
  }
 

src/main/java/io/swagger/codegen/v3/generators/java/JavaClientCodegen.java

@@ -10,6 +10,7 @@
 import io.swagger.codegen.v3.SupportingFile;
 import io.swagger.codegen.v3.generators.features.BeanValidationFeatures;
 import io.swagger.codegen.v3.generators.features.GzipFeatures;
+import io.swagger.codegen.v3.generators.features.NotNullAnnotationFeatures;
 import io.swagger.codegen.v3.generators.features.PerformBeanValidationFeatures;
 import io.swagger.codegen.v3.generators.util.OpenAPIUtil;
 import org.apache.commons.lang3.BooleanUtils;
@@ -32,7 +33,7 @@
 import static io.swagger.codegen.v3.generators.handlebars.ExtensionHelper.getBooleanValue;
 import static java.util.Collections.sort;
 
-public class JavaClientCodegen extends AbstractJavaCodegen implements BeanValidationFeatures, PerformBeanValidationFeatures, GzipFeatures {
+public class JavaClientCodegen extends AbstractJavaCodegen implements BeanValidationFeatures, PerformBeanValidationFeatures, GzipFeatures, NotNullAnnotationFeatures {
  static final String MEDIA_TYPE = "mediaType";
 
  private static final Logger LOGGER = LoggerFactory.getLogger(JavaClientCodegen.class);
@@ -62,6 +63,7 @@ public class JavaClientCodegen extends AbstractJavaCodegen implements BeanValida
  protected boolean performBeanValidation = false;
  protected boolean useGzipFeature = false;
  protected boolean useRuntimeException = false;
+ private boolean notNullJacksonAnnotation = false;
 
 
  public JavaClientCodegen() {
@@ -611,4 +613,13 @@ static boolean isJsonVendorMimeType(String mime) {
  return mime != null && JSON_VENDOR_MIME_PATTERN.matcher(mime).matches();
  }
 
+ @Override
+ public void setNotNullJacksonAnnotation(boolean notNullJacksonAnnotation) {
+ this.notNullJacksonAnnotation = notNullJacksonAnnotation;
+ }
+
+ @Override
+ public boolean isNotNullJacksonAnnotation() {
+ return notNullJacksonAnnotation;
+ }
 }

src/main/java/io/swagger/codegen/v3/generators/java/SpringCodegen.java

@@ -14,6 +14,7 @@
 import io.swagger.codegen.v3.CodegenType;
 import io.swagger.codegen.v3.SupportingFile;
 import io.swagger.codegen.v3.generators.features.BeanValidationFeatures;
+import io.swagger.codegen.v3.generators.features.NotNullAnnotationFeatures;
 import io.swagger.codegen.v3.generators.features.OptionalFeatures;
 import io.swagger.codegen.v3.generators.util.OpenAPIUtil;
 import io.swagger.codegen.v3.utils.URLPathUtil;
@@ -36,7 +37,7 @@
 import static io.swagger.codegen.v3.CodegenConstants.IS_ENUM_EXT_NAME;
 import static io.swagger.codegen.v3.generators.handlebars.ExtensionHelper.getBooleanValue;
 
-public class SpringCodegen extends AbstractJavaCodegen implements BeanValidationFeatures, OptionalFeatures {
+public class SpringCodegen extends AbstractJavaCodegen implements BeanValidationFeatures, OptionalFeatures, NotNullAnnotationFeatures {
  static Logger LOGGER = LoggerFactory.getLogger(SpringCodegen.class);
  public static final String DEFAULT_LIBRARY = "spring-boot";
  public static final String TITLE = "title";
@@ -78,6 +79,7 @@ public class SpringCodegen extends AbstractJavaCodegen implements BeanValidation
  protected boolean defaultInterfaces = true;
  protected String springBootVersion = "1.5.22.RELEASE";
  protected boolean throwsException = false;
+ private boolean notNullJacksonAnnotation = false;
 
  public SpringCodegen() {
  super();
@@ -591,6 +593,16 @@ public void setReturnContainer(final String returnContainer) {
  return objs;
  }
 
+ @Override
+ public void setNotNullJacksonAnnotation(boolean notNullJacksonAnnotation) {
+ this.notNullJacksonAnnotation = notNullJacksonAnnotation;
+ }
+
+ @Override
+ public boolean isNotNullJacksonAnnotation() {
+ return notNullJacksonAnnotation;
+ }
+
  private interface DataTypeAssigner {
  void setReturnType(String returnType);
  void setReturnContainer(String returnContainer);