package com.jilongda.optometrist.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.optometrist.authority.model.SecUser; import com.jilongda.optometrist.authority.service.SecUserService; import com.jilongda.optometrist.model.TOptometrist; import com.jilongda.optometrist.service.TOptometristService; import com.jilongda.optometrist.service.TOptometryService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; @Component public class LoginInfoUtil { @Autowired private SecUserService secUserService; @Autowired private TOptometristService optometristService; public Integer getUserId(){ String username = JwtTokenUtils.getUsername(); TOptometrist userName = optometristService.getOne(new QueryWrapper().eq("phone", 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)); } } }