1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package com.puppycrawl.tools.checkstyle.checks.coding;
21
22 import java.util.HashMap;
23 import java.util.Map;
24
25 import com.puppycrawl.tools.checkstyle.FileStatefulCheck;
26 import com.puppycrawl.tools.checkstyle.api.AbstractCheck;
27 import com.puppycrawl.tools.checkstyle.api.DetailAST;
28 import com.puppycrawl.tools.checkstyle.api.FullIdent;
29 import com.puppycrawl.tools.checkstyle.api.TokenTypes;
30 import com.puppycrawl.tools.checkstyle.utils.CheckUtil;
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50 @FileStatefulCheck
51 public class EqualsHashCodeCheck
52 extends AbstractCheck {
53
54
55
56
57
58
59
60
61 public static final String MSG_KEY_HASHCODE = "equals.noHashCode";
62
63
64
65
66
67 public static final String MSG_KEY_EQUALS = "equals.noEquals";
68
69
70 private final Map<DetailAST, DetailAST> objBlockWithEquals = new HashMap<>();
71
72
73 private final Map<DetailAST, DetailAST> objBlockWithHashCode = new HashMap<>();
74
75 @Override
76 public int[] getDefaultTokens() {
77 return getRequiredTokens();
78 }
79
80 @Override
81 public int[] getAcceptableTokens() {
82 return getRequiredTokens();
83 }
84
85 @Override
86 public int[] getRequiredTokens() {
87 return new int[] {TokenTypes.METHOD_DEF};
88 }
89
90 @Override
91 public void beginTree(DetailAST rootAST) {
92 objBlockWithEquals.clear();
93 objBlockWithHashCode.clear();
94 }
95
96 @Override
97 public void visitToken(DetailAST ast) {
98 if (isEqualsMethod(ast)) {
99 objBlockWithEquals.put(ast.getParent(), ast);
100 }
101 else if (isHashCodeMethod(ast)) {
102 objBlockWithHashCode.put(ast.getParent(), ast);
103 }
104 }
105
106
107
108
109
110
111
112 private static boolean isEqualsMethod(DetailAST ast) {
113 final DetailAST modifiers = ast.getFirstChild();
114 final DetailAST parameters = ast.findFirstToken(TokenTypes.PARAMETERS);
115
116 return CheckUtil.isEqualsMethod(ast)
117 && isObjectParam(parameters.getFirstChild())
118 && (ast.findFirstToken(TokenTypes.SLIST) != null
119 || modifiers.findFirstToken(TokenTypes.LITERAL_NATIVE) != null);
120 }
121
122
123
124
125
126
127
128 private static boolean isHashCodeMethod(DetailAST ast) {
129 final DetailAST modifiers = ast.getFirstChild();
130 final DetailAST methodName = ast.findFirstToken(TokenTypes.IDENT);
131 final DetailAST parameters = ast.findFirstToken(TokenTypes.PARAMETERS);
132
133 return "hashCode".equals(methodName.getText())
134 && parameters.getFirstChild() == null
135 && (ast.findFirstToken(TokenTypes.SLIST) != null
136 || modifiers.findFirstToken(TokenTypes.LITERAL_NATIVE) != null);
137 }
138
139
140
141
142
143
144
145 private static boolean isObjectParam(DetailAST paramNode) {
146 final DetailAST typeNode = paramNode.findFirstToken(TokenTypes.TYPE);
147 final FullIdent fullIdent = FullIdent.createFullIdentBelow(typeNode);
148 final String name = fullIdent.getText();
149 return "Object".equals(name) || "java.lang.Object".equals(name);
150 }
151
152 @Override
153 public void finishTree(DetailAST rootAST) {
154 objBlockWithEquals
155 .entrySet().stream().filter(detailASTDetailASTEntry -> {
156 return objBlockWithHashCode.remove(detailASTDetailASTEntry.getKey()) == null;
157 }).forEach(detailASTDetailASTEntry -> {
158 final DetailAST equalsAST = detailASTDetailASTEntry.getValue();
159 log(equalsAST, MSG_KEY_HASHCODE);
160 });
161 objBlockWithHashCode.forEach((key, equalsAST) -> log(equalsAST, MSG_KEY_EQUALS));
162 }
163
164 }