Skip to content

Commit 002adfe

Browse files
owen-obrienrozza
andcommitted
JAVA-5971 Reverse equals check to avoid NPEs (#1806)
Fixes NPE due to BSONIgnore fields introduced in #1610 JAVA-5971 --------- Co-authored-by: Ross Lawley <[email protected]>
1 parent e5adcfd commit 002adfe

File tree

3 files changed

+70
-1
lines changed

3 files changed

+70
-1
lines changed

bson/src/main/org/bson/codecs/pojo/ConventionAnnotationImpl.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
import java.lang.reflect.Type;
3535
import java.util.ArrayList;
3636
import java.util.List;
37+
import java.util.Objects;
3738
import java.util.Map;
3839

3940
import static java.lang.String.format;
@@ -244,7 +245,7 @@ private void cleanPropertyBuilders(final ClassModelBuilder<?> classModelBuilder)
244245
if (!propertyModelBuilder.isReadable() && !propertyModelBuilder.isWritable()) {
245246
propertiesToRemove.add(propertyModelBuilder.getName());
246247
}
247-
if (classModelBuilder.useDiscriminator() && propertyModelBuilder.getReadName().equals(classModelBuilder.getDiscriminatorKey())) {
248+
if (classModelBuilder.useDiscriminator() && Objects.equals(classModelBuilder.getDiscriminatorKey(), propertyModelBuilder.getReadName())) {
248249
propertiesToRemove.add(propertyModelBuilder.getName());
249250
LOGGER.warn(
250251
format(

bson/src/test/unit/org/bson/codecs/pojo/PojoCodecDiscriminatorTest.java

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
import org.bson.codecs.pojo.entities.DiscriminatorModel;
2020
import org.bson.codecs.pojo.entities.DiscriminatorWithGetterModel;
2121
import org.bson.codecs.pojo.entities.DiscriminatorWithProperty;
22+
import org.bson.codecs.pojo.entities.DiscriminatorWithPropertyAndIgnore;
2223
import org.junit.jupiter.api.Test;
2324

2425
import static org.junit.jupiter.api.Assertions.assertArrayEquals;
@@ -70,4 +71,12 @@ public void testDiscriminatorRoundTripWhenItIsAlsoAProperty() {
7071
"{discriminatorKey:'discriminatorValue'}"
7172
);
7273
}
74+
75+
@Test
76+
public void testDiscriminatorRoundTripWhenItIsAlsoAPropertyWithIgnoredProperty() {
77+
roundTrip(
78+
new DiscriminatorWithPropertyAndIgnore(),
79+
"{discriminatorKey:'discriminatorValue'}"
80+
);
81+
}
7382
}
Lines changed: 59 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,59 @@
1+
/*
2+
* Copyright 2008-present MongoDB, Inc.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
package org.bson.codecs.pojo.entities;
18+
19+
import org.bson.codecs.pojo.annotations.BsonDiscriminator;
20+
import org.bson.codecs.pojo.annotations.BsonIgnore;
21+
import org.bson.codecs.pojo.annotations.BsonProperty;
22+
23+
import java.util.Objects;
24+
25+
@BsonDiscriminator(key = "discriminatorKey", value = "discriminatorValue")
26+
public class DiscriminatorWithPropertyAndIgnore {
27+
28+
public DiscriminatorWithPropertyAndIgnore() {
29+
}
30+
31+
@BsonIgnore
32+
public String getFoo() {
33+
return "foo";
34+
}
35+
36+
@BsonProperty("discriminatorKey")
37+
public String getDiscriminator() {
38+
return "discriminatorValue";
39+
}
40+
41+
@Override
42+
public boolean equals(final Object o) {
43+
if (o == null || getClass() != o.getClass()){
44+
return false;
45+
}
46+
final DiscriminatorWithPropertyAndIgnore that = (DiscriminatorWithPropertyAndIgnore) o;
47+
return Objects.equals(getDiscriminator(), that.getDiscriminator());
48+
}
49+
50+
@Override
51+
public int hashCode() {
52+
return Objects.hashCode(getDiscriminator());
53+
}
54+
55+
@Override
56+
public String toString() {
57+
return "DiscriminatorWithProperty{}";
58+
}
59+
}

0 commit comments

Comments
 (0)