Skip to content

Commit 68350ce

Browse files
committed
fix: open api support Collection.class
1 parent 44140ff commit 68350ce

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

dongtai-api-gather/dongtai-api-gather-openapi/src/main/java/io/dongtai/iast/api/openapi/convertor/CollectionOpenApiSchemaConvertor.java

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
import io.dongtai.iast.api.openapi.domain.Schema;
44

5+
import java.util.Collection;
6+
57
/**
68
* 用于Java内置的集合类型的转换,比如List、Set、Map
79
*
@@ -31,10 +33,15 @@ public boolean canConvert(Class clazz) {
3133
return clazz != null && (
3234
listOpenApiSchemaConvertor.canConvert(clazz) ||
3335
setOpenApiSchemaConvertor.canConvert(clazz) ||
34-
mapOpenApiSchemaConvertor.canConvert(clazz)
36+
mapOpenApiSchemaConvertor.canConvert(clazz) ||
37+
isCollectionClass(clazz)
3538
);
3639
}
3740

41+
private boolean isCollectionClass(Class clazz) {
42+
return clazz != null && Collection.class.isAssignableFrom(clazz);
43+
}
44+
3845
@Override
3946
public Schema convert(Class clazz) {
4047
if (listOpenApiSchemaConvertor.canConvert(clazz)) {
@@ -43,6 +50,8 @@ public Schema convert(Class clazz) {
4350
return setOpenApiSchemaConvertor.convert(clazz);
4451
} else if (mapOpenApiSchemaConvertor.canConvert(clazz)) {
4552
return mapOpenApiSchemaConvertor.convert(clazz);
53+
} else if (isCollectionClass(clazz)) {
54+
return listOpenApiSchemaConvertor.convert(clazz);
4655
} else {
4756
return null;
4857
}

0 commit comments

Comments
 (0)