Parcourir la source

controller层增加信息校验

mashengyi il y a 2 ans
Parent
commit
02adbd50d7

+ 41 - 2
src/main/java/com/jkcredit/invoice/controller/business/CustomerController.java

@@ -174,7 +174,7 @@ public class CustomerController {
     @LoginRequired
     public RespR addCustomer(@RequestBody Customer customer, User user) {
         checkHasAuthRole.checkCustomerRole(user, customer.getCustomerName());
-        if(!checkHasAuthRole.checkMessage(customer.getCustomerName())){
+        if(!checkHasAuthRole.checkMessage(customer.getCustomerName(),customer.getCompany())){
             return new RespR(false, "客户添加失败,请查看数据是否维护准确");
         }
         return new RespR(customerService.addCustomers(customer), "客户添加失败,请查看是否重复");
@@ -189,7 +189,9 @@ public class CustomerController {
     @ApiOperation(value = "更新用户", notes = "更新用户")
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR updateCustomer(@RequestBody Customer customer) {
-
+        if(!checkHasAuthRole.checkMessage(customer.getCustomerName(),customer.getCompany())){
+            return new RespR(false, "客户更新失败,请查看数据是否维护准确");
+        }
         return new RespR(customerService.updateCustomer(customer));
     }
 
@@ -228,6 +230,14 @@ public class CustomerController {
     @ApiOperation(value = "手工添加备案信息", notes = "手工添加备案信息")
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR customerRecAdd(@RequestBody CustomerRec customerRec) {
+        if(!checkHasAuthRole.checkMessage(customerRec.getCustomerName(),customerRec.getCompanyLeader(),customerRec.getCompanyLeaderPhone(),
+                customerRec.getCompanyName(),customerRec.getCompanyReferencenum(),
+                customerRec.getCompanyOpenbank(),customerRec.getCompanyOpenbankAcc(),
+                customerRec.getCompanyAdress(),customerRec.getCompanyPhone(),
+                customerRec.getBussinessType())){
+            return new RespR(false, "手工添加备案信息失败,请查看数据是否维护准确");
+        }
+
         customerRec.setInterType(1);
         List<CustomerRec> customerRecList = new ArrayList<>();
         customerRecList.add(customerRec);
@@ -243,6 +253,10 @@ public class CustomerController {
     @ApiOperation(value = "查询备案信息", notes = "手工添加备案信息")
     @LoginRequired
     public RespR customeRecQueryUpper(@RequestBody CustomerRec customerRec, User user) {
+        if(!checkHasAuthRole.checkMessage(customerRec.getCustomerName(),customerRec.getCompanyName(),customerRec.getCompanyReferencenum())){
+            return new RespR(false, "查询备案信息失败,请查看数据是否维护准确");
+        }
+
         checkHasAuthRole.checkCustomerRole(user, customerRec.getCustomerName());
         RespR<List<B2bCompanyModel>> respR = customerService.customerRecQuery(customerRec);
 
@@ -298,6 +312,9 @@ public class CustomerController {
     @ApiOperation(value = "用户备案确认", notes = "用户备案确认")
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR customeRec(@RequestBody CustomerRec customerRec) {
+        if(!checkHasAuthRole.checkMessage(customerRec.getCustomerName(),customerRec.getCompanyReferencenum())){
+            return new RespR(false, "用户备案确认失败,请查看数据是否维护准确");
+        }
         return customerService.customeRec(customerRec);
     }
 
@@ -305,6 +322,11 @@ public class CustomerController {
     @ApiOperation(value = "协议上传确认", notes = "协议上传确认")
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR contractAdd(@RequestBody CustomerRec customerRec) {
+        if(!checkHasAuthRole.checkMessage(customerRec.getCompanyNum(),
+                customerRec.getContractFileName(),customerRec.getServiceStartTime(),
+                customerRec.getServiceEndTime())){
+            return new RespR(false, "协议上传确认失败,请查看数据是否维护准确");
+        }
         return customerService.contractAdd(customerRec);
     }
 
@@ -312,6 +334,9 @@ public class CustomerController {
     @ApiOperation(value = "协议上传失败状态确认", notes = "协议上传失败状态确认")
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR contractStatusFail(@RequestBody CustomerRec customerRec) {
+        if(null == customerRec.getRecStatus()){
+            return new RespR(false, "协议上传失败状态确认失败,请查看数据是否维护准确");
+        }
         return customerService.contractStatusFail(customerRec);
     }
 
@@ -319,6 +344,9 @@ public class CustomerController {
     @ApiOperation(value = "协议上传成功,审批中状态确认", notes = "协议上传成功,审批中状态确认")
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR contractStatusProcess(@RequestBody CustomerRec customerRec) {
+        if(null == customerRec.getRecStatus()){
+            return new RespR(false, "协议上传成功,审批中状态确认失败,请查看数据是否维护准确");
+        }
         return customerService.contractStatusProcess(customerRec);
     }
 
@@ -326,6 +354,9 @@ public class CustomerController {
     @ApiOperation(value = "备案成功状态确认", notes = "备案成功状态确认")
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR contractStatusSuccess(@RequestBody CustomerRec customerRec) {
+        if(null == customerRec.getRecStatus()){
+            return new RespR(false, "备案成功状态确认失败,请查看数据是否维护准确");
+        }
         return customerService.contractStatusSuccess(customerRec);
     }
 
@@ -333,6 +364,10 @@ public class CustomerController {
     @ApiOperation(value = "协议下载", notes = "协议下载")
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR contractDownload(String customerRecId, HttpServletResponse response) {
+        if(!checkHasAuthRole.checkMessage(customerRecId)){
+            return new RespR(false, "协议下载失败,请查看数据是否维护准确");
+        }
+
         CustomerRec customerRec = (CustomerRec) customerService.contractDownload(customerRecId).getData();
         String fileName = customerRec.getLowerFileName();
         InputStream inputStream = null;
@@ -362,6 +397,10 @@ public class CustomerController {
     @ApiOperation(value = "文档下载", notes = "文档下载")
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR generateWord(String customerRecId, HttpServletResponse response) {
+        if(!checkHasAuthRole.checkMessage(customerRecId)){
+            return new RespR(false, "文档下载失败,请查看数据是否维护准确");
+        }
+
         CustomerRec customerRec = (CustomerRec) customerService.generateWordDownload(customerRecId).getData();
         if (null == customerRec || null == customerRec.getServiceStartTime() || null == customerRec.getServiceEndTime()) {
             return new RespR(false, "企业维护信息缺失服务时间,请手动维护");

+ 27 - 0
src/main/java/com/jkcredit/invoice/controller/business/NoCarController.java

@@ -339,6 +339,10 @@ public class NoCarController {
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR batchImportNocarInvoices(@RequestParam("file") MultipartFile file) {
         try {
+            if(null == file){
+                return new RespR(false, "批量查询无车发票信息失败,请查看数据是否维护准确");
+            }
+
             ImportParams params = new ImportParams();
             params.setTitleRows(1);
             params.setHeadRows(1);
@@ -363,6 +367,11 @@ public class NoCarController {
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR batchImportNocarBillWay(@RequestParam("file") MultipartFile file) {
         try {
+
+            if(null == file){
+                return new RespR(false, "批量查询无车运单信息失败,请查看数据是否维护准确");
+            }
+
             ImportParams params = new ImportParams();
             params.setTitleRows(1);
             params.setHeadRows(1);
@@ -417,6 +426,11 @@ public class NoCarController {
     @LoginRequired
     public RespR batchImprotBillWay(@RequestParam("file") MultipartFile file, String customerName, User user) {
         try {
+
+            if(null == file){
+                return new RespR(false, "批量运单导入失败,请查看数据是否维护准确");
+            }
+
             if(!checkHasAuthRole.checkMessage(customerName)){
                 return new RespR(false, "批量运单导入失败,请查看数据是否维护准确");
             }
@@ -446,6 +460,11 @@ public class NoCarController {
     @LoginRequired
     public RespR batchImprotHistoryBillWay(@RequestParam("file") MultipartFile file, String customerName, User user) {
         try {
+
+            if(null == file){
+                return new RespR(false, "批量历史运单导入失败,请查看数据是否维护准确");
+            }
+
             if(!checkHasAuthRole.checkMessage(customerName)){
                 return new RespR(false, "批量历史运单导入失败,请查看数据是否维护准确");
             }
@@ -476,6 +495,10 @@ public class NoCarController {
     @LoginRequired
     public RespR batchImprotEndBillWay(@RequestParam("file") MultipartFile file, String customerName, User user) {
 
+        if(null == file){
+            return new RespR(false, "批量运单结束失败,请查看数据是否维护准确");
+        }
+
         if(!checkHasAuthRole.checkMessage(customerName)){
             return new RespR(false, "批量运单结束失败,请查看数据是否维护准确");
         }
@@ -685,6 +708,10 @@ public class NoCarController {
     @GetMapping("/templateDownload")
     @ApiOperation(value = "模板下载", notes = "模板下载")
     public RespR contractDownload(String fileName, HttpServletResponse response) {
+
+        if(!checkHasAuthRole.checkMessage(fileName)){
+            return new RespR(false, "模板下载失败,请查看数据是否维护准确");
+        }
         if (ONE.equals(fileName)) {
             fileName = "运单模板.xlsx";
         } else if (TWO.equals(fileName)) {

+ 27 - 0
src/main/java/com/jkcredit/invoice/controller/business/SelfCarController.java

@@ -386,6 +386,11 @@ public class SelfCarController {
     @ApiOperation(value = "获取需要开票的交易Id列表", notes = "获取需要开票的交易Id列表")
     @LoginRequired
     public RespR getTradeList(@RequestBody TradeRequestCarNumVo tradeRequestCarNumVo, User user) {
+
+        if(!checkHasAuthRole.checkMessage(tradeRequestCarNumVo.getCarNum())){
+            return new RespR(false, "获取需要开票的交易Id列表失败,请查看数据是否维护准确");
+        }
+
         checkHasAuthRole.checkCompanyRole(user, tradeRequestCarNumVo.getCompanyName());
         //通过车牌号查询所有的Etc卡号
         List<CustomerCarRec> customerCarRecs = selfCarBussService.getEtcsByCarNum(tradeRequestCarNumVo.getCarNum());
@@ -419,6 +424,11 @@ public class SelfCarController {
     @ApiOperation(value = "根据车牌号进行开票操作", notes = "根据车牌号进行开票操作")
     @LoginRequired
     public RespR applTradeList(@RequestBody TradeRequestCarNumVo tradeRequestCarNumVo, User user) {
+
+        if(!checkHasAuthRole.checkMessage(tradeRequestCarNumVo.getCarIdStr())){
+            return new RespR(false, "根据车牌号进行开票操作失败,请查看数据是否维护准确");
+        }
+
         checkHasAuthRole.checkCompanyRole(user, tradeRequestCarNumVo.getCompanyName());
         String[] cardAndTrades = tradeRequestCarNumVo.getCarIdStr().split(",");
         Map<String, InvoiceApplVo> map = new HashMap<>(10);
@@ -465,6 +475,12 @@ public class SelfCarController {
     @LoginRequired
     public RespR findSelfcarInvoiceByTime(@RequestBody SelfcarInvoiceByTimeVo selfcarInvoiceByTimeVo, User user) {
         try {
+
+            if(!checkHasAuthRole.checkMessage(selfcarInvoiceByTimeVo.getCustomerName())){
+                return new RespR(false, "根据车牌号进行开票操作失败,请查看数据是否维护准确");
+            }
+
+
             checkHasAuthRole.checkCompanyRole(user, selfcarInvoiceByTimeVo.getCompanyName());
             if (!customerService.checkMoneyEnough(selfcarInvoiceByTimeVo.getCustomerName())) {
                 return new RespR(false, "余额不足");
@@ -519,6 +535,12 @@ public class SelfCarController {
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR batchImportSelfcarTrades(@RequestParam("file") MultipartFile file) {
         try {
+
+            if(null == file){
+                return new RespR(false, "批量查询自有车交易信息失败,请查看数据是否维护准确");
+            }
+
+
             ImportParams params = new ImportParams();
             params.setTitleRows(1);
             params.setHeadRows(1);
@@ -543,6 +565,11 @@ public class SelfCarController {
     @LoginRequired(role = AuthenticationInterceptor.AUTH_ADMIN)
     public RespR batchImportSelfcarInvoices(@RequestParam("file") MultipartFile file) {
         try {
+
+            if(null == file){
+                return new RespR(false, "批量查询自有车发票信息失败,请查看数据是否维护准确");
+            }
+
             ImportParams params = new ImportParams();
             params.setTitleRows(1);
             params.setHeadRows(1);