diff --git a/src/main/java/com/sczx/car/controller/CarRentalController.java b/src/main/java/com/sczx/car/controller/CarRentalController.java
index 2c2b9c6..83311e8 100644
--- a/src/main/java/com/sczx/car/controller/CarRentalController.java
+++ b/src/main/java/com/sczx/car/controller/CarRentalController.java
@@ -3,7 +3,7 @@ package com.sczx.car.controller;
import com.sczx.car.common.Result;
import com.sczx.car.dto.CarRentalDetailDTO;
-import com.sczx.car.dto.RentBatteyRuleDTO;
+import com.sczx.car.dto.RentBatteyRuleRsp;
import com.sczx.car.service.CarRentalService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
@@ -13,8 +13,6 @@ import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
-import java.util.List;
-
/**
*
* 车型表 前端控制器
@@ -40,7 +38,7 @@ public class CarRentalController {
@ApiOperation(value = "根据租车套餐id查询租店套餐列表")
@GetMapping("/listRentBatteyRuleByCarRuleId")
- public Result> listRentBatteyRuleByCarRuleId(@RequestParam(name = "carRuleId") Integer carRuleId){
+ public Result listRentBatteyRuleByCarRuleId(@RequestParam(name = "carRuleId") Integer carRuleId){
return Result.ok(carRentalService.queryRentBatteyRuleByCarRuleId(carRuleId));
}
}
diff --git a/src/main/java/com/sczx/car/dto/RentBatteyRuleDTO.java b/src/main/java/com/sczx/car/dto/RentBatteyRuleDTO.java
index 0b75239..bb2293c 100644
--- a/src/main/java/com/sczx/car/dto/RentBatteyRuleDTO.java
+++ b/src/main/java/com/sczx/car/dto/RentBatteyRuleDTO.java
@@ -72,6 +72,9 @@ public class RentBatteyRuleDTO {
@ApiModelProperty("电池类型")
private Integer categoryId;
+ @ApiModelProperty("电池类型名称")
+ private String categoryName;
+
@ApiModelProperty("套餐类型 换电/租电")
private Integer mealType;
diff --git a/src/main/java/com/sczx/car/dto/RentBatteyRuleRsp.java b/src/main/java/com/sczx/car/dto/RentBatteyRuleRsp.java
new file mode 100644
index 0000000..f90baad
--- /dev/null
+++ b/src/main/java/com/sczx/car/dto/RentBatteyRuleRsp.java
@@ -0,0 +1,20 @@
+package com.sczx.car.dto;
+
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+
+import java.util.List;
+
+@ApiModel(value = "租电套餐计费规则返回对象")
+@Data
+public class RentBatteyRuleRsp {
+
+ @ApiModelProperty("电池名称列表")
+ private List categoryNameList;
+
+ @ApiModelProperty("租电套餐计费规则列表")
+ private List rentBatteyRuleList;
+
+
+}
diff --git a/src/main/java/com/sczx/car/service/CarRentalService.java b/src/main/java/com/sczx/car/service/CarRentalService.java
index 1d8531a..79961f6 100644
--- a/src/main/java/com/sczx/car/service/CarRentalService.java
+++ b/src/main/java/com/sczx/car/service/CarRentalService.java
@@ -1,9 +1,7 @@
package com.sczx.car.service;
import com.sczx.car.dto.CarRentalDetailDTO;
-import com.sczx.car.dto.RentBatteyRuleDTO;
-
-import java.util.List;
+import com.sczx.car.dto.RentBatteyRuleRsp;
public interface CarRentalService {
@@ -22,5 +20,5 @@ public interface CarRentalService {
* @param carRuleId
* @return
*/
- List queryRentBatteyRuleByCarRuleId(Integer carRuleId);
+ RentBatteyRuleRsp queryRentBatteyRuleByCarRuleId(Integer carRuleId);
}
diff --git a/src/main/java/com/sczx/car/service/impl/CarRentalServiceImpl.java b/src/main/java/com/sczx/car/service/impl/CarRentalServiceImpl.java
index 297a045..2d417f0 100644
--- a/src/main/java/com/sczx/car/service/impl/CarRentalServiceImpl.java
+++ b/src/main/java/com/sczx/car/service/impl/CarRentalServiceImpl.java
@@ -3,6 +3,8 @@ package com.sczx.car.service.impl;
import com.sczx.car.convert.CarModelConvert;
import com.sczx.car.dto.CarRentalDetailDTO;
import com.sczx.car.dto.RentBatteyRuleDTO;
+import com.sczx.car.dto.RentBatteyRuleRsp;
+import com.sczx.car.exception.BizException;
import com.sczx.car.po.CarModelPO;
import com.sczx.car.repository.CarModelRepo;
import com.sczx.car.repository.RentBatteyRuleRepo;
@@ -12,8 +14,10 @@ import com.sczx.car.thirdpart.integration.StoreInteg;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
+import org.springframework.util.CollectionUtils;
import java.util.List;
+import java.util.stream.Collectors;
@Slf4j
@Service
@@ -47,7 +51,17 @@ public class CarRentalServiceImpl implements CarRentalService {
}
@Override
- public List queryRentBatteyRuleByCarRuleId(Integer carRuleId) {
- return rentBatteyRuleRepo.queryRentBatteyRuleByCarRuleId(carRuleId);
+ public RentBatteyRuleRsp queryRentBatteyRuleByCarRuleId(Integer carRuleId) {
+
+ List rentBatteyRuleList = rentBatteyRuleRepo.queryRentBatteyRuleByCarRuleId(carRuleId);
+ if(CollectionUtils.isEmpty(rentBatteyRuleList)){
+ throw new BizException("无对应租电套餐,请重新选择租车套餐");
+ }else {
+ RentBatteyRuleRsp rentBatteyRuleRsp = new RentBatteyRuleRsp();
+ rentBatteyRuleRsp.setRentBatteyRuleList(rentBatteyRuleList);
+ List categoryNameList = rentBatteyRuleList.stream().map(RentBatteyRuleDTO::getCategoryName).distinct().collect(Collectors.toList());
+ rentBatteyRuleRsp.setCategoryNameList(categoryNameList);
+ return rentBatteyRuleRsp;
+ }
}
}
diff --git a/src/main/resources/mapper/RentBatteyRuleMapper.xml b/src/main/resources/mapper/RentBatteyRuleMapper.xml
index fad1b9b..b1b5bc3 100644
--- a/src/main/resources/mapper/RentBatteyRuleMapper.xml
+++ b/src/main/resources/mapper/RentBatteyRuleMapper.xml
@@ -4,11 +4,13 @@