package com.jilongda.manage.utils; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.jilongda.common.security.JwtTokenUtils; import com.jilongda.manage.authority.model.SecUser; import com.jilongda.manage.authority.service.SecUserService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; @Component public class LoginInfoUtil { @Autowired private SecUserService secUserService; public Long getUserId(){ String username = JwtTokenUtils.getUsername(); SecUser userName = secUserService.getOne(new QueryWrapper().eq("account", username).eq("isDelete",0)); return userName.getId(); } public SecUser getLoginUser(){ String username = JwtTokenUtils.getUsername(); if("admin".equals(username)){ return secUserService.getOne(Wrappers.lambdaQuery(SecUser.class).eq(SecUser::getAccount,username).eq(SecUser::getIsDelete,0)); }else { return secUserService.getOne(Wrappers.lambdaQuery(SecUser.class).eq(SecUser::getPhone,username).eq(SecUser::getIsDelete,0)); } } }