package com.tievd.cube.modules.system.service.impl;
|
|
import cn.hutool.core.util.ObjectUtil;
|
import cn.hutool.json.JSONUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.fasterxml.jackson.databind.JsonNode;
|
import com.tievd.cube.modules.system.mapper.SysGatewayRouteMapper;
|
import com.tievd.cube.modules.system.service.ISysGatewayRouteService;
|
import lombok.extern.slf4j.Slf4j;
|
import com.tievd.cube.commons.base.BaseMap;
|
import com.tievd.cube.commons.constant.CacheConst;
|
import com.tievd.cube.modules.system.entity.SysGatewayRoute;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.redis.core.RedisTemplate;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.List;
|
|
@Slf4j
|
@Service
|
@Transactional
|
public class SysGatewayRouteServiceImpl extends ServiceImpl<SysGatewayRouteMapper, SysGatewayRoute> implements ISysGatewayRouteService {
|
|
@Autowired
|
private RedisTemplate<String, Object> redisTemplate;
|
|
@Override
|
public void addRoute2Redis(String key) {
|
List<SysGatewayRoute> ls = this.list(new LambdaQueryWrapper<SysGatewayRoute>().eq(SysGatewayRoute::getStatus, 1));
|
redisTemplate.opsForValue().set(key, JSONUtil.toJsonStr(ls));
|
}
|
|
@Override
|
public void deleteById(String id) {
|
this.removeById(id);
|
this.refreshRouter();
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void updateAll(JsonNode json) {
|
try {
|
json = json.get("router");
|
String id = json.get("id").asText();
|
SysGatewayRoute route = getById(id);
|
if (ObjectUtil.isEmpty(route)) {
|
route = new SysGatewayRoute();
|
}
|
route.setRouterId(json.get("routerId").asText());
|
route.setName(json.get("name").asText());
|
route.setPredicates(json.get("predicates").asText());
|
String filters = json.get("filters").asText();
|
if (ObjectUtil.isEmpty(filters)) {
|
filters = "[]";
|
}
|
route.setFilters(filters);
|
route.setUri(json.get("uri").asText());
|
if (json.get("status") == null) {
|
route.setStatus(1);
|
} else {
|
route.setStatus(json.get("status").asInt());
|
}
|
this.saveOrUpdate(route);
|
} catch (Exception e) {
|
log.error("路由配置解析失败！", e);
|
} finally {
|
refreshRouter();
|
}
|
}
|
|
/**
|
* 更新redis路由缓存
|
*/
|
private void refreshRouter() {
|
// 更新redis路由缓存
|
addRoute2Redis(CacheConst.GATEWAY_ROUTES);
|
BaseMap params = new BaseMap();
|
params.put("handlerName", "refreshRouterListener");
|
// 通过redis消息通知网关刷新路由
|
redisTemplate.convertAndSend("cube_redis_topic", params);
|
}
|
|
@Override
|
public void clearRedis() {
|
redisTemplate.opsForValue().set(CacheConst.GATEWAY_ROUTES, "");
|
}
|
}
|
