Fix missing imports for schemaMapping in API interfaces (#24672)

* [core] fix missing imports for schemaMapping with importMapping in API interfaces (#24232)

* trigger CI

* [core] support parameter and response schemaMapping imports and refactor test

Author: SubhamAshok

modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultCodegen.java

@@ -4888,6 +4888,9 @@ public CodegenOperation fromOperation(String path,
                     imports.add(r.baseType);
                 }
 
+                addImportMappedType(imports, r.dataType);
+                addImportMappedType(imports, r.baseType);
+
                 if ("set".equals(r.containerType) && typeMapping.containsKey(r.containerType)) {
                     op.uniqueItems = true;
                     imports.add(typeMapping.get(r.containerType));
@@ -5132,6 +5135,18 @@ public CodegenOperation fromOperation(String path,
         return op;
     }
 
+    /**
+     * Helper method to add an import for a data type if it exists in the importMapping.
+     *
+     * @param imports  The set of imports to add to.
+     * @param dataType The data type to check for a mapping.
+     */
+    protected void addImportMappedType(Set<String> imports, String dataType) {
+        if (imports != null && dataType != null && importMapping.containsKey(dataType)) {
+            imports.add(dataType);
+        }
+    }
+
     public void SortParametersByRequiredFlag(List<CodegenParameter> parameters) {
         Collections.sort(parameters, new Comparator<CodegenParameter>() {
             @Override
@@ -5728,6 +5743,8 @@ public CodegenParameter fromParameter(Parameter parameter, Set<String> imports)
         if (codegenProperty.complexType != null) {
             imports.add(codegenProperty.complexType);
         }
+        addImportMappedType(imports, codegenParameter.dataType);
+        addImportMappedType(imports, codegenParameter.baseType);
 
         codegenParameter.pattern = toRegularExpression(parameterSchema.getPattern());
 

modules/openapi-generator/src/test/java/org/openapitools/codegen/java/spring/SpringCodegenTest.java

@@ -9314,4 +9314,25 @@ public void testReactiveSpringHttpInterfaceSupportListOfStringReturnTypeNoRespon
                 "Mono<Set<String>> getUserIdSet"
         );
     }
+
+    @Test
+    public void issue_24232() throws IOException {
+        Map<String, File> files = generateFromContract(
+                "src/test/resources/3_0/spring/issue_24232.yaml", SPRING_BOOT,
+                Map.of(USE_SPRING_BOOT4, true),
+                codegenConfigurator ->
+                        codegenConfigurator
+                                .addTypeMapping("string+custom", "MyCustomId")
+                                .addSchemaMapping("MyKey", "MyCustomKey")
+                                .addImportMapping("MyCustomId", "org.myorg.MyCustomId")
+                                .addImportMapping("MyCustomKey", "org.myorg.MyCustomKey"));
+
+        JavaFileAssert.assertThat(files.get("SomeApi.java"))
+                .assertMethod("getDummy", "MyCustomId", "MyCustomKey")
+                .toFileAssert()
+                .fileContains("import org.myorg.MyCustomId;", "import org.myorg.MyCustomKey;");
+
+        JavaFileAssert.assertThat(files.get("Dummy.java"))
+                .fileContains("import org.myorg.MyCustomId;", "import org.myorg.MyCustomKey;");
+    }
 }

modules/openapi-generator/src/test/resources/3_0/spring/issue_24232.yaml

@@ -0,0 +1,42 @@
+openapi: 3.0.0
+info:
+  version: "1.0.0"
+  title: mapping
+paths:
+  /some/dummy/endpoint/{id}:
+    get:
+      operationId: getDummy
+      parameters:
+        - name: id
+          in: path
+          required: true
+          description: My Custom Id
+          schema:
+            $ref: '#/components/schemas/MyId'
+        - name: key
+          in: query
+          required: true
+          description: filter on key
+          schema:
+            $ref: '#/components/schemas/MyKey'
+      responses:
+        200:
+          description: Successfully created reverse listings for retail
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/Dummy'
+components:
+  schemas:
+    Dummy:
+      type: object
+      properties:
+        id:
+          $ref: '#/components/schemas/MyId'
+        key:
+          $ref: '#/components/schemas/MyKey'
+    MyId:
+      type: string
+      format: custom
+    MyKey:
+      type: string