package com.xcong.farmer.cms; 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; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.xcong.farmer.cms.common.response.Result; import com.xcong.farmer.cms.common.utils.FileUtils; 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.*; 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; import com.xcong.farmer.cms.modules.test.dao.TestUserDao; import com.xcong.farmer.cms.modules.test.entity.TestUserEntity; import org.jsoup.Jsoup; import org.junit.jupiter.api.Test; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; import javax.annotation.Resource; import java.io.File; 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; import java.util.ArrayList; import java.util.Date; import java.util.List; @SpringBootTest class KssframeworkApplicationTests { @Resource private TestUserDao testUserDao; @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++) { TestUserEntity testUser = new TestUserEntity(); testUser.setCreateBy("123"); testUser.setCreateTime(new Date()); testUser.setUpdateBy("123"); testUser.setUpdateTime(new Date()); testUser.setAccount("12345" + i); testUser.setName("hehe" + i); testUser.setPassword("33333"); testUserDao.insert(testUser); } } @Test public void start() { System.out.println(1); } @Autowired private CmsProperties cmsProperties; @Test public void viewTemplateInfo() { Long id = 4L; CmsTemplateEntity cmsTemplateEntity = cmsTemplateMapper.selectById(id); String htmlUrl = "" + cmsProperties.getTemplatePath(); File uploadDir = new File(htmlUrl); if (!uploadDir.isDirectory()) { uploadDir.mkdir(); } htmlUrl = "D:\\cmshtml"; String pathName = htmlUrl+"\\"+cmsTemplateEntity.getPath(); System.out.println(pathName); byte[] bytes = new byte[0]; try { bytes = Files.readAllBytes(Paths.get(pathName)); } catch (IOException e) { e.printStackTrace(); System.out.println("未找到模板"); } String content = new String(bytes, StandardCharsets.UTF_8); System.out.println(content); } @Test public void path(){ Long id = 6L; CmsTemplateEntity cmsTemplateEntity = cmsTemplateMapper.selectById(id); String templatePath = cmsProperties.getTemplatePath(); String pathName = FileUtils.path(templatePath, cmsTemplateEntity.getPath()); System.out.println(templatePath); System.out.println(pathName); byte[] bytes = new byte[0]; try { bytes = Files.readAllBytes(Paths.get(pathName)); } catch (IOException e) { e.printStackTrace(); System.out.println("未找到模板"); } String content = new String(bytes, StandardCharsets.UTF_8); System.out.println(content); } @Autowired private ColumnMapper columnMapper; @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 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(); webArticleInPageDto.setTimeType("一周内"); webArticleInPageDto.setPageNum(1); webArticleInPageDto.setPageSize(8); Long companyId = 24L; Page page = new Page<>(webArticleInPageDto.getPageNum(), webArticleInPageDto.getPageSize()); ArticleEntity articleEntity = new ArticleEntity(); Long columnId = webArticleInPageDto.getColumnId() == null ? 0L : webArticleInPageDto.getColumnId(); if(columnId != 0L){ articleEntity.setColumnId(columnId); } String author = webArticleInPageDto.getAuthor(); if(StrUtil.isNotEmpty(author)){ articleEntity.setAuthor(author); } String queryKey = webArticleInPageDto.getQueryKey(); if(StrUtil.isNotEmpty(queryKey)){ articleEntity.setTitle(queryKey); } if(UserEntity.USER_BELONG_TOP != companyId){ articleEntity.setCompanyId(companyId); } String timeType = webArticleInPageDto.getTimeType(); if(StrUtil.isNotEmpty(timeType)){ articleEntity.setTimeType(timeType); } IPage list = articleMapper.selectWebArticleInPage(page,articleEntity); System.out.println(list.getRecords()); } @Test public void dateTest(){ String columnIdStrs = "68,63"; AdminArticleDto adminArticleDto = new AdminArticleDto(); adminArticleDto.setColumnIdStr(columnIdStrs); adminArticleDto.setPageNum(1); adminArticleDto.setPageSize(8); Page page = new Page<>(adminArticleDto.getPageNum(), adminArticleDto.getPageSize()); ArticleEntity articleEntity = new ArticleEntity(); String columnIdStr = adminArticleDto.getColumnIdStr(); if(StrUtil.isNotEmpty(columnIdStr)){ long[] columnIdLongs = StrUtil.splitToLong(columnIdStr, StrUtil.COMMA); List columnList = new ArrayList<>(); for(long columnIdLong : columnIdLongs){ columnList.add(columnIdLong); } articleEntity.setColumnList(columnList); } String title = adminArticleDto.getTitle(); if(StrUtil.isNotEmpty(title)){ articleEntity.setTitle(title); } Integer contentType = adminArticleDto.getContentType() == null ? 0 : adminArticleDto.getContentType(); if(contentType != 0){ QueryWrapper objectQueryWrapper = new QueryWrapper<>(); objectQueryWrapper.eq("content_type",contentType); List columnEntities = columnMapper.selectList(objectQueryWrapper); if(CollUtil.isNotEmpty(columnEntities)){ List columIds = new ArrayList<>(); for(ColumnEntity columnEntity : columnEntities){ Long id = columnEntity.getId(); columIds.add(id); } articleEntity.setColumnIds(columIds); } } articleEntity.setCompanyId(24L); IPage list = articleMapper.selectAdminArticleInPage(page,articleEntity); System.out.println(list.getRecords()); } }