diff --git a/app/src/main/java/com/tinyengine/it/config/context/DefaultLoginUserContext.java b/app/src/main/java/com/tinyengine/it/config/context/DefaultLoginUserContext.java index b5add8e6..d0494f7b 100644 --- a/app/src/main/java/com/tinyengine/it/config/context/DefaultLoginUserContext.java +++ b/app/src/main/java/com/tinyengine/it/config/context/DefaultLoginUserContext.java @@ -43,4 +43,7 @@ public int getAppId() { public int getPlatformId() { return 1; } + + @Override + public String getSiteId() { return "1"; } } diff --git a/base/src/main/java/com/tinyengine/it/common/context/LoginUserContext.java b/base/src/main/java/com/tinyengine/it/common/context/LoginUserContext.java index 6f8bde93..a0e8adbd 100644 --- a/base/src/main/java/com/tinyengine/it/common/context/LoginUserContext.java +++ b/base/src/main/java/com/tinyengine/it/common/context/LoginUserContext.java @@ -45,4 +45,6 @@ public interface LoginUserContext { * @return 设计器ID */ public int getPlatformId(); + + public String getSiteId(); } diff --git a/base/src/main/java/com/tinyengine/it/common/handler/MyMetaObjectHandler.java b/base/src/main/java/com/tinyengine/it/common/handler/MyMetaObjectHandler.java index b3673728..b1b2e701 100644 --- a/base/src/main/java/com/tinyengine/it/common/handler/MyMetaObjectHandler.java +++ b/base/src/main/java/com/tinyengine/it/common/handler/MyMetaObjectHandler.java @@ -42,6 +42,8 @@ public void insertFill(MetaObject metaObject) { this.setFieldValByName("lastUpdatedBy", loginUserContext.getLoginUserId(), metaObject); this.setFieldValByName("tenantId", loginUserContext.getTenantId(), metaObject); this.setFieldValByName("renterId", loginUserContext.getRenterId(), metaObject); + this.setFieldValByName("siteId", loginUserContext.getSiteId(), metaObject); + this.setFieldValByName("platformId", loginUserContext.getPlatformId(), metaObject); } @Override diff --git a/base/src/main/java/com/tinyengine/it/controller/BlockGroupController.java b/base/src/main/java/com/tinyengine/it/controller/BlockGroupController.java index 32fd1c6f..12f7dd70 100644 --- a/base/src/main/java/com/tinyengine/it/controller/BlockGroupController.java +++ b/base/src/main/java/com/tinyengine/it/controller/BlockGroupController.java @@ -13,6 +13,7 @@ package com.tinyengine.it.controller; import com.tinyengine.it.common.base.Result; +import com.tinyengine.it.common.context.LoginUserContext; import com.tinyengine.it.common.exception.ExceptionEnum; import com.tinyengine.it.common.exception.ServiceException; import com.tinyengine.it.common.log.SystemControllerLog; @@ -59,6 +60,8 @@ public class BlockGroupController { private BlockGroupService blockGroupService; @Autowired private BlockGroupMapper blockGroupMapper; + @Autowired + private LoginUserContext loginUserContext; /** * 获取区块分组 @@ -111,8 +114,7 @@ public Result> getAllBlockGroups( ) @SystemControllerLog(description = "创建区块分组") @PostMapping("/block-groups/create") - public Result> createBlockGroups(@Valid @RequestBody BlockGroup blockGroup) { - blockGroup.setPlatformId(1); + public Result createBlockGroups(@Valid @RequestBody BlockGroup blockGroup) { return blockGroupService.createBlockGroup(blockGroup); } diff --git a/base/src/main/java/com/tinyengine/it/service/material/BlockGroupService.java b/base/src/main/java/com/tinyengine/it/service/material/BlockGroupService.java index b4d40294..bf6d321c 100644 --- a/base/src/main/java/com/tinyengine/it/service/material/BlockGroupService.java +++ b/base/src/main/java/com/tinyengine/it/service/material/BlockGroupService.java @@ -71,7 +71,7 @@ public interface BlockGroupService { * @param blockGroup the block group * @return the result */ - Result> createBlockGroup(BlockGroup blockGroup); + Result createBlockGroup(BlockGroup blockGroup); /** * 根据ids或者appId获取区块信息 diff --git a/base/src/main/java/com/tinyengine/it/service/material/impl/BlockGroupServiceImpl.java b/base/src/main/java/com/tinyengine/it/service/material/impl/BlockGroupServiceImpl.java index 3f6bb081..aa06da42 100644 --- a/base/src/main/java/com/tinyengine/it/service/material/impl/BlockGroupServiceImpl.java +++ b/base/src/main/java/com/tinyengine/it/service/material/impl/BlockGroupServiceImpl.java @@ -164,7 +164,7 @@ public Integer updateBlockGroupById(BlockGroup blockGroup) { * @return insert number */ @Override - public Result> createBlockGroup(BlockGroup blockGroup) { + public Result createBlockGroup(BlockGroup blockGroup) { List blockGroupsList = blockGroupMapper.queryBlockGroupByCondition(blockGroup); if (blockGroupsList.isEmpty()) { blockGroupMapper.createBlockGroup(blockGroup); @@ -173,7 +173,7 @@ public Result> createBlockGroup(BlockGroup blockGroup) { } // 页面返回数据显示 BlockGroup blockGroupResult = findBlockGroupById(blockGroup.getId()); - return Result.success(Collections.singletonList(blockGroupResult)); + return Result.success(blockGroupResult); } /** @@ -213,7 +213,7 @@ public List getBlockGroupByIdsOrAppId(List ids, Integer app for (Block block : blockGroupTemp.getBlocks()) { BlockCarriersRelation queryParam = new BlockCarriersRelation(); queryParam.setBlockId(block.getId()); - queryParam.setHostId(blockGroup.getId()); + queryParam.setHostId(blockGroupTemp.getId()); queryParam.setHostType(Enums.BlockGroup.BLOCK_GROUP.getValue()); List blockCarriersRelations = blockCarriersRelationMapper.queryBlockCarriersRelationByCondition(queryParam); if (blockCarriersRelations.isEmpty()) { diff --git a/base/src/test/java/com/tinyengine/it/common/handler/MockUserContext.java b/base/src/test/java/com/tinyengine/it/common/handler/MockUserContext.java index e39eebcf..de554cd5 100644 --- a/base/src/test/java/com/tinyengine/it/common/handler/MockUserContext.java +++ b/base/src/test/java/com/tinyengine/it/common/handler/MockUserContext.java @@ -41,4 +41,7 @@ public int getAppId() { public int getPlatformId() { return 1; } + + @Override + public String getSiteId() { return "1"; } } diff --git a/base/src/test/java/com/tinyengine/it/controller/BlockGroupControllerTest.java b/base/src/test/java/com/tinyengine/it/controller/BlockGroupControllerTest.java index 535be04f..8461b4aa 100644 --- a/base/src/test/java/com/tinyengine/it/controller/BlockGroupControllerTest.java +++ b/base/src/test/java/com/tinyengine/it/controller/BlockGroupControllerTest.java @@ -63,10 +63,10 @@ void testGetAllBlockGroups() { @Test void testCreateBlockGroups() { - when(blockGroupService.createBlockGroup(any(BlockGroup.class))).thenReturn(new Result>()); + when(blockGroupService.createBlockGroup(any(BlockGroup.class))).thenReturn(new Result()); - Result> result = blockGroupController.createBlockGroups(new BlockGroup()); - Assertions.assertEquals(new Result>(), result); + Result result = blockGroupController.createBlockGroups(new BlockGroup()); + Assertions.assertEquals(new Result(), result); } @Test diff --git a/base/src/test/java/com/tinyengine/it/service/material/impl/BlockGroupServiceImplTest.java b/base/src/test/java/com/tinyengine/it/service/material/impl/BlockGroupServiceImplTest.java index 24059616..00592696 100644 --- a/base/src/test/java/com/tinyengine/it/service/material/impl/BlockGroupServiceImplTest.java +++ b/base/src/test/java/com/tinyengine/it/service/material/impl/BlockGroupServiceImplTest.java @@ -111,7 +111,7 @@ void testCreateBlockGroup() { when(loginUserContext.getLoginUserId()).thenReturn("1"); BlockGroup blockGroupParam = new BlockGroup(); blockGroupParam.setId(1); - Result> result = blockGroupServiceImpl.createBlockGroup(blockGroupParam); + Result result = blockGroupServiceImpl.createBlockGroup(blockGroupParam); Assertions.assertNotNull(result.getData()); }