Explorar o código

fix 清除claude code tokens

zoujiajian hai 1 ano
pai
achega
51ff3c186b

+ 6 - 4
netflix-service/src/main/java/com/cyksj/service/relation/impl/GroupRelationClearServiceImpl.java

@@ -3,10 +3,8 @@ package com.cyksj.service.relation.impl;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.cyksj.common.constant.Constant;
 import com.cyksj.mapper.*;
-import com.cyksj.model.entity.ClaudeCodeUser;
-import com.cyksj.model.entity.GoodsDonSku;
-import com.cyksj.model.entity.GroupsRelation;
-import com.cyksj.model.entity.UserTicketClearedRecord;
+import com.cyksj.mapper.claudecode.OAuthTokenMapper;
+import com.cyksj.model.entity.*;
 import com.cyksj.model.manage.views.GroupsRelationView;
 import com.cyksj.service.claude.ClaudeCodeService;
 import com.cyksj.service.mange.MirrorUserCleatService;
@@ -45,6 +43,8 @@ public class GroupRelationClearServiceImpl implements GroupRelationClearService
 
 	private final ClaudeCodeUserMapper claudeCodeUserMapper;
 
+	private final OAuthTokenMapper oAuthTokenMapper;
+
 	@Override
 	public void clearTicket(GroupsRelation relation, UserTicketClearedRecord.Source source) {
 		Long relationId = relation.getId();
@@ -142,6 +142,8 @@ public class GroupRelationClearServiceImpl implements GroupRelationClearService
 					if (claudeCodeService.delUserPackages(groupsRelation.getUserId())) {
 						//清除claude code user
 						claudeCodeUserMapper.delete(Wrappers.lambdaQuery(ClaudeCodeUser.class).eq(ClaudeCodeUser::getRelationId, relationId));
+						//清除claude code tokens
+						oAuthTokenMapper.delete(Wrappers.lambdaQuery(OAuthToken.class).eq(OAuthToken::getUserId, userId));
 					}
 				}
 			}