From 8f8b6835bbe3a3811191a66797486e6a097e2ecb Mon Sep 17 00:00:00 2001
From: dy-hu <dy-hu@haihangyun.com>
Date: Mon, 24 Jun 2024 10:48:10 +0800
Subject: [PATCH] =?UTF-8?q?=E5=95=86=E5=AE=B6=E6=96=B0=E5=A2=9EPC=E5=95=86?=
 =?UTF-8?q?=E5=9F=8E=E5=9F=9F=E5=90=8D=E3=80=81=E7=A7=BB=E5=8A=A8=E7=AB=AF?=
 =?UTF-8?q?=E5=95=86=E5=9F=8E=E5=9F=9F=E5=90=8D=E9=85=8D=E7=BD=AE=E4=B8=9A?=
 =?UTF-8?q?=E5=8A=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../app/dao/business/CerePlatformBusinessDAO.java  |  2 ++
 .../app/interceptor/AuthorizationInterceptor.java  | 35 +++++++++++++++++++++-
 .../mapper/business/CerePlatformBusinessDAO.xml    |  4 +++
 .../domain/business/CerePlatformBusiness.java      | 12 ++++++++
 doc/3.0/update.sql                                 |  2 ++
 5 files changed, 54 insertions(+), 1 deletion(-)

diff --git a/cereshop-app/src/main/java/com/shop/cereshop/app/dao/business/CerePlatformBusinessDAO.java b/cereshop-app/src/main/java/com/shop/cereshop/app/dao/business/CerePlatformBusinessDAO.java
index 17d9072..c65f7db 100644
--- a/cereshop-app/src/main/java/com/shop/cereshop/app/dao/business/CerePlatformBusinessDAO.java
+++ b/cereshop-app/src/main/java/com/shop/cereshop/app/dao/business/CerePlatformBusinessDAO.java
@@ -35,4 +35,6 @@ public interface CerePlatformBusinessDAO extends BaseMapper<CerePlatformBusiness
 
     CerePlatformBusiness checkBusinessIdByPhone(@Param("businessPhone") String businessPhone, @Param("businessid") Long businessid);
 
+    CerePlatformBusiness findByDomain(@Param("domain") String domain);
+
 }
diff --git a/cereshop-app/src/main/java/com/shop/cereshop/app/interceptor/AuthorizationInterceptor.java b/cereshop-app/src/main/java/com/shop/cereshop/app/interceptor/AuthorizationInterceptor.java
index 4594e21..99f2199 100644
--- a/cereshop-app/src/main/java/com/shop/cereshop/app/interceptor/AuthorizationInterceptor.java
+++ b/cereshop-app/src/main/java/com/shop/cereshop/app/interceptor/AuthorizationInterceptor.java
@@ -7,12 +7,14 @@ package com.shop.cereshop.app.interceptor;
 
 import cn.hutool.core.util.ObjectUtil;
 import com.alibaba.fastjson.JSON;
+import com.shop.cereshop.app.dao.business.CerePlatformBusinessDAO;
 import com.shop.cereshop.app.redis.service.api.UserRedisService;
 import com.shop.cereshop.app.service.buyer.CereBuyerUserService;
 import com.shop.cereshop.app.utils.ContextUtil;
 import com.shop.cereshop.app.utils.TokenProvider;
 import com.shop.cereshop.commons.config.SecurityProperties;
 import com.shop.cereshop.commons.constant.CoReturnFormat;
+import com.shop.cereshop.commons.domain.business.CerePlatformBusiness;
 import com.shop.cereshop.commons.domain.buyer.CereBuyerUser;
 import com.shop.cereshop.commons.result.Result;
 import com.shop.cereshop.commons.utils.EmptyUtils;
@@ -48,6 +50,9 @@ public class AuthorizationInterceptor extends HandlerInterceptorAdapter {
     @Autowired
     private TokenProvider tokenProvider;
 
+    @Autowired
+    private CerePlatformBusinessDAO cerePlatformBusinessDAO;
+
     @Override
     public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
         Result result = new Result();
@@ -118,7 +123,20 @@ public class AuthorizationInterceptor extends HandlerInterceptorAdapter {
             }
         }
 
