Skip to content

Commit e553e44

Browse files
committed
[app-builder] 修改检视意见,优化 repo 层与 service 层的职责边界。
1 parent 8713e76 commit e553e44

File tree

6 files changed

+12
-12
lines changed

6 files changed

+12
-12
lines changed

app-builder/plugins/aipp-custom-model-center/src/main/java/modelengine/fit/jade/aipp/model/repository/UserModelRepo.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ public interface UserModelRepo {
2525
* @param userId 表示用户标识的 {@link String}。
2626
* @return 该用户可用的模型列表的 {@link List}{@code <}{@link ModelPo}{@code >}。
2727
*/
28-
List<ModelPo> getModelList(String userId);
28+
List<ModelPo> listModelsByUserId(String userId);
2929

3030
/**
3131
* 查询特定的模型访问信息。
@@ -51,7 +51,7 @@ public interface UserModelRepo {
5151
* @param userId 表示用户标识的 {@link String}。
5252
* @return 该用户可用的用户模型列表的 {@link List}{@code <}{@link UserModelPo}{@code >}。
5353
*/
54-
List<UserModelPo> listUserModels(String userId);
54+
List<UserModelPo> listUserModelsByUserId(String userId);
5555

5656
/**
5757
* 根据模型标识列表批量查询模型信息。

app-builder/plugins/aipp-custom-model-center/src/main/java/modelengine/fit/jade/aipp/model/repository/impl/UserModelRepoImpl.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ public UserModelRepoImpl(ModelMapper modelMapper, UserModelMapper userModelMappe
3737
}
3838

3939
@Override
40-
public List<ModelPo> getModelList(String userId) {
40+
public List<ModelPo> listModelsByUserId(String userId) {
4141
return this.getModelPos(this.userModelMapper.listUserModels(userId));
4242
}
4343

@@ -79,7 +79,7 @@ public ModelPo getDefaultModel(String userId) {
7979
}
8080

8181
@Override
82-
public List<UserModelPo> listUserModels(String userId) {
82+
public List<UserModelPo> listUserModelsByUserId(String userId) {
8383
return this.userModelMapper.listUserModels(userId);
8484
}
8585

app-builder/plugins/aipp-custom-model-center/src/main/java/modelengine/fit/jade/aipp/model/service/impl/CustomAippModelCenter.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ public CustomAippModelCenter(UserModelRepo userModelRepo,
5050
@Override
5151
public ModelListDto fetchModelList(String type, String scene, OperationContext context) {
5252
LOG.info("[Custom][fetchModelList] operator={}, type={}, scene={}.", context.getOperator(), type, scene);
53-
List<ModelPo> modelList = this.userModelRepo.getModelList(context.getOperator());
53+
List<ModelPo> modelList = this.userModelRepo.listModelsByUserId(context.getOperator());
5454
if (CollectionUtils.isEmpty(modelList)) {
5555
return this.defaultModelCenter.fetchModelList(type, scene, context);
5656
}

app-builder/plugins/aipp-custom-model-center/src/main/java/modelengine/fit/jade/aipp/model/service/impl/UserModelConfigService.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
public class UserModelConfigService implements UserModelConfig {
4141
private static final Logger log = Logger.get(UserModelConfig.class);
4242
private static final String FITABLE_ID = "aipp.model.service.impl";
43-
public static final String DEFAULT_MODEL_TYPE = "chat_completions";
43+
private static final String DEFAULT_MODEL_TYPE = "chat_completions";
4444
private final UserModelRepo userModelRepo;
4545

4646
/**
@@ -60,7 +60,7 @@ public UserModelConfigService(UserModelRepo userModelRepo) {
6060
@Property(description = "返回该用户可用的模型列表")
6161
public List<UserModelDetailDto> getUserModelList(String userId) {
6262
log.info("start get model list for {}.", userId);
63-
List<UserModelPo> userModelPos = this.userModelRepo.listUserModels(userId);
63+
List<UserModelPo> userModelPos = this.userModelRepo.listUserModelsByUserId(userId);
6464
if (CollectionUtils.isEmpty(userModelPos)) {
6565
log.warn("No user model records found for userId={}.", userId);
6666
return Collections.emptyList();
@@ -130,7 +130,7 @@ public String addUserModel(String userId, String apiKey, String modelName, Strin
130130
@Property(description = "删除用户绑定的模型信息")
131131
public String deleteUserModel(String userId, String modelId) {
132132
log.info("start delete user model for {}.", userId);
133-
List<UserModelPo> userModels = this.userModelRepo.listUserModels(userId);
133+
List<UserModelPo> userModels = this.userModelRepo.listUserModelsByUserId(userId);
134134
if (CollectionUtils.isEmpty(userModels)) {
135135
return "删除模型失败,当前用户没有任何模型记录。";
136136
}

app-builder/plugins/aipp-custom-model-center/src/test/java/modelengine/fit/jade/aipp/model/service/impl/CustomAippModelCenterTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ void shouldGetResultWhenFetchModelListGivenUserModelRepoHasData() {
4949
.type("type")
5050
.tag("tag1")
5151
.build();
52-
Mockito.when(this.userModelRepo.getModelList(userId)).thenReturn(Collections.singletonList(modelPo));
52+
Mockito.when(this.userModelRepo.listModelsByUserId(userId)).thenReturn(Collections.singletonList(modelPo));
5353
OperationContext context = new OperationContext();
5454
context.setOperator(userId);
5555

@@ -78,7 +78,7 @@ void shouldGetDefaultWhenFetchModelListGivenUserModelRepoNoData() {
7878
.tag("tag1")
7979
.build();
8080
ModelAccessInfo model1 = ModelAccessInfo.builder().serviceName("gpt").baseUrl("").tag("").build();
81-
Mockito.when(this.userModelRepo.getModelList(userId)).thenReturn(Collections.emptyList());
81+
Mockito.when(this.userModelRepo.listModelsByUserId(userId)).thenReturn(Collections.emptyList());
8282
ModelListDto expectModelList = ModelListDto.builder()
8383
.models(Collections.singletonList(model1))
8484
.total(1)

app-builder/plugins/aipp-custom-model-center/src/test/java/modelengine/fit/jade/aipp/model/service/impl/UserModelConfigServiceTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ void shouldReturnUserModelListWhenUserHasModels() {
6060

6161
ModelPo modelPo = ModelPo.builder().modelId(modelId).name("gpt").baseUrl("http://testUrl").build();
6262

63-
Mockito.when(userModelRepo.listUserModels(userId)).thenReturn(Collections.singletonList(userModelPo));
63+
Mockito.when(userModelRepo.listUserModelsByUserId(userId)).thenReturn(Collections.singletonList(userModelPo));
6464
Mockito.when(userModelRepo.listModels(Collections.singletonList(modelId)))
6565
.thenReturn(Collections.singletonList(modelPo));
6666

@@ -98,7 +98,7 @@ void shouldDeleteModelWhenItIsNotDefault() {
9898
.createdAt(LocalDateTime.now())
9999
.build();
100100

101-
Mockito.when(userModelRepo.listUserModels(userId)).thenReturn(Collections.singletonList(userModelPo));
101+
Mockito.when(userModelRepo.listUserModelsByUserId(userId)).thenReturn(Collections.singletonList(userModelPo));
102102

103103
String result = userModelConfigService.deleteUserModel(userId, modelId);
104104
assertEquals("删除模型成功。", result);

0 commit comments

Comments
 (0)