Skip to content

Commit cb93ce2

Browse files
Use plain string instead of wrapper
1 parent dacd58a commit cb93ce2

File tree

4 files changed

+23
-36
lines changed

4 files changed

+23
-36
lines changed

src/main/java/org/springframework/data/repository/aot/generate/AotQueryMethodGenerationContext.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@
2828
import org.springframework.core.annotation.MergedAnnotation;
2929
import org.springframework.core.annotation.MergedAnnotationSelectors;
3030
import org.springframework.core.annotation.MergedAnnotations;
31-
import org.springframework.data.repository.aot.generate.VariableNameFactory.VariableName;
3231
import org.springframework.data.repository.core.RepositoryInformation;
3332
import org.springframework.data.repository.query.Parameter;
3433
import org.springframework.data.repository.query.QueryMethod;
@@ -135,7 +134,7 @@ public TypeName getReturnTypeName() {
135134
* @param variableName the intended variable name.
136135
* @return the suggested VariableName
137136
*/
138-
public VariableName suggestLocalVariableName(String variableName) {
137+
public String suggestLocalVariableName(String variableName) {
139138
return variableNameFactory.generateName(variableName);
140139
}
141140

src/main/java/org/springframework/data/repository/aot/generate/LocalVariableNameFactory.java

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
*/
3030
class LocalVariableNameFactory implements VariableNameFactory {
3131

32-
private final MultiValueMap<String, VariableName> variables;
32+
private final MultiValueMap<String, String> variables;
3333

3434
static LocalVariableNameFactory forMethod(MethodMetadata methodMetadata) {
3535
return of(methodMetadata.getMethodArguments().keySet());
@@ -47,24 +47,22 @@ static LocalVariableNameFactory of(Set<String> variables) {
4747

4848
variables = new LinkedMultiValueMap<>();
4949
for (String parameterName : predefinedVariableNames) {
50-
variables.add(parameterName, new VariableName(parameterName));
50+
variables.add(parameterName, parameterName);
5151
}
5252
}
5353

5454
@Override
55-
public VariableName generateName(String suggestedName) {
55+
public String generateName(String suggestedName) {
5656

5757
if (!variables.containsKey(suggestedName)) {
58-
VariableName variableName = new VariableName(suggestedName);
59-
variables.add(suggestedName, variableName);
60-
return variableName;
58+
variables.add(suggestedName, suggestedName);
59+
return suggestedName;
6160
}
6261

6362
String targetName = suggestTargetName(suggestedName);
64-
VariableName variableName = new VariableName(suggestedName, targetName);
65-
variables.add(suggestedName, variableName);
66-
variables.add(targetName, variableName);
67-
return variableName;
63+
variables.add(suggestedName, targetName);
64+
variables.add(targetName, targetName);
65+
return targetName;
6866
}
6967

7068
String suggestTargetName(String suggested) {

src/main/java/org/springframework/data/repository/aot/generate/VariableNameFactory.java

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -21,16 +21,6 @@
2121
*/
2222
public interface VariableNameFactory {
2323

24-
VariableName generateName(String suggestedName);
24+
String generateName(String suggestedName);
2525

26-
record VariableName(String source, String target) {
27-
28-
public VariableName(String source) {
29-
this(source, source);
30-
}
31-
32-
public String name() {
33-
return target;
34-
}
35-
}
3626
}

src/test/java/org/springframework/data/repository/aot/generate/LocalVariableNameFactoryUnitTests.java

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -37,26 +37,26 @@ void beforeEach() {
3737
@Test // GH-3270
3838
void resolvesNameClashesInNames() {
3939

40-
assertThat(variableNameFactory.generateName("name").name()).isEqualTo("name");
41-
assertThat(variableNameFactory.generateName("name").name()).isEqualTo("name_1");
42-
assertThat(variableNameFactory.generateName("name").name()).isEqualTo("name_2");
43-
assertThat(variableNameFactory.generateName("name1").name()).isEqualTo("name1");
44-
assertThat(variableNameFactory.generateName("name3").name()).isEqualTo("name3");
45-
assertThat(variableNameFactory.generateName("name3").name()).isEqualTo("name3_1");
46-
assertThat(variableNameFactory.generateName("name4_1").name()).isEqualTo("name4_1");
47-
assertThat(variableNameFactory.generateName("name4").name()).isEqualTo("name4");
48-
assertThat(variableNameFactory.generateName("name4_1_1").name()).isEqualTo("name4_1_1");
49-
assertThat(variableNameFactory.generateName("name4_1").name()).isEqualTo("name4_1_2");
50-
assertThat(variableNameFactory.generateName("name4_1").name()).isEqualTo("name4_1_3");
40+
assertThat(variableNameFactory.generateName("name")).isEqualTo("name");
41+
assertThat(variableNameFactory.generateName("name")).isEqualTo("name_1");
42+
assertThat(variableNameFactory.generateName("name")).isEqualTo("name_2");
43+
assertThat(variableNameFactory.generateName("name1")).isEqualTo("name1");
44+
assertThat(variableNameFactory.generateName("name3")).isEqualTo("name3");
45+
assertThat(variableNameFactory.generateName("name3")).isEqualTo("name3_1");
46+
assertThat(variableNameFactory.generateName("name4_1")).isEqualTo("name4_1");
47+
assertThat(variableNameFactory.generateName("name4")).isEqualTo("name4");
48+
assertThat(variableNameFactory.generateName("name4_1_1")).isEqualTo("name4_1_1");
49+
assertThat(variableNameFactory.generateName("name4_1")).isEqualTo("name4_1_2");
50+
assertThat(variableNameFactory.generateName("name4_1")).isEqualTo("name4_1_3");
5151
}
5252

5353
@Test // GH-3270
5454
void considersPredefinedNames() {
55-
assertThat(variableNameFactory.generateName("firstname").name()).isEqualTo("firstname_1");
55+
assertThat(variableNameFactory.generateName("firstname")).isEqualTo("firstname_1");
5656
}
5757

5858
@Test // GH-3270
5959
void considersCase() {
60-
assertThat(variableNameFactory.generateName("firstName").name()).isEqualTo("firstName");
60+
assertThat(variableNameFactory.generateName("firstName")).isEqualTo("firstName");
6161
}
6262
}

0 commit comments

Comments
 (0)