com.github.xiaoymin.swaggerbootstrapui.service.SpringAddtionalModelService Maven / Gradle / Ivy
/*
* Copyright (C) 2018 Zhejiang xiaominfo Technology CO.,LTD.
* All rights reserved.
* Official Web Site: http://www.xiaominfo.com.
* Developer Web Site: http://open.xiaominfo.com.
*/
package com.github.xiaoymin.swaggerbootstrapui.service;
import com.fasterxml.classmate.TypeResolver;
import com.github.xiaoymin.swaggerbootstrapui.io.ResourceUtil;
import com.github.xiaoymin.swaggerbootstrapui.model.SpringAddtionalModel;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import java.util.Set;
/***
* some error unresolved,don't use
* @since:swagger-bootstrap-ui 1.9.4
* @author [email protected]
* 2019/06/01 20:58
*/
@Component
public class SpringAddtionalModelService{
@Autowired
private TypeResolver typeResolver;
/***
* 扫描包,获取对象
* @param basePackage 扫描包路径
* @return SpringAddtionModel实例
*/
public SpringAddtionalModel scan(String... basePackage){
if (basePackage==null||basePackage.length==0){
throw new IllegalArgumentException("basePackage can't be empty!!!");
}
SpringAddtionalModel springAddtionalModel=new SpringAddtionalModel();
ResourceUtil resourceUtil=new ResourceUtil();
resourceUtil.find(basePackage);
Set> classSets=resourceUtil.getClasses();
if (classSets==null||classSets.isEmpty()){
throw new IllegalArgumentException("can't find any Models in basePackage");
}
int a=0;
for (Class clazz:classSets){
if (a==0){
springAddtionalModel.setFirst(typeResolver.resolve(clazz));
}else{
springAddtionalModel.add(typeResolver.resolve(clazz));
}
a++;
}
return springAddtionalModel;
}
}