From da6e4d69389a1a2daf60dc3b392a133929bce270 Mon Sep 17 00:00:00 2001
From: ky <592468495@qq.com>
Date: Sun, 10 May 2020 17:01:18 +0800
Subject: [PATCH] p

---
 src/main/java/com/gszc/service/impl/InvoiceServiceImpl.java | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/main/java/com/gszc/service/impl/InvoiceServiceImpl.java b/src/main/java/com/gszc/service/impl/InvoiceServiceImpl.java
index a6759b4..f6b38b3 100644
--- a/src/main/java/com/gszc/service/impl/InvoiceServiceImpl.java
+++ b/src/main/java/com/gszc/service/impl/InvoiceServiceImpl.java
@@ -76,6 +76,7 @@ public class InvoiceServiceImpl extends ServiceImpl<InvoiceMapper, Invoice> impl
                 invoice.setCustomType(customs.getJSONObject(i).getJSONObject("custom").getString("registerType"));
                 String invoiceHeaderId = invoice.getInvoiceHeaderId();
                 InvoiceHeader invoiceHeader = invoiceHeaderMapper.selectById(invoiceHeaderId);
+
                 invoice.setInvoiceHeaderName(invoiceHeader.getCompanyName());
                 invoice.setTaxCode(invoiceHeader.getTaxCode());
             }
@@ -115,6 +116,10 @@ public class InvoiceServiceImpl extends ServiceImpl<InvoiceMapper, Invoice> impl
     }
 
     public void deleteInvoiceHeader(String invoiceHeaderId){
+
+        if(invoiceMapper.selectCount(new EntityWrapper<Invoice>().eq("invoice_header_id", invoiceHeaderId))>0){
+            throw new RuntimeException("抬头已被使用");
+        }
         invoiceHeaderMapper.deleteById(invoiceHeaderId);
     }