From cf566cbe51e917833f3e4e9820b8656b9c7c6a7c Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Mon, 11 Jul 2022 11:21:30 +0800 Subject: [PATCH] fix --- src/test/java/com/xcong/farmer/cms/KssframeworkApplicationTests.java | 68 ++++++++++++++++++++++++++++++++-- 1 files changed, 64 insertions(+), 4 deletions(-) diff --git a/src/test/java/com/xcong/farmer/cms/KssframeworkApplicationTests.java b/src/test/java/com/xcong/farmer/cms/KssframeworkApplicationTests.java index 5053fe0..a571fb4 100644 --- a/src/test/java/com/xcong/farmer/cms/KssframeworkApplicationTests.java +++ b/src/test/java/com/xcong/farmer/cms/KssframeworkApplicationTests.java @@ -2,7 +2,10 @@ import cn.hutool.core.collection.CollUtil; import cn.hutool.core.date.DateUtil; +import cn.hutool.core.io.FileUtil; +import cn.hutool.core.io.file.FileWriter; import cn.hutool.core.util.ObjectUtil; +import cn.hutool.core.util.RandomUtil; import cn.hutool.core.util.StrUtil; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; @@ -12,13 +15,11 @@ import com.xcong.farmer.cms.configurations.properties.CmsProperties; import com.xcong.farmer.cms.modules.system.dto.AdminArticleDto; import com.xcong.farmer.cms.modules.system.dto.WebArticleInPageDto; -import com.xcong.farmer.cms.modules.system.entity.ArticleEntity; -import com.xcong.farmer.cms.modules.system.entity.CmsTemplateEntity; -import com.xcong.farmer.cms.modules.system.entity.ColumnEntity; -import com.xcong.farmer.cms.modules.system.entity.UserEntity; +import com.xcong.farmer.cms.modules.system.entity.*; import com.xcong.farmer.cms.modules.system.mapper.ArticleMapper; import com.xcong.farmer.cms.modules.system.mapper.CmsTemplateMapper; import com.xcong.farmer.cms.modules.system.mapper.ColumnMapper; +import com.xcong.farmer.cms.modules.system.mapper.CompanyMapper; import com.xcong.farmer.cms.modules.system.util.LoginUserUtil; import com.xcong.farmer.cms.modules.system.vo.AdminArticleVo; import com.xcong.farmer.cms.modules.system.vo.WebArticleVo; @@ -34,6 +35,8 @@ import java.io.FileInputStream; import java.io.IOException; import java.io.InputStream; +import java.math.BigDecimal; +import java.math.RoundingMode; import java.nio.charset.StandardCharsets; import java.nio.file.Files; import java.nio.file.Paths; @@ -50,6 +53,32 @@ @Resource private CmsTemplateMapper cmsTemplateMapper; + @Test + public void testFileWrite(){ + Long companyId = 24L; + Long id = 38L; + CmsTemplateEntity cmsTemplateEntity = cmsTemplateMapper.selectByIdAndCompanyId(id,companyId); + String name = cmsTemplateEntity.getName(); + Integer type = cmsTemplateEntity.getType(); + String templatePath = "D:\\cmshtml"; + String path = cmsTemplateEntity.getPath(); + + cmsTemplateMapper.delete(id,companyId); + String pathNew = FileUtils.path(templatePath, path); + File file = new File(pathNew); + FileUtil.touch(file); + + //文件写入,直接覆盖 + FileWriter writer = new FileWriter(file); + writer.write("测试121634964949", false); + + CmsTemplateEntity cmsTemplate = new CmsTemplateEntity(); + cmsTemplate.setCompanyId(companyId); + cmsTemplate.setType(type); + cmsTemplate.setName(name); + cmsTemplate.setPath(path); + cmsTemplateMapper.insert(cmsTemplate); + } @Test public void testUserInsert() { for(int i = 0; i < 20; i++) { @@ -125,6 +154,37 @@ @Autowired private ArticleMapper articleMapper; + @Autowired + private CompanyMapper companyMapper; + + @Test + public void testCompanyId(){ + String website = "127.0.0.1"; + Long companyIdFromWebsite = getCompanyIdFromWebsite(website); + System.out.print(companyIdFromWebsite); + } + + 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(companyEntity.getWebAddress(), website); + if(contains){ + companyId = companyEntity.getId(); + } + } + } + return companyId; + } + + public static void main(String[] args) { + for(int i=0;i<10;i++){ + BigDecimal partnerIncomeNum = RandomUtil.randomBigDecimal(new BigDecimal("5000.001"), new BigDecimal("10000.0001")).setScale(6, RoundingMode.HALF_DOWN); + System.out.println(partnerIncomeNum); + } + } + @Test public void articleTest(){ WebArticleInPageDto webArticleInPageDto = new WebArticleInPageDto(); -- Gitblit v1.9.1