Skip to content

Commit 74d3b30

Browse files
committed
fix: fix object list param generate error
1 parent 78ffdcc commit 74d3b30

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/main/java/io/github/linyimin/plugin/service/SqlParamGenerateService.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,15 +170,15 @@ private String parseParamNameTypeList(List<ParamNameType> paramNameTypes) {
170170
param.put(name, getPrimitiveDefaultValue(name, paramType));
171171
} else {
172172
Map<String, Object> classParam = getFieldFromClass(psiClass);
173-
if (paramNameType.isParamAnnotation) {
173+
if (paramNameType.isParamAnnotation || isArray(type)) {
174174
param.put(name, classParam);
175175
} else {
176176
param.putAll(classParam);
177177
}
178178
}
179179

180180
// 数组或者列表
181-
if (type instanceof PsiArrayType || CollectionUtils.isCollectionClassOrInterface(type)) {
181+
if (isArray(type)) {
182182
param.put(name, Lists.newArrayList(param.get(name)));
183183
}
184184

@@ -188,6 +188,10 @@ private String parseParamNameTypeList(List<ParamNameType> paramNameTypes) {
188188
return JSON.toJSONString(params, true);
189189
}
190190

191+
private boolean isArray(PsiType type) {
192+
return type instanceof PsiArrayType || CollectionUtils.isCollectionClassOrInterface(type);
193+
}
194+
191195
private Map<String, Object> getFieldFromClass(PsiClass psiClass) {
192196
Map<String, Object> param = new HashMap<>();
193197

0 commit comments

Comments
 (0)