码匠社区学习笔记(二)

package com.majiang.community.controller; import cn.hutool.core.util.StrUtil; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.majiang.community.dto.AccessTokenDTO; import com.majiang.community.dto.GiteeUserDTO; import com.majiang.community.mapper.UserMapper; import com.majiang.community.model.User; import com.majiang.community.util.CookieUtil; import com.majiang.community.util.HttpClientUtil; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.RequestParam; import javax.servlet.http.Cookie; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.util.UUID; /** * @date 2022-04-13 */ @Controller public class AuthorizeController { Logger logger=LoggerFactory.getLogger(AuthorizeController.class); @Autowired(required = false) private UserMapper userMapper; @Autowired private CookieUtil cookieUtil; @Autowired private HttpClientUtil httpClientUtil; @Value("${gitee.client.id}") private String client_id; @Value("${gitee.client.secrets}") private String client_secret; @Value("${gitee.redirect_uri}") private String redirect_uri; @GetMapping("/callBack") public String callBack(@RequestParam(name = "state")String state, @RequestParam(name = "code")String code, HttpServletRequest request, HttpServletResponse response){ String cookieValue = cookieUtil.getCookieValue(request); //cookie已经过期 if(StrUtil.isEmpty(cookieValue)){ AccessTokenDTO accessTokenDto = new AccessTokenDTO(); accessTokenDto.setCode(code); accessTokenDto.setClient_id(client_id); accessTokenDto.setRedirect_uri(redirect_uri); accessTokenDto.setClient_secret(client_secret); accessTokenDto.setState(state); //获取token String access_token = httpClientUtil.httpResult(accessTokenDto); //根据token获取用户信息 GiteeUserDTO giteeUserDto = httpClientUtil.getUser(access_token); logger.info(giteeUserDto.getId()+" "+giteeUserDto.getLogin()+" "+giteeUserDto.getBio()+" "+giteeUserDto.getAvatar_url()); //将用户信息存储到数据库 if(giteeUserDto!=null&&StrUtil.isNotEmpty(giteeUserDto.getId().toString())){ QueryWrapper queryWrapper=new QueryWrapper(); queryWrapper.eq("fd_account_id", giteeUserDto.getId().toString()); User selectOne = userMapper.selectOne(queryWrapper); String fdToken = UUID.randomUUID().toString(); if(selectOne!=null){ selectOne.setFdToken(fdToken); userMapper.updateById(selectOne); }else{ User user=new User(); user.setFdName(giteeUserDto.getLogin()); user.setFdToken(fdToken); user.setFdAccountId(giteeUserDto.getId().toString()); user.setGmtCreateTime(System.currentTimeMillis()); user.setAvatarUrl(giteeUserDto.getAvatar_url()); user.setGmtAlterTime(System.currentTimeMillis()); userMapper.insert(user); } Cookie cookie = new Cookie("token", fdToken); response.addCookie(cookie); } } //重定向 return "redirect:/"; } }