|
|
|
|
@ -0,0 +1,64 @@
|
|
|
|
|
package cn.iocoder.yudao.module.system.service.oauth2;
|
|
|
|
|
|
|
|
|
|
import cn.hutool.core.util.IdUtil;
|
|
|
|
|
import cn.iocoder.yudao.framework.common.util.date.DateUtils;
|
|
|
|
|
import cn.iocoder.yudao.module.system.dal.dataobject.auth.OAuth2CodeDO;
|
|
|
|
|
import cn.iocoder.yudao.module.system.dal.mysql.oauth2.OAuth2CodeMapper;
|
|
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
import org.springframework.validation.annotation.Validated;
|
|
|
|
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
|
|
|
import java.util.Calendar;
|
|
|
|
|
import java.util.List;
|
|
|
|
|
|
|
|
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
|
|
|
|
import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.OAUTH2_CODE_EXPIRE;
|
|
|
|
|
import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.OAUTH2_CODE_NOT_EXISTS;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* OAuth2.0 授权码 Service 实现类
|
|
|
|
|
*
|
|
|
|
|
* @author 芋道源码
|
|
|
|
|
*/
|
|
|
|
|
@Service
|
|
|
|
|
@Validated
|
|
|
|
|
public class OAuth2CodeServiceImpl implements OAuth2CodeService {
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 授权码的过期时间，默认 5 分钟
|
|
|
|
|
*/
|
|
|
|
|
private static final Integer TIMEOUT = 5 * 60;
|
|
|
|
|
|
|
|
|
|
@Resource
|
|
|
|
|
private OAuth2CodeMapper oauth2CodeMapper;
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public OAuth2CodeDO createAuthorizationCode(Long userId, Integer userType, String clientId, List<String> scopes,
|
|
|
|
|
String redirectUri, String state) {
|
|
|
|
|
OAuth2CodeDO codeDO = new OAuth2CodeDO().setCode(generateCode())
|
|
|
|
|
.setUserId(userId).setUserType(userType)
|
|
|
|
|
.setClientId(clientId).setScopes(scopes)
|
|
|
|
|
.setExpiresTime(DateUtils.addDate(Calendar.SECOND, TIMEOUT))
|
|
|
|
|
.setRedirectUri(redirectUri).setState(state);
|
|
|
|
|
oauth2CodeMapper.insert(codeDO);
|
|
|
|
|
return codeDO;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public OAuth2CodeDO consumeAuthorizationCode(String code) {
|
|
|
|
|
OAuth2CodeDO codeDO = oauth2CodeMapper.selectByCode(code);
|
|
|
|
|
if (codeDO == null) {
|
|
|
|
|
throw exception(OAUTH2_CODE_NOT_EXISTS);
|
|
|
|
|
}
|
|
|
|
|
if (DateUtils.isExpired(codeDO.getExpiresTime())) {
|
|
|
|
|
throw exception(OAUTH2_CODE_EXPIRE);
|
|
|
|
|
}
|
|
|
|
|
oauth2CodeMapper.deleteById(codeDO.getId());
|
|
|
|
|
return codeDO;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static String generateCode() {
|
|
|
|
|
return IdUtil.fastSimpleUUID();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
YunaiV
4 years ago