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