From b23f33df3e51ff1bfb1edde4e2aa2624f46de0d7 Mon Sep 17 00:00:00 2001 From: Burce Date: Fri, 18 Oct 2019 15:46:30 +0800 Subject: [PATCH] =?UTF-8?q?[=E6=8B=9B=E5=95=86][=E6=B6=88=E6=81=AF?= =?UTF-8?q?=E6=8F=90=E9=86=92]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../domain/po/invest/InvestMessageEntity.java | 6 ++---- .../domain/vo/invest/InvestPageResult.java | 9 +++++++++ .../invest/impl/InvestBizServiceImpl.java | 20 ++++++++++--------- 3 files changed, 22 insertions(+), 13 deletions(-) diff --git a/mallinkService/src/main/java/com/iformall/domain/po/invest/InvestMessageEntity.java b/mallinkService/src/main/java/com/iformall/domain/po/invest/InvestMessageEntity.java index 80e69a5d7..a729146a5 100755 --- a/mallinkService/src/main/java/com/iformall/domain/po/invest/InvestMessageEntity.java +++ b/mallinkService/src/main/java/com/iformall/domain/po/invest/InvestMessageEntity.java @@ -28,15 +28,13 @@ public class InvestMessageEntity extends InvestBaseEntity { /** * 数据主键ID */ - @JsonIgnore - //@io.swagger.annotations.ApiModelProperty(value = "数据主键ID", name = "tid") + @io.swagger.annotations.ApiModelProperty(value = "数据主键ID", name = "tid") private Long tid; /** * 数据类型:0-招商任务;1-客户需求 */ - @JsonIgnore - //@io.swagger.annotations.ApiModelProperty(value = "数据类型:0-招商任务;1-客户需求", name = "type") + @io.swagger.annotations.ApiModelProperty(value = "数据类型:0-招商任务;1-客户需求", name = "type") private EnumFollowType type; /** * 消息内容 diff --git a/mallinkService/src/main/java/com/iformall/domain/vo/invest/InvestPageResult.java b/mallinkService/src/main/java/com/iformall/domain/vo/invest/InvestPageResult.java index 3aeebee39..725611355 100644 --- a/mallinkService/src/main/java/com/iformall/domain/vo/invest/InvestPageResult.java +++ b/mallinkService/src/main/java/com/iformall/domain/vo/invest/InvestPageResult.java @@ -27,5 +27,14 @@ public class InvestPageResult { private boolean hasNextPage = false ; + public Collection getRecords() { + return records; + } + public void setRecords(Collection records) { + this.records = records; + if (total > (records.size() + current * size)) { + hasNextPage = true; + } + } } diff --git a/mallinkService/src/main/java/com/iformall/service/invest/impl/InvestBizServiceImpl.java b/mallinkService/src/main/java/com/iformall/service/invest/impl/InvestBizServiceImpl.java index c796f97ba..af6b28910 100644 --- a/mallinkService/src/main/java/com/iformall/service/invest/impl/InvestBizServiceImpl.java +++ b/mallinkService/src/main/java/com/iformall/service/invest/impl/InvestBizServiceImpl.java @@ -340,15 +340,10 @@ public class InvestBizServiceImpl implements InvestBizService { @Override public boolean saveCustomerAndDemand(InvestDemandDto demandDto) { InvestHelper.notNull(demandDto, demandDto.getCustomer(), demandDto.getDemand()); - InvestCustomerEntity customerEntityDB = customerService.getByIdNotNull(demandDto.getCustomer().getId()); - if (Objects.isNull(customerEntityDB)) { - customerEntityDB = demandDto.getCustomer(); - customerService.save(customerEntityDB); - } else { - customerService.updateById(demandDto.getCustomer()); - } + InvestCustomerEntity customerEntity = demandDto.getCustomer() ; + customerService.save(customerEntity); InvestDemandEntity demandEntity = demandDto.getDemand(); - demandEntity.setCustomerId(customerEntityDB.getId()); + demandEntity.setCustomerId(customerEntity.getId()); return demandService.save(demandEntity); } @@ -356,9 +351,16 @@ public class InvestBizServiceImpl implements InvestBizService { @Override public boolean updateCustomerAndDemand(InvestDemandDto demandDto) { InvestHelper.notNull(demandDto, demandDto.getCustomer(), demandDto.getDemand()); + Long customerId = demandDto.getCustomer().getId(); + if (Objects.isNull(customerId)) { + customerId = demandDto.getDemand().getCustomerId(); + demandDto.getCustomer().setId(customerId); + } customerService.updateById(demandDto.getCustomer()); InvestDemandEntity demandEntity = demandDto.getDemand(); - demandEntity.setCustomerId(demandDto.getCustomer().getId()); + if (Objects.isNull(demandEntity.getCustomerId())) { + demandEntity.setCustomerId(customerId); + } return demandService.updateById(demandEntity); }