From 3725c177bb38ee248d0769eebddbc13379cbdc98 Mon Sep 17 00:00:00 2001 From: 无关风月 <443237572@qq.com> Date: 星期三, 08 一月 2025 09:21:24 +0800 Subject: [PATCH] bug修改 --- ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/service/impl/MenuServiceImpl.java | 30 +++++++++++++++++++++--------- 1 files changed, 21 insertions(+), 9 deletions(-) diff --git a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/service/impl/MenuServiceImpl.java b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/service/impl/MenuServiceImpl.java index 065451f..d7d6c1d 100644 --- a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/service/impl/MenuServiceImpl.java +++ b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/service/impl/MenuServiceImpl.java @@ -2,17 +2,17 @@ import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.ruoyi.admin.entity.Menu; +import com.ruoyi.admin.entity.SysUser; import com.ruoyi.admin.mapper.MenuMapper; +import com.ruoyi.admin.mapper.SysUserMapper; import com.ruoyi.admin.service.MenuService; import com.ruoyi.common.core.constant.RedisConstants; +import com.ruoyi.common.security.service.TokenService; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.stereotype.Service; import javax.annotation.Resource; -import java.util.ArrayList; -import java.util.Collection; -import java.util.List; -import java.util.Map; +import java.util.*; import java.util.concurrent.TimeUnit; import java.util.stream.Collectors; @@ -29,17 +29,29 @@ @Resource private RedisTemplate<Object, Object> redisTemplate; - + @Resource + private SysUserMapper sysUserMapper; + @Resource + private TokenService tokenService; @Override public List<Menu> menuTree() { // redis缓存 Map<Object, Object> regionList = redisTemplate.opsForHash().entries(RedisConstants.MENU_TREE); // 获取所有地区信息 省市区三级 List<Menu> regions = getReginList(regionList); - // 所有地区 - Map<String, Menu> courseTypeMap = regions.stream(). - collect(Collectors.toMap(region -> region.getMenuId().toString() - , region -> region)); + Map<String, Menu> courseTypeMap = new HashMap<>(); + Long userid = tokenService.getLoginUser().getUserid(); + SysUser sysUser = sysUserMapper.selectById(userid); + if (sysUser.getFranchiseeId()==null){ + // 加盟商才有加盟商余额管理 + courseTypeMap = regions.stream().filter(region -> region.getMenuId() != 88900&& + region.getMenuId() != 88901&& + region.getMenuId() != 88902&& + region.getMenuId() != 88903). + collect(Collectors.toMap(region -> region.getMenuId().toString() + , region -> region)); + } + redisTemplate.opsForHash().putAll(RedisConstants.MENU_TREE, courseTypeMap); redisTemplate.expire(RedisConstants.MENU_TREE, 30, TimeUnit.MINUTES); // 生成map集合 -- Gitblit v1.7.1