-
Notifications
You must be signed in to change notification settings - Fork 12
/
JdbcRouterManger.java
108 lines (85 loc) · 4.6 KB
/
JdbcRouterManger.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package com.weweibuy.gateway.router.dynamic;
import com.weweibuy.gateway.router.mapper.*;
import com.weweibuy.gateway.router.model.example.*;
import com.weweibuy.gateway.router.model.po.*;
import com.weweibuy.gateway.router.model.vo.FilterVo;
import com.weweibuy.gateway.router.model.vo.PredicateVo;
import com.weweibuy.gateway.router.model.vo.RouterVo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
/**
* @author durenhao
* @date 2019/5/20 22:41
**/
@Slf4j
public class JdbcRouterManger {
@Autowired
private GatewayRouterMapper routerMapper;
@Autowired
private RouterPredicateMapper predicateMapper;
@Autowired
private RouterPredicateArgsMapper predicateArgsMapper;
@Autowired
private RouterFilterMapper routerFilterMapper;
@Autowired
private RouterFilterArgsMapper filterArgsMapper;
public List<RouterVo> getAllRouter() {
log.info("【路由加载】>>> 从DB中捞取路由数据");
GatewayRouterExample example = new GatewayRouterExample();
example.createCriteria()
.andIsDeleteEqualTo(false);
List<GatewayRouter> gatewayRouters = routerMapper.selectByExample(example);
RouterPredicateExample predicateExample = new RouterPredicateExample();
return gatewayRouters.stream()
.map(router -> {
String routerId = router.getRouterId();
predicateExample.createCriteria()
.andRouterIdEqualTo(routerId)
.andIsDeleteEqualTo(false);
// 断言
List<RouterPredicate> routerPredicates = predicateMapper.selectByExample(predicateExample);
predicateExample.clear();
RouterPredicateArgsExample predicateArgsExample = new RouterPredicateArgsExample();
// 获取断言
List<PredicateVo> predicateVoList = routerPredicates.stream()
.map(predicate -> {
predicateArgsExample.createCriteria()
.andPredicateIdEqualTo(predicate.getPredicateId())
.andIsDeleteEqualTo(false);
// 断言参数
List<RouterPredicateArgs> predicateArgs = predicateArgsMapper.selectByExample(predicateArgsExample);
predicateArgsExample.clear();
Map<String, String> predicateArgsMap = predicateArgs.stream()
.collect(Collectors.toMap(RouterPredicateArgs::getArgsName, RouterPredicateArgs::getArgsValue));
return PredicateVo.convert(predicate, predicateArgsMap);
})
.collect(Collectors.toList());
RouterFilterExample routerFilterExample = new RouterFilterExample();
routerFilterExample.createCriteria()
.andRouterIdEqualTo(routerId)
.andIsDeleteEqualTo(false);
// 过滤器
List<RouterFilter> routerFilters = routerFilterMapper.selectByExample(routerFilterExample);
routerFilterExample.clear();
RouterFilterArgsExample filterArgsExample = new RouterFilterArgsExample();
List<FilterVo> filterVoList = routerFilters.stream()
.map(filter -> {
filterArgsExample.createCriteria()
.andFilterIdEqualTo(filter.getFilterId())
.andIsDeleteEqualTo(false);
List<RouterFilterArgs> filterArgs = filterArgsMapper.selectByExample(filterArgsExample);
filterArgsExample.clear();
// 过滤器参数
Map<String, String> filterArgsMap = filterArgs.stream()
.collect(Collectors.toMap(RouterFilterArgs::getArgsName, RouterFilterArgs::getArgsValue));
return FilterVo.convert(filter, filterArgsMap);
})
.collect(Collectors.toList());
return RouterVo.convert(router, predicateVoList, filterVoList);
})
.collect(Collectors.toList());
}
}