|
@@ -10,6 +10,7 @@ import com.cyksj.mapper.UserMapper;
|
|
|
import com.cyksj.model.entity.GptTeamAccount;
|
|
import com.cyksj.model.entity.GptTeamAccount;
|
|
|
import com.cyksj.model.entity.GptTeamUserBindRecord;
|
|
import com.cyksj.model.entity.GptTeamUserBindRecord;
|
|
|
import com.cyksj.model.entity.User;
|
|
import com.cyksj.model.entity.User;
|
|
|
|
|
+import com.cyksj.model.views.GptTeamUserBindRecordView;
|
|
|
import com.cyksj.service.chatgpt.GptTeamAccountService;
|
|
import com.cyksj.service.chatgpt.GptTeamAccountService;
|
|
|
import com.ejlchina.searcher.BeanSearcher;
|
|
import com.ejlchina.searcher.BeanSearcher;
|
|
|
import com.ejlchina.searcher.SearchResult;
|
|
import com.ejlchina.searcher.SearchResult;
|
|
@@ -66,11 +67,11 @@ public class CmsGptTeamAccountController {
|
|
|
userIds.addAll(collect);
|
|
userIds.addAll(collect);
|
|
|
}
|
|
}
|
|
|
if (CollUtil.isNotEmpty(userIds)) {
|
|
if (CollUtil.isNotEmpty(userIds)) {
|
|
|
- List<String> user_tokens = gptTeamUserBindRecordMapper.selectList(Wrappers.lambdaQuery(GptTeamUserBindRecord.class).in(GptTeamUserBindRecord::getUserId, userIds)).stream().map(GptTeamUserBindRecord::getToken).collect(Collectors.toList());
|
|
|
|
|
- if (user_tokens.isEmpty()) {
|
|
|
|
|
|
|
+ List<String> accountIds = gptTeamUserBindRecordMapper.selectList(Wrappers.lambdaQuery(GptTeamUserBindRecord.class).in(GptTeamUserBindRecord::getUserId, userIds)).stream().map(GptTeamUserBindRecord::getAccountId).collect(Collectors.toList());
|
|
|
|
|
+ if (accountIds.isEmpty()) {
|
|
|
return GatewayResponse.SUCCESS.newBuilder().toResult();
|
|
return GatewayResponse.SUCCESS.newBuilder().toResult();
|
|
|
}
|
|
}
|
|
|
- builder.field(GptTeamAccount::getToken, user_tokens).op(Operator.InList);
|
|
|
|
|
|
|
+ builder.field(GptTeamAccount::getAccountId, accountIds).op(Operator.InList);
|
|
|
}
|
|
}
|
|
|
SearchResult<GptTeamAccount> search = beanSearcher.search(GptTeamAccount.class, builder
|
|
SearchResult<GptTeamAccount> search = beanSearcher.search(GptTeamAccount.class, builder
|
|
|
.orderBy(GptTeamAccount::getId).desc()
|
|
.orderBy(GptTeamAccount::getId).desc()
|
|
@@ -82,11 +83,11 @@ public class CmsGptTeamAccountController {
|
|
|
* gpt team账号用户加入列表
|
|
* gpt team账号用户加入列表
|
|
|
*/
|
|
*/
|
|
|
@GetMapping("/get/users")
|
|
@GetMapping("/get/users")
|
|
|
- public Result<List<GptTeamUserBindRecord>> getTeamUsers(String token) {
|
|
|
|
|
- List<GptTeamUserBindRecord> list = beanSearcher.searchAll(GptTeamUserBindRecord.class, MapUtils.flatBuilder(request.getParameterMap())
|
|
|
|
|
- .field(GptTeamUserBindRecord::getToken, token)
|
|
|
|
|
- .field(GptTeamUserBindRecord::getStatus, Boolean.TRUE)
|
|
|
|
|
- .orderBy(GptTeamUserBindRecord::getId).desc()
|
|
|
|
|
|
|
+ public Result<List<GptTeamUserBindRecordView>> getTeamUsers(String accountId) {
|
|
|
|
|
+ List<GptTeamUserBindRecordView> list = beanSearcher.searchAll(GptTeamUserBindRecordView.class, MapUtils.flatBuilder(request.getParameterMap())
|
|
|
|
|
+ .field(GptTeamUserBindRecordView::getAccountId, accountId)
|
|
|
|
|
+ .field(GptTeamUserBindRecordView::getStatus, Boolean.TRUE)
|
|
|
|
|
+ .orderBy(GptTeamUserBindRecordView::getId).desc()
|
|
|
.build());
|
|
.build());
|
|
|
return GatewayResponse.SUCCESS.newBuilder().toResult(list);
|
|
return GatewayResponse.SUCCESS.newBuilder().toResult(list);
|
|
|
}
|
|
}
|
|
@@ -99,6 +100,7 @@ public class CmsGptTeamAccountController {
|
|
|
GptTeamUserBindRecord gptTeamUserBindRecord = gptTeamUserBindRecordMapper.selectById(id);
|
|
GptTeamUserBindRecord gptTeamUserBindRecord = gptTeamUserBindRecordMapper.selectById(id);
|
|
|
if (gptTeamUserBindRecord != null) {
|
|
if (gptTeamUserBindRecord != null) {
|
|
|
gptTeamUserBindRecordMapper.deleteById(id);
|
|
gptTeamUserBindRecordMapper.deleteById(id);
|
|
|
|
|
+ gptTeamAccountService.removeTeamUser(gptTeamUserBindRecord.getAccountId(), gptTeamUserBindRecord.getUserKey());
|
|
|
}
|
|
}
|
|
|
return GatewayResponse.SUCCESS.newBuilder().toResult();
|
|
return GatewayResponse.SUCCESS.newBuilder().toResult();
|
|
|
}
|
|
}
|