From eb3d6df3d5e48f32fc7714ec357f7b6ceccc3659 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 07 Jul 2022 19:57:47 +0800
Subject: [PATCH] fix

---
 src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/ArticleServiceImpl.java |   40 ++++++++++++++++++++++++++--------------
 1 files changed, 26 insertions(+), 14 deletions(-)

diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/ArticleServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/ArticleServiceImpl.java
index 8435fda..962cef3 100644
--- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/ArticleServiceImpl.java
+++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/ArticleServiceImpl.java
@@ -6,6 +6,7 @@
 import com.baomidou.mybatisplus.core.toolkit.StringPool;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.xcong.farmer.cms.common.contants.AppContants;
 import com.xcong.farmer.cms.common.response.Result;
 import com.xcong.farmer.cms.modules.core.service.ICmsCoreService;
 import com.xcong.farmer.cms.modules.system.dto.*;
@@ -24,6 +25,7 @@
 import org.springframework.stereotype.Service;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 import java.util.Date;
 
@@ -53,9 +55,14 @@
         long companyId = userlogin.getCompanyId() == null ? UserEntity.USER_BELONG_TOP : userlogin.getCompanyId();
         Page<AdminArticleVo> page = new Page<>(adminArticleDto.getPageNum(), adminArticleDto.getPageSize());
         ArticleEntity articleEntity = new ArticleEntity();
-        Long columnId = adminArticleDto.getColumnId() == null ? 0L : adminArticleDto.getColumnId();
-        if(columnId != 0L){
-            articleEntity.setColumnId(columnId);
+        String columnIdStrs = adminArticleDto.getColumnIdStr();
+        if(StrUtil.isNotEmpty(columnIdStrs)){
+            String[] columnIdStr = columnIdStrs.split(StringPool.COMMA);
+            List<Long> columnList = new ArrayList<>();
+            for(String columnIdString : columnIdStr){
+                columnList.add(Long.valueOf(columnIdString));
+            }
+            articleEntity.setColumnList(columnList);
         }
         String title = adminArticleDto.getTitle();
         if(StrUtil.isNotEmpty(title)){
@@ -259,17 +266,8 @@
 
     @Override
     public Result webArticleInPage(HttpServletRequest request, WebArticleInPageDto webArticleInPageDto) {
-        StringBuffer requestURL = request.getRequestURL();
-        List<CompanyEntity> companyEntities = companyMapper.selectList(new QueryWrapper<>());
-        Long companyId = 0L;
-        if(CollUtil.isNotEmpty(companyEntities)){
-            for(CompanyEntity companyEntity : companyEntities){
-                boolean contains = StrUtil.contains(requestURL, companyEntity.getWebAddress());
-                if(contains){
-                    companyId = companyEntity.getId();
-                }
-            }
-        }
+        String website = request.getHeader(AppContants.WEBSITE_HEADER);
+        Long companyId = getCompanyIdFromWebsite(website);
 
         Page<WebArticleVo> page = new Page<>(webArticleInPageDto.getPageNum(), webArticleInPageDto.getPageSize());
         ArticleEntity articleEntity = new ArticleEntity();
@@ -305,4 +303,18 @@
         releaseService.releaseArticle(articleEntity.getId());
         return Result.ok("操作成功");
     }
+
+    private Long getCompanyIdFromWebsite(String website){
+        List<CompanyEntity> companyEntities = companyMapper.selectList(new QueryWrapper<>());
+        Long companyId = 0L;
+        if(CollUtil.isNotEmpty(companyEntities)){
+            for(CompanyEntity companyEntity : companyEntities){
+                boolean contains = StrUtil.contains(website, companyEntity.getWebAddress());
+                if(contains){
+                    companyId = companyEntity.getId();
+                }
+            }
+        }
+        return companyId;
+    }
 }

--
Gitblit v1.9.1