18
18
import cn .hutool .jwt .JWTPayload ;
19
19
import cn .hutool .jwt .JWTUtil ;
20
20
import com .robin .basis .dto .LoginUserDTO ;
21
- import com .robin .basis .dto .SysMenuDTO ;
21
+ import com .robin .basis .dto .RouterDTO ;
22
22
import com .robin .basis .model .system .SysResource ;
23
23
import com .robin .basis .sercurity .SysLoginUser ;
24
24
import com .robin .basis .service .system .SysResourceService ;
@@ -168,18 +168,18 @@ public Map<String, Object> getRouter(HttpServletRequest request) {
168
168
Map <String , Object > retMap = new HashMap <>();
169
169
SysLoginUser loginUser =SecurityUtils .getLoginUser ();
170
170
if (loginUser != null ) {
171
- List <SysMenuDTO > routers = getMenuList (loginUser .getId ());
171
+ List <RouterDTO > routers = getMenuList (loginUser .getId ());
172
172
retMap = wrapObject (routers );
173
173
} else {
174
174
wrapError (retMap , "not login" );
175
175
}
176
176
return retMap ;
177
177
}
178
- public List <SysMenuDTO > getMenuList (Long userId ) {
178
+ public List <RouterDTO > getMenuList (Long userId ) {
179
179
List <SysResource > allList = resourceService .getAllValidate ();
180
- List <SysMenuDTO > dtoList = allList .stream ().map (SysMenuDTO ::fromVO ).collect (Collectors .toList ());
181
- Map <Long , SysMenuDTO > dtoMap = dtoList .stream ().collect (Collectors .toMap (SysMenuDTO ::getId , Function .identity ()));
182
- SysMenuDTO root = new SysMenuDTO ();
180
+ List <RouterDTO > dtoList = allList .stream ().map (RouterDTO ::fromVO ).collect (Collectors .toList ());
181
+ Map <Long , RouterDTO > dtoMap = dtoList .stream ().collect (Collectors .toMap (RouterDTO ::getId , Function .identity ()));
182
+ RouterDTO root = new RouterDTO ();
183
183
dtoMap .put (0L , root );
184
184
Map <Long , Integer > readMap = new HashMap <>();
185
185
@@ -190,11 +190,11 @@ public List<SysMenuDTO> getMenuList(Long userId) {
190
190
jdbcDao .queryBySelectId (query1 );
191
191
try {
192
192
if (!query1 .getRecordSet ().isEmpty ()) {
193
- Map <Long , List <SysMenuDTO >> aMap = query1 .getRecordSet ().stream ().map (SysMenuDTO ::fromMap ).collect (Collectors .groupingBy (SysMenuDTO ::getPid , Collectors .toList ()));
193
+ Map <Long , List <RouterDTO >> aMap = query1 .getRecordSet ().stream ().map (RouterDTO ::fromMap ).collect (Collectors .groupingBy (RouterDTO ::getPid , Collectors .toList ()));
194
194
195
- List <SysMenuDTO > tops = aMap .get (0L );
196
- tops .sort (Comparator .comparing (SysMenuDTO ::getSeqNo ));
197
- for (SysMenuDTO dto : tops ) {
195
+ List <RouterDTO > tops = aMap .get (0L );
196
+ tops .sort (Comparator .comparing (RouterDTO ::getSeqNo ));
197
+ for (RouterDTO dto : tops ) {
198
198
if (!readMap .containsKey (dto .getId ()) && !dto .getAssignType ().equals (Const .RESOURCE_ASSIGN_DENIED )) {
199
199
if (dtoMap .get (dto .getPid ()).getChildren ()==null ){
200
200
dtoMap .get (dto .getPid ()).setChildren (new ArrayList <>());
@@ -212,10 +212,10 @@ public List<SysMenuDTO> getMenuList(Long userId) {
212
212
return dtoMap .get (0L ).getChildren ();
213
213
}
214
214
215
- private void doScanChildren (Map <Long , SysMenuDTO > cmap , Map <Long , List <SysMenuDTO >> pMap , SysMenuDTO dto , Map <Long , Integer > readMap ) {
215
+ private void doScanChildren (Map <Long , RouterDTO > cmap , Map <Long , List <RouterDTO >> pMap , RouterDTO dto , Map <Long , Integer > readMap ) {
216
216
if (!CollectionUtils .isEmpty (pMap .get (dto .getId ()))) {
217
- pMap .get (dto .getId ()).sort (Comparator .comparing (SysMenuDTO ::getSeqNo ));
218
- for (SysMenuDTO childs : pMap .get (dto .getId ())) {
217
+ pMap .get (dto .getId ()).sort (Comparator .comparing (RouterDTO ::getSeqNo ));
218
+ for (RouterDTO childs : pMap .get (dto .getId ())) {
219
219
if (!readMap .containsKey (childs .getId ()) && !childs .getAssignType ().equals (Const .RESOURCE_ASSIGN_DENIED )) {
220
220
if (cmap .get (childs .getPid ()).getChildren ()==null ){
221
221
cmap .get (childs .getPid ()).setChildren (new ArrayList <>());
0 commit comments