修复bug

xx-order/xx-order-server/src/main/java/com/xxfc/platform/order/rest/OrderRentVehicleController.java

@@ -18,7 +18,11 @@ import com.xxfc.platform.order.service.OrderRentVehicleService;
 import com.xxfc.platform.order.pojo.AddRentVehicleDTO;
 import com.xxfc.platform.order.pojo.order.RentVehicleBO;
 import com.xxfc.platform.order.pojo.price.RentVehiclePriceVO;
+import com.xxfc.platform.vehicle.entity.BranchCompany;
+import com.xxfc.platform.vehicle.entity.Vehicle;
+import com.xxfc.platform.vehicle.feign.VehicleFeign;
 import com.xxfc.platform.vehicle.pojo.BookVehicleVO;
+import com.xxfc.platform.vehicle.pojo.CompanyDetail;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiModelProperty;
 import io.swagger.annotations.ApiOperation;
@@ -32,6 +36,7 @@ import springfox.documentation.annotations.ApiIgnore;
 import javax.persistence.Column;
 import java.util.Arrays;
 import java.util.List;
+import java.util.Map;
 import java.util.stream.Collectors;
 
 @Controller
@@ -52,13 +57,28 @@ public class OrderRentVehicleController extends CommonBaseController {
         @Autowired
         UserFeign userFeign;
 
+        @Autowired
+        VehicleFeign vehicleFeign;
+
         @RequestMapping(value = "add",method = RequestMethod.POST)
         @ResponseBody
         @ApiOperation(value = "确认租车订单")
         public ObjectRestResponse<BaseOrder> add(@RequestBody AddRentVehicleDTO vo){
-            if(null == vo.getBookEndDate() || null == vo.getBookStartDate()
-                    || vo.getBookStartDate().equals(0) || vo.getBookEndDate().equals(0)) {
-                    throw new BaseException(ResultCode.PARAM_ILLEGAL_CODE, Sets.newSet("公司参数不正确"));
+            if(null == vo.getEndCompanyId() || vo.getEndCompanyId().equals(0)) {
+                if(StrUtil.isBlank(vo.getEndAddr())) {
+                        throw new BaseException(ResultCode.PARAM_ILLEGAL_CODE, Sets.newSet("公司参数不正确"));
+                }else {
+                        Map<String, Object> map = BeanUtil.beanToMap(new BranchCompany(){{
+                                setName(vo.getEndAddr());
+                        }});
+                        List<BranchCompany> data = vehicleFeign.branchCompanyEntityList(map).getData();
+                        if(null == data) {
+                                throw new BaseException(ResultCode.PARAM_ILLEGAL_CODE, Sets.newSet("公司参数不正确"));
+                        }else {
+                                BranchCompany endCompany = vehicleFeign.branchCompanyEntityList(map).getData().get(0);
+                                vo.setEndCity(endCompany.getId());
+                        }
+                }
             }
             RentVehicleBO bo = BeanUtil.toBean(vo, RentVehicleBO.class);
             bo.setBookVehicleVO(new BookVehicleVO(){{