OMS-GW-企业单位

mhd_oms/src/main/java/com/mhd/oms/application/service/erpEnterpriseUnit/ErpEnterpriseUnitApplicationService.java

@@ -1,15 +1,20 @@
 package com.mhd.oms.application.service.erpEnterpriseUnit;
 
 
+import com.mhd.common.core.utils.bean.BeanUtils;
 import com.mhd.common.core.web.page.TableDataInfo;
+import com.mhd.common.security.utils.SecurityUtils;
+import com.mhd.oms.domain.erpEnterpriseUnit.entity.ErpEnterpriseUnit;
 import com.mhd.oms.domain.erpEnterpriseUnit.repository.mapper.ErpEnterpriseUnitMapper;
 import com.mhd.oms.domain.erpEnterpriseUnit.repository.po.ErpEnterpriseUnitPO;
 import com.mhd.oms.domain.erpEnterpriseUnit.repository.todo.ErpEnterpriseUnitDO;
 import com.mhd.oms.interfaces.dto.erpEnterpriseUnit.ErpEnterpriseUnitDTO;
+import com.mhd.system.api.model.LoginUser;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
+import java.util.Date;
 import java.util.List;
 
 @Service
@@ -27,4 +32,32 @@ public class ErpEnterpriseUnitApplicationService {
         tableDataInfo.setCode(200);
         return tableDataInfo;
     }
+
+    public int update(ErpEnterpriseUnitDTO erpEnterpriseUnitDTO) {
+        ErpEnterpriseUnit erpEnterpriseUnit = new ErpEnterpriseUnit();
+        BeanUtils.copyProperties(erpEnterpriseUnitDTO,erpEnterpriseUnit);
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        if(loginUser != null){
+            erpEnterpriseUnit.setUpdateBy(loginUser.getUserid());
+            erpEnterpriseUnit.setUpdateTime(new Date());
+        }
+        return erpEnterpriseUnitMapper.updateById(erpEnterpriseUnit);
+    }
+
+    public int insert(ErpEnterpriseUnitDTO erpEnterpriseUnitDTO) {
+        ErpEnterpriseUnit erpEnterpriseUnit = new ErpEnterpriseUnit();
+        BeanUtils.copyProperties(erpEnterpriseUnitDTO,erpEnterpriseUnit);
+        erpEnterpriseUnit.setPushStatus(1);
+        erpEnterpriseUnit.setDelFlag(1);
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        if(loginUser != null){
+            erpEnterpriseUnit.setCreateBy(loginUser.getUserid());
+            //erpEnterpriseUnit.setCreateTime(new Date());
+            erpEnterpriseUnit.setOrganizationId(loginUser.getUserPo().getOrganizationId());
+            erpEnterpriseUnit.setTopOrganizationId(loginUser.getUserPo().getTopOrganizationId());
+        }
+        erpEnterpriseUnit.setCreateTime(new Date());
+        return erpEnterpriseUnitMapper.insert(erpEnterpriseUnit);
+
+    }
 }

mhd_oms/src/main/java/com/mhd/oms/interfaces/facade/erpEnterpriseUnit/ErpEnterpriseUnitApi.java

@@ -2,6 +2,7 @@ package com.mhd.oms.interfaces.facade.erpEnterpriseUnit;
 
 
 import com.mhd.common.core.web.controller.BaseController;
+import com.mhd.common.core.web.domain.AjaxResult;
 import com.mhd.common.core.web.page.TableDataInfo;
 import com.mhd.oms.application.service.erpEnterpriseUnit.ErpEnterpriseUnitApplicationService;
 import com.mhd.oms.domain.erpEnterpriseUnit.repository.todo.ErpEnterpriseUnitDO;
@@ -10,9 +11,7 @@ import com.mhd.oms.interfaces.dto.erpEnterpriseUnit.ErpEnterpriseUnitDTO;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
-import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.*;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
 
 import javax.annotation.Resource;
 
@@ -40,4 +39,15 @@ public class ErpEnterpriseUnitApi extends BaseController {
 
     }
 
+
+    @ApiOperation("编辑企业单位")
+    @PostMapping("/edit")
+    public AjaxResult edit(@RequestBody ErpEnterpriseUnitDTO erpEnterpriseUnitDTO){
+        if(erpEnterpriseUnitDTO.getId() !=null){
+            return toAjax(erpEnterpriseUnitApplicationService.update(erpEnterpriseUnitDTO));
+        }else{
+            return toAjax(erpEnterpriseUnitApplicationService.insert(erpEnterpriseUnitDTO));
+        }
+    }
+
 }