Fix builder setters to use the correct Boxed or primitive type

Author: adwsingh

codegen/codegen-core/src/main/java/software/amazon/smithy/java/codegen/generators/BuilderGenerator.java

@@ -159,8 +159,9 @@ protected void generateMemberValueSetters(JavaWriter writer) {
             writer.putContext("type", symbolProvider.toSymbol(member));
             writer.putContext("memberSchema", CodegenUtils.toMemberSchemaName(symbolProvider.toMemberName(member)));
             writer.putContext("schemaUtilsClass", SchemaUtils.class);
+            writer.putContext("isNullable", CodegenUtils.isNullableMember(model, member));
             writer.write(
-                    "case $L -> ${memberName:L}((${type:T}) ${schemaUtilsClass:T}.validateSameMember(${memberSchema:L}, member, value));",
+                    "case $L -> ${memberName:L}((${?isNullable}${type:B}${/isNullable}${^isNullable}${type:N}${/isNullable}) ${schemaUtilsClass:T}.validateSameMember(${memberSchema:L}, member, value));",
                     idx);
             writer.popState();
         }

codegen/codegen-core/src/main/java/software/amazon/smithy/java/codegen/generators/StructureGenerator.java

@@ -754,11 +754,12 @@ protected void generateSetters(JavaWriter writer) {
                 writer.putContext("memberSymbol", symbolProvider.toSymbol(member));
                 writer.putContext("tracked", CodegenUtils.isRequiredWithNoDefault(member));
                 writer.putContext("check", CodegenUtils.requiresSetterNullCheck(symbolProvider, member));
+                writer.putContext("isNullable", CodegenUtils.isNullableMember(model, member));
                 writer.putContext("schemaName", CodegenUtils.toMemberSchemaName(symbolProvider.toMemberName(member)));
 
                 writer.write(
                         """
-                                public Builder ${memberName:L}(${memberSymbol:T} ${memberName:L}) {
+                                public Builder ${memberName:L}(${?isNullable}${memberSymbol:B}${/isNullable}${^isNullable}${memberSymbol:N}${/isNullable} ${memberName:L}) {
                                     this.${memberName:L} = ${?check}${objects:T}.requireNonNull(${/check}${memberName:L}${?check}, "${memberName:L} cannot be null")${/check};${?tracked}
                                     tracker.setMember(${schemaName:L});${/tracked}
                                     return this;

codegen/plugins/server-codegen/src/it/java/software/amazon/smithy/java/codegen/server/ServiceBuilderTest.java

@@ -90,7 +90,7 @@ void testRouting() throws ExecutionException, InterruptedException {
         var output = getBeer.asyncFunction().apply(GetBeerInput.builder().id(1).build(), null);
         assertThat(output.get().getValue())
                 .hasSize(1)
-                .containsExactly(Beer.builder().id(1).name("TestBeer").build());
+                .containsExactly(Beer.builder().id(1L).name("TestBeer").build());
 
         Operation<EchoInput, EchoOutput> echo = service.getOperation("Echo");
         assertThat("Echo").isEqualTo(echo.name());

examples/lambda/src/main/java/software/amazon/smithy/java/example/BeerServiceProvider.java

@@ -33,7 +33,7 @@ public final class BeerServiceProvider implements SmithyServiceProvider {
     private static final Logger LOGGER = Logger.getLogger(BeerServiceProvider.class.getName());
     private static final Service SERVICE;
     private static final Map<String, Beer> FRIDGE = new HashMap<>(
-            Map.of("TXVuaWNoIEhlbGxlcw==", Beer.builder().name("Munich Helles").quantity(1).build()));
+            Map.of("TXVuaWNoIEhlbGxlcw==", Beer.builder().name("Munich Helles").quantity(1L).build()));
     private static final Base64.Encoder ENCODER = Base64.getEncoder();
 
     static {