添加复制仓库时，复制用户仓库关系表

pengcheng
1 parent 13ae0bb3
Showing 3 changed files with 11 additions and 0 deletions
src/main/java/com/huaheng/pc/config/warehouse/mapper/WarehouseMapper.java
src/main/java/com/huaheng/pc/config/warehouse/service/WarehouseServiceImpl.java
src/main/resources/mybatis/config/WarehouseMapper.xml
@@ -17,4 +17,6 @@ public interface WarehouseMapper extends BaseMapper&lt;Warehouse&gt; {
  
     int companyCopy(@Param("warehouseCode") String warehouseCode, @Param("newWarehouseCode")String newWarehouseCode);
  
+    int userCopy(@Param("warehouseCode") String warehouseCode, @Param("newWarehouseCode")String newWarehouseCode);
+
 }
 \ No newline at end of file
@@ -163,6 +163,10 @@ public class WarehouseServiceImpl extends ServiceImpl&lt;WarehouseMapper, Warehouse
         if (warehouseMapper.companyCopy(warehouseCode, newWarehouseCode) < 1){
             throw new ServiceException("复制仓库货主关系表失败");
         }
+        /* 复制仓库用户关系*/
+        if (warehouseMapper.userCopy(warehouseCode, newWarehouseCode) < 1){
+            throw new ServiceException("复制仓库货主关系表失败");
+        }
         /* 复制角色*/
         if (!roleService.roleCopy(warehouseCode, newWarehouseCode)){
             throw new ServiceException("复制角色表失败");
@@ -66,4 +66,9 @@
     insert into warehouse_company(warehouseCode,companyId,companyCode)
     select #{newWarehouseCode},companyId,companyCode from warehouse_company where warehouseCode=#{warehouseCode}
   </insert>
+
+  <insert id="userCopy">
+    insert into sys_user_warehouse(userId,warehouseCode)
+    select userId,#{newWarehouseCode} from warehouse_company where warehouseCode=#{warehouseCode}
+  </insert>
 </mapper>
 \ No newline at end of file