import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.StrUtil;
import com.auth0.jwt.JWT;
import com.auth0.jwt.algorithms.Algorithm;
import com.example.springboot.entity.User;
import com.example.springboot.service.IUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.springframework.web.context.support.RequestHandledEvent;
import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Date;
@Component
public class TokenUtils {
private static IUserService staticUserService;
@Resource
private static IUserService userService;
@PostConstruct
public void setUserService(){
staticUserService=userService;
}
public static String gentoken(String userid,String sign){
return JWT.create().withAudience(userid)//将userid保存在token中
.withExpiresAt(DateUtil.offsetHour(new Date(),2))//两小时token过期
.sign(Algorithm.HMAC256(sign));//以pasword作为密钥
}
//后端获取当前登录用户信息
public static User getCurrentUser(){
try {
HttpServletRequest request=((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
String token=request.getHeader("token");
if (StrUtil.isNotBlank(token))
{
String userid= JWT.decode(token).getAudience().get(0);;
return staticUserService.getById(Integer.valueOf(userid));}
}catch (Exception e){
return null;
}
return null;
}
}