From a3297554085eabcd90fd6668dca0f1ad22fbf54d Mon Sep 17 00:00:00 2001
From: anthonywj <yewenjie20@vip.qq.com>
Date: Mon, 3 Jul 2023 19:20:57 +0800
Subject: [PATCH] =?UTF-8?q?=E8=8E=B7=E5=8F=96=E4=BC=81=E4=B8=9AID?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../api/controller/purchase/SupCertController.java    | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/glxp/api/controller/purchase/SupCertController.java b/src/main/java/com/glxp/api/controller/purchase/SupCertController.java
index 612e23c..9a1cb40 100644
--- a/src/main/java/com/glxp/api/controller/purchase/SupCertController.java
+++ b/src/main/java/com/glxp/api/controller/purchase/SupCertController.java
@@ -93,7 +93,6 @@ public class SupCertController {
 //    }
 
 
-
     @AuthRuleAnnotation("")
     @GetMapping("/sup/company/cert/filter")
     public BaseResponse filterUserCompanyCert(FilterSupCertRequest filterSupCertRequest,
@@ -427,6 +426,16 @@ public class SupCertController {
         return authAdmin.getCustomerId() + "";
     }
 
+    public String getCompanyId() {
+        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
+        if (attributes == null) {
+            throw new JsonException(ResultEnum.NOT_NETWORK);
+        }
+        HttpServletRequest request = attributes.getRequest();
+        String companyId = request.getHeader("companyId");
+        return companyId;
+    }
+
     /**
      * 校验模板文件
      *