From f04b8922283120e2bc4a476ddc302ebd3b0894e4 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Thu, 07 Jul 2022 20:42:05 +0800 Subject: [PATCH] Merge branch 'master' of http://120.27.238.55:7000/r/farmer-cms --- src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MessageBoardServiceImpl.java | 32 ++++++++++++++++++++++---------- 1 files changed, 22 insertions(+), 10 deletions(-) diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MessageBoardServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MessageBoardServiceImpl.java index 79cd77e..bee91c1 100644 --- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MessageBoardServiceImpl.java +++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MessageBoardServiceImpl.java @@ -5,6 +5,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.system.dto.AdminAddMessageDto; import com.xcong.farmer.cms.modules.system.dto.AdminDeleteDto; @@ -70,16 +71,13 @@ @Override public Result addMessage(HttpServletRequest request, AdminAddMessageDto adminAddMessageDto) { - 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); + + String title = adminAddMessageDto.getTitle(); + String remark = adminAddMessageDto.getRemark(); + if(StrUtil.isEmpty(title) && StrUtil.isEmpty(remark)){ + return Result.fail("请输入留言"); } MessageBoardEntity messageBoardEntity = new MessageBoardEntity(); messageBoardEntity.setName(adminAddMessageDto.getName()); @@ -95,4 +93,18 @@ this.baseMapper.insert(messageBoardEntity); 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