From a4a682c38f96741897d47a839301ee521b1585fd Mon Sep 17 00:00:00 2001
From: Andrea Cavalli <andrea@cavallium.it>
Date: Mon, 28 Feb 2022 14:38:57 +0100
Subject: [PATCH] Add needed NotNull annotations

---
 .../java/it/cavallium/data/generator/SourcesGenerator.java | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/main/java/it/cavallium/data/generator/SourcesGenerator.java b/src/main/java/it/cavallium/data/generator/SourcesGenerator.java
index cb51f0c..d54b53d 100644
--- a/src/main/java/it/cavallium/data/generator/SourcesGenerator.java
+++ b/src/main/java/it/cavallium/data/generator/SourcesGenerator.java
@@ -2239,7 +2239,12 @@ public class SourcesGenerator {
 						}
 						returnMapNewInstanceStamentBuilder.add("($T) fields.get(\"" + field + "\")", typeTypes.get(fieldType));
 						returnOfNewInstanceStamentBuilder.add("($T) " + field + "", typeTypes.get(fieldType));
-						ofConstructor.addParameter(typeTypes.get(fieldType),"" + field);
+						var fieldTypeName = typeTypes.get(fieldType);
+						var parameterSpecBuilder = ParameterSpec.builder(fieldTypeName, "" + field);
+						if (!fieldTypeName.isPrimitive()) {
+							parameterSpecBuilder.addAnnotation(NotNull.class);
+						}
+						ofConstructor.addParameter(parameterSpecBuilder.build());
 						if (requiresNotNull) {
 							returnMapNewInstanceStamentBuilder.add(")");
 							returnOfNewInstanceStamentBuilder.add(")");