|
|
@@ -2,6 +2,7 @@ package com.cyksj.service.claude.impl;
|
|
|
|
|
|
import cn.hutool.core.collection.CollectionUtil;
|
|
|
import cn.hutool.json.JSONObject;
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
import com.cyksj.mapper.claudecode.OAuthClientMapper;
|
|
|
@@ -29,6 +30,7 @@ import java.util.List;
|
|
|
public class OAuthClaudeCodeServiceImpl implements OAuthClaudeCodeService {
|
|
|
|
|
|
private final OAuthClientMapper oauthClientMapper;
|
|
|
+
|
|
|
private final OAuthTokenMapper oauthTokenMapper;
|
|
|
|
|
|
private final RedisService redisService;
|
|
|
@@ -101,18 +103,14 @@ public class OAuthClaudeCodeServiceImpl implements OAuthClaudeCodeService {
|
|
|
*/
|
|
|
@Override
|
|
|
public OAuthToken validateToken(String code, String key) {
|
|
|
+ LambdaQueryWrapper<OAuthToken> wrapper = Wrappers.lambdaQuery(OAuthToken.class);
|
|
|
if (StringUtils.isNotBlank(key)) {
|
|
|
- QueryWrapper<OAuthToken> wrapper = new QueryWrapper<>();
|
|
|
- wrapper.eq("key_value", key)
|
|
|
- .eq("status", 1);
|
|
|
- return oauthTokenMapper.selectOne(wrapper);
|
|
|
-
|
|
|
+ wrapper.eq(OAuthToken::getKeyValue, key).eq(OAuthToken::getStatus, 1);
|
|
|
} else {
|
|
|
- QueryWrapper<OAuthToken> wrapper = new QueryWrapper<>();
|
|
|
- wrapper.eq("token", code)
|
|
|
- .eq("status", 1);
|
|
|
- return oauthTokenMapper.selectOne(wrapper);
|
|
|
+ wrapper.eq(OAuthToken::getTokenValue, key).eq(OAuthToken::getStatus, 1);
|
|
|
+
|
|
|
}
|
|
|
+ return oauthTokenMapper.selectOne(wrapper);
|
|
|
}
|
|
|
|
|
|
}
|