-        String project = request.getHeader("project");
+        String project = "";
+
+        //获取访问的二级域名
+        String subDomain = getSubDomain(request);
+        if (ObjectUtil.isNotEmpty(subDomain)){
+            //从商家设置的商城访问域名表中获取商家id，即project
+            CerePlatformBusiness business =  cerePlatformBusinessDAO.findByDomain(subDomain);
+            if (ObjectUtil.isNotEmpty(business)){
+                project = String.valueOf(business.getBusinessId());
+            }
+        }else {
+            project = request.getHeader("project");
+        }
+
         if (EmptyUtils.isEmpty(project)  ||  project.equals("undefined")) {
             project = "0";
         }
@@ -178,4 +196,19 @@ public class AuthorizationInterceptor extends HandlerInterceptorAdapter {
         //防止乱码，适用于传输JSON数据
         response.setHeader("Content-Type", "application/json;charset=UTF-8");
     }
+
+    /**
+     * 获取访问的二级域名，如subdomain.example.com。获取的是subdomain
+     * @param request
+     * @return
+     */
+    private String getSubDomain(HttpServletRequest request) {
+        String host = request.getHeader("Host");
+        String[] parts = host.split("\\.");
+        if (parts.length >= 3){
+            return parts[0];
+        }
+
+        return "";
+    }
 }
diff --git a/cereshop-app/src/main/resources/mybatis/mapper/business/CerePlatformBusinessDAO.xml b/cereshop-app/src/main/resources/mybatis/mapper/business/CerePlatformBusinessDAO.xml
index 7226499..1769437 100644
--- a/cereshop-app/src/main/resources/mybatis/mapper/business/CerePlatformBusinessDAO.xml
+++ b/cereshop-app/src/main/resources/mybatis/mapper/business/CerePlatformBusinessDAO.xml
@@ -321,4 +321,8 @@
   <select id="checkBusinessIdByPhone" resultType="com.shop.cereshop.commons.domain.business.CerePlatformBusiness">
     SELECT business_id FROM cere_platform_business where business_phone=#{businessPhone} and business_id<![CDATA[!= ]]>#{businessId}
   </select>
+
+  <select id="findByDomain" resultType="com.shop.cereshop.commons.domain.business.CerePlatformBusiness">
+    SELECT business_id FROM cere_platform_business where pc_domain=#{domain} or mobile_domain=#{domain}
+  </select>
 </mapper>
diff --git a/cereshop-commons/src/main/java/com/shop/cereshop/commons/domain/business/CerePlatformBusiness.java b/cereshop-commons/src/main/java/com/shop/cereshop/commons/domain/business/CerePlatformBusiness.java
index feb8e84..6cd2faa 100644
--- a/cereshop-commons/src/main/java/com/shop/cereshop/commons/domain/business/CerePlatformBusiness.java
+++ b/cereshop-commons/src/main/java/com/shop/cereshop/commons/domain/business/CerePlatformBusiness.java
@@ -191,6 +191,18 @@ public class CerePlatformBusiness implements Serializable {
     @ApiModelProperty(value = "分账账期")
     private Integer shareDays;
 
+    /**
+     * PC端商城域名
+     */
+    @ApiModelProperty(value = "PC端商城域名")
+    private String pcDomain;
+
+    /**
+     * 移动端商城域名
+     */
+    @ApiModelProperty(value = "移动端商城域名")
+    private String mobileDomain;
+
 
     private static final long serialVersionUID = 1L;
 
diff --git a/doc/3.0/update.sql b/doc/3.0/update.sql
index 4bd42dd..e8a1009 100644
--- a/doc/3.0/update.sql
+++ b/doc/3.0/update.sql
@@ -287,3 +287,5 @@ ALTER TABLE `cere_shop_product_lang_info` MODIFY COLUMN `product_brief` varchar(
 ALTER TABLE `cere_shop_product` MODIFY COLUMN `classify_id` bigint DEFAULT NULL COMMENT '关联分类id';
 
 ALTER TABLE cere_platform_shop ADD shop_type tinyint(1) DEFAULT 1 COMMENT '店铺类型  1国内贸易 2跨境保税 3海外直邮';
+ALTER TABLE cere_platform_business ADD pc_domain varchar(128) DEFAULT '' COMMENT 'pc端商城域名';
+ALTER TABLE cere_platform_business ADD mobile_domain varchar(128) DEFAULT '' COMMENT '移动端商城域名';