package cc.mrbird.febs.gateway.enhance.service.impl;
|
|
import cc.mrbird.febs.common.core.entity.QueryRequest;
|
import cc.mrbird.febs.common.core.entity.constant.StringConstant;
|
import cc.mrbird.febs.common.core.utils.DateUtil;
|
import cc.mrbird.febs.gateway.enhance.entity.RateLimitRule;
|
import cc.mrbird.febs.gateway.enhance.mapper.RateLimitRuleMapper;
|
import cc.mrbird.febs.gateway.enhance.service.RateLimitRuleService;
|
import cc.mrbird.febs.gateway.enhance.service.RouteEnhanceCacheService;
|
import cc.mrbird.febs.gateway.enhance.utils.PageableExecutionUtil;
|
import lombok.RequiredArgsConstructor;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.mongodb.core.ReactiveMongoTemplate;
|
import org.springframework.data.mongodb.core.query.Criteria;
|
import org.springframework.data.mongodb.core.query.Query;
|
import org.springframework.stereotype.Service;
|
import reactor.core.publisher.Flux;
|
import reactor.core.publisher.Mono;
|
|
import java.time.LocalDateTime;
|
import java.util.Arrays;
|
|
/**
|
* @author MrBird
|
*/
|
@Service
|
@RequiredArgsConstructor
|
public class RateLimitRuleServiceImpl implements RateLimitRuleService {
|
|
private final RouteEnhanceCacheService routeEnhanceCacheService;
|
private RateLimitRuleMapper rateLimitRuleMapper;
|
private ReactiveMongoTemplate template;
|
|
@Autowired(required = false)
|
public void setRateLimitRuleMapper(RateLimitRuleMapper rateLimitRuleMapper) {
|
this.rateLimitRuleMapper = rateLimitRuleMapper;
|
}
|
|
@Autowired(required = false)
|
public void setTemplate(ReactiveMongoTemplate template) {
|
this.template = template;
|
}
|
|
@Override
|
public Flux<RateLimitRule> findAll() {
|
return rateLimitRuleMapper.findAll();
|
}
|
|
@Override
|
public Flux<RateLimitRule> findByRequestUriAndRequestMethod(String requestUri, String requestMethod) {
|
return rateLimitRuleMapper.findByRequestUriAndRequestMethod(requestUri, requestMethod);
|
}
|
|
@Override
|
public Flux<RateLimitRule> findPages(QueryRequest request, RateLimitRule rateLimitRule) {
|
Query query = getQuery(rateLimitRule);
|
return PageableExecutionUtil.getPages(query, request, RateLimitRule.class, template);
|
}
|
|
@Override
|
public Mono<Long> findCount(RateLimitRule rateLimitRule) {
|
Query query = getQuery(rateLimitRule);
|
return template.count(query, RateLimitRule.class);
|
}
|
|
@Override
|
public Mono<RateLimitRule> create(RateLimitRule rateLimitRule) {
|
rateLimitRule.setCreateTime(DateUtil.formatFullTime(LocalDateTime.now(), DateUtil.FULL_TIME_SPLIT_PATTERN));
|
return rateLimitRuleMapper.insert(rateLimitRule)
|
.doOnSuccess(routeEnhanceCacheService::saveRateLimitRule);
|
}
|
|
@Override
|
public Mono<RateLimitRule> update(RateLimitRule rateLimitRule) {
|
return this.rateLimitRuleMapper.findById(rateLimitRule.getId())
|
.flatMap(r -> {
|
routeEnhanceCacheService.removeRateLimitRule(r);
|
BeanUtils.copyProperties(rateLimitRule, r);
|
return this.rateLimitRuleMapper.save(r);
|
}).doOnSuccess(routeEnhanceCacheService::saveRateLimitRule);
|
}
|
|
@Override
|
public Flux<RateLimitRule> delete(String ids) {
|
String[] idArray = StringUtils.splitByWholeSeparatorPreserveAllTokens(ids, StringConstant.COMMA);
|
return rateLimitRuleMapper.deleteByIdIn(Arrays.asList(idArray))
|
.doOnNext(routeEnhanceCacheService::removeRateLimitRule);
|
}
|
|
private Query getQuery(RateLimitRule rateLimitRule) {
|
Query query = new Query();
|
Criteria criteria = new Criteria();
|
if (StringUtils.isNotBlank(rateLimitRule.getRequestMethod())) {
|
criteria.and("requestMethod").is(rateLimitRule.getRequestMethod());
|
}
|
if (StringUtils.isNotBlank(rateLimitRule.getRequestUri())) {
|
criteria.and("requestUri").is(rateLimitRule.getRequestUri());
|
}
|
if (StringUtils.isNotBlank(rateLimitRule.getStatus())) {
|
criteria.and("status").is(rateLimitRule.getStatus());
|
}
|
query.addCriteria(criteria);
|
return query;
|
}
|
}
|
