package com.ydtech.modules.base.controller; import cn.dev33.satoken.session.SaSession; import cn.dev33.satoken.stp.StpUtil; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.ydtech.core.page.PageResult; import com.ydtech.modules.admin.model.SysUser; import com.ydtech.modules.base.model.dto.PageDto; import lombok.extern.slf4j.Slf4j; import org.apache.poi.ss.formula.functions.T; /** * 基础控制器 * * @author: lig * @date: 2023-06-28 */ @Slf4j public class BaseController { /** * 获取用户 * * @author: lig * @date: 2023年06月28日 0028 */ public SysUser getUser() { SaSession saSession = StpUtil.getSessionByLoginId(StpUtil.getLoginId()); SysUser sysUser = (SysUser) saSession.get(SaSession.USER); return sysUser; } /** * 获取用户ID * * @author: lig * @date: 2023-06-28 */ public String getUserId() { return StpUtil.getLoginId().toString(); } /** * 获取用户名 系统用户名与ID相同 * * @author: lig * @date: 2023-06-28 */ public String getUserName() { return StpUtil.getLoginId().toString(); } /** * 获取部门ID * * @author: lig * @date: 2023年06月28日 0028 */ public String getDeptId() { SaSession saSession = StpUtil.getSessionByLoginId(StpUtil.getLoginId()); SysUser sysUser = (SysUser) saSession.get(SaSession.USER); return sysUser.getDeptId(); } /** * 构造分页 请求 数据 * * @author: lig * @date: 2023年06月28日 0028 */ public Page buildPageRequest(PageDto pageDto){ int pageNum = pageDto.getPageNum() < 1?1: pageDto.getPageNum(); int pageSize = pageDto.getPageSize() <1?1: pageDto.getPageSize(); Page p = new Page<>(); p.setSize(pageSize); p.setCurrent(pageNum); return p; } /** * 构造分页 响应 数据 * * @author: lig * @date: 2023-06-28 */ public PageResult buildPageResponse(IPage page) { PageResult p = new PageResult(); p.setPageNum((int)page.getCurrent()); p.setPageSize((int)page.getSize()); p.setTotalPages((int)page.getPages()); p.setTotalSize((int)page.getTotal()); p.setContent(page.getRecords()) ; return p; } }