whyczh
2021-06-03 18f848916fd1d0e915bb51148fa85609f335aaaa
src/main/java/com/whyc/service/LoginService.java
@@ -4,6 +4,7 @@
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.whyc.dto.Response;
import com.whyc.dto.UserLoginInfo;
import com.whyc.mapper.MenuMapper;
import com.whyc.mapper.RoleMenuMapper;
import com.whyc.mapper.UserMapper;
import com.whyc.pojo.Menu;
@@ -28,6 +29,9 @@
public class LoginService {
    @Resource
    private UserMapper userMapper;
    @Resource
    private MenuMapper menuMapper;
    @Resource
    private RoleMenuMapper roleMenuMapper;
@@ -63,7 +67,12 @@
            info.setUsername(userName);
            info.setTimestamp(System.currentTimeMillis());
            info.setUser(user);
            info.setMenus(menus);
            if ("superadmin".equals(user.getName())){
                List<Menu> menuList = menuMapper.selectList(null);
                info.setMenus(menuList);
            }else{
                info.setMenus(menus);
            }
            return new Response<>().set(1,info);
        }
        return new Response<>().set(0,false);