chenbiao 1 年間 前
コミット
c1e709c077

+ 7 - 9
netflix-service/src/main/java/com/cyksj/service/claude/impl/OAuthClaudeCodeServiceImpl.java

@@ -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);
     }
 
 }