com.github.zuihou.scan.ScanConfiguration Maven / Gradle / Ivy
package com.github.zuihou.scan;
import com.alibaba.fastjson.JSONObject;
import com.github.zuihou.base.R;
import com.github.zuihou.mq.constant.QueueConstants;
import com.github.zuihou.scan.model.SystemApiScanSaveDTO;
import com.github.zuihou.scan.service.SystemApiScanService;
import com.github.zuihou.utils.SpringUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.cloud.openfeign.EnableFeignClients;
import org.springframework.cloud.openfeign.FeignClient;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
/**
* 系统api扫描配置类
*
* @author zuihou
* @date 2019/12/16
*/
@ComponentScan(
basePackages = {
"com.github.zuihou.scan.controller"
}
)
@Import({ScanConfiguration.ScanRabbitConfiguration.class, ScanConfiguration.ScanFeignConfiguration.class})
@Slf4j
public class ScanConfiguration {
@Bean
@ConditionalOnMissingBean(SpringUtils.class)
public SpringUtils getSpringUtils(ApplicationContext applicationContext) {
SpringUtils.setApplicationContext(applicationContext);
return new SpringUtils();
}
@Configuration
@ConditionalOnProperty(name = "zuihou.scan.type", havingValue = "FEIGN")
@EnableFeignClients(basePackageClasses = ScanFeignConfiguration.SystemApiApi.class)
public static class ScanFeignConfiguration {
@Bean("systemApiScanService")
@ConditionalOnMissingBean(SystemApiScanService.class)
public SystemApiScanService getSystemApiService(SystemApiApi systemApiApi) {
return new SystemApiFeignServiceImpl(systemApiApi);
}
/**
* 系统接口
*
* @author zuihou
* @date 2019/12/16
*/
@FeignClient(name = "${zuihou.feign.authority-server:zuihou-authority-server}", path = "/systemApi", fallback = SystemApiApiFallback.class)
public interface SystemApiApi {
/**
* 批量保存
*
* @param data
* @return
*/
@PostMapping("/batch")
R batchSave(@RequestBody SystemApiScanSaveDTO data);
}
public class SystemApiFeignServiceImpl implements SystemApiScanService {
private SystemApiApi systemApiApi;
public SystemApiFeignServiceImpl(SystemApiApi systemApiApi) {
this.systemApiApi = systemApiApi;
}
@Override
public Boolean batchSave(SystemApiScanSaveDTO data) {
R result = systemApiApi.batchSave(data);
return result.getIsSuccess() ? result.getData() : false;
}
}
/**
* 熔断
*
* @author zuihou
* @date 2019/12/16
*/
@Component
public class SystemApiApiFallback implements SystemApiApi {
@Override
public R batchSave(SystemApiScanSaveDTO data) {
return R.timeout();
}
}
}
/**
* 使用队列时，消费者需要自行实现
*/
@ConditionalOnProperty(name = "zuihou.scan.type", havingValue = "RABBIT")
public static class ScanRabbitConfiguration {
@Bean("systemApiScanService")
@ConditionalOnMissingBean(SystemApiScanService.class)
public SystemApiScanService getSystemApiService() {
return new SystemApiRabbitServiceImpl();
}
public class SystemApiRabbitServiceImpl implements SystemApiScanService {
@Autowired
private RabbitTemplate rabbitTemplate;
@Override
public Boolean batchSave(SystemApiScanSaveDTO data) {
try {
rabbitTemplate.convertAndSend(QueueConstants.QUEUE_SCAN_API_RESOURCE, JSONObject.toJSONString(data));
return true;
} catch (Exception e) {
log.warn("发送消息失败", e);
return false;
}
}
}
}
}