Merge branch 'v5-dev' of github.com:dromara/hutool into v5-dev

This commit is contained in:
Looly 2024-08-08 11:16:48 +08:00
commit 0aaa9019ad
3 changed files with 173 additions and 139 deletions

View File

@ -159,9 +159,18 @@ public class BigExcelWriter extends ExcelWriter {
return this;
}
@Override
public BigExcelWriter autoSizeColumnAll(float widthRatio) {
final SXSSFSheet sheet = (SXSSFSheet) this.sheet;
sheet.trackAllColumnsForAutoSizing();
super.autoSizeColumnAll(widthRatio);
sheet.untrackAllColumnsForAutoSizing();
return this;
}
@Override
public ExcelWriter flush(OutputStream out, boolean isCloseOut) throws IORuntimeException {
if (false == isFlushed) {
if (!isFlushed) {
isFlushed = true;
return super.flush(out, isCloseOut);
}
@ -170,7 +179,7 @@ public class BigExcelWriter extends ExcelWriter {
@Override
public void close() {
if (null != this.destFile && false == isFlushed) {
if (null != this.destFile && !isFlushed) {
flush();
}

View File

@ -30,12 +30,7 @@ import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.Charset;
import java.util.Comparator;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;
import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;
/**
@ -241,9 +236,22 @@ public class ExcelWriter extends ExcelBase<ExcelWriter> {
* @since 4.0.12
*/
public ExcelWriter autoSizeColumnAll() {
return autoSizeColumnAll(0f);
}
/**
* 设置所有列为自动宽度不考虑合并单元格<br>
* 此方法必须在指定列数据完全写出后调用才有效<br>
* 列数计算是通过第一行计算的
*
* @param widthRatio 列宽的倍数如果所有内容都是英文可以设为1如果有中文建议设置为 1.6-2.0之间
* @return this
* @since 5.8.30
*/
public ExcelWriter autoSizeColumnAll(float widthRatio) {
final int columnCount = this.getColumnCount();
for (int i = 0; i < columnCount; i++) {
autoSizeColumn(i);
autoSizeColumn(i, widthRatio);
}
return this;
}
@ -257,8 +265,7 @@ public class ExcelWriter extends ExcelBase<ExcelWriter> {
* @since 4.0.12
*/
public ExcelWriter autoSizeColumn(int columnIndex) {
this.sheet.autoSizeColumn(columnIndex);
return this;
return autoSizeColumn(columnIndex, false);
}
/**
@ -271,7 +278,37 @@ public class ExcelWriter extends ExcelBase<ExcelWriter> {
* @since 3.3.0
*/
public ExcelWriter autoSizeColumn(int columnIndex, boolean useMergedCells) {
this.sheet.autoSizeColumn(columnIndex, useMergedCells);
return autoSizeColumn(columnIndex, useMergedCells, 0f);
}
/**
* 设置某列为自动宽度注意有中文的情况下需要根据需求调整宽度扩大比例<br>
* 此方法必须在指定列数据完全写出后调用才有效
*
* @param columnIndex 第几列从0计数
* @param widthRatio 列宽的倍数如果所有内容都是英文可以设为1如果有中文建议设置为 1.6-2.0之间
* @return this
* @since 5.8.30
*/
public ExcelWriter autoSizeColumn(int columnIndex, float widthRatio) {
return autoSizeColumn(columnIndex, false, widthRatio);
}
/**
* 设置某列为自动宽度注意有中文的情况下需要根据需求调整宽度扩大比例<br>
* 此方法必须在指定列数据完全写出后调用才有效
*
* @param columnIndex 第几列从0计数
* @param useMergedCells 是否适用于合并单元格
* @param widthRatio 列宽的倍数如果所有内容都是英文可以设为1如果有中文建议设置为 1.6-2.0之间
* @return this
* @since 5.8.30
*/
public ExcelWriter autoSizeColumn(int columnIndex, boolean useMergedCells, float widthRatio) {
sheet.autoSizeColumn(columnIndex, useMergedCells);
if (widthRatio > 0) {
sheet.setColumnWidth(columnIndex, (int) (sheet.getColumnWidth(columnIndex) * widthRatio));
}
return this;
}

View File

@ -12,15 +12,7 @@ import cn.hutool.core.util.ObjectUtil;
import cn.hutool.poi.excel.cell.setters.EscapeStrCellSetter;
import cn.hutool.poi.excel.style.StyleUtil;
import org.apache.poi.common.usermodel.HyperlinkType;
import org.apache.poi.ss.usermodel.BorderStyle;
import org.apache.poi.ss.usermodel.BuiltinFormats;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.FillPatternType;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.usermodel.Hyperlink;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.ss.usermodel.VerticalAlignment;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddressList;
import org.apache.poi.xssf.usermodel.XSSFRichTextString;
import org.junit.Assert;
@ -28,15 +20,7 @@ import org.junit.Ignore;
import org.junit.Test;
import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;
import java.util.*;
/**
* 写出Excel单元测试
@ -45,6 +29,23 @@ import java.util.TreeMap;
*/
public class ExcelWriteTest {
public static Map<String, Object> MAP_DATA_1 = new LinkedHashMap<>();
public static Map<String, Object> MAP_DATA_2 = new LinkedHashMap<>();
static {
MAP_DATA_1.put("姓名", "张三");
MAP_DATA_1.put("年龄", 23);
MAP_DATA_1.put("成绩", 88.32);
MAP_DATA_1.put("是否合格", true);
MAP_DATA_1.put("考试日期", DateUtil.date());
MAP_DATA_2.put("姓名", "张三");
MAP_DATA_2.put("年龄", 23);
MAP_DATA_2.put("成绩", 88.32);
MAP_DATA_2.put("是否合格", true);
MAP_DATA_2.put("考试日期", DateUtil.date());
}
@Test
public void writeNoFlushTest() {
List<?> row1 = CollUtil.newArrayList("aaaaa", "bb", "cc", "dd", DateUtil.date(), 3.22676575765);
@ -200,26 +201,13 @@ public class ExcelWriteTest {
@Test
@Ignore
public void mergeTest2() {
Map<String, Object> row1 = new LinkedHashMap<>();
row1.put("姓名", "张三");
row1.put("年龄", 23);
row1.put("成绩", 88.32);
row1.put("是否合格", true);
row1.put("考试日期", DateUtil.date());
Map<String, Object> row2 = new LinkedHashMap<>();
row2.put("姓名", "李四");
row2.put("年龄", 33);
row2.put("成绩", 59.50);
row2.put("是否合格", false);
row2.put("考试日期", DateUtil.date());
ArrayList<Map<String, Object>> rows = CollUtil.newArrayList(row1, row2);
ArrayList<Map<String, Object>> rows = CollUtil.newArrayList(MAP_DATA_1, MAP_DATA_2);
// 通过工具类创建writer
ExcelWriter writer = ExcelUtil.getWriter("d:/test/writeMapTest.xlsx");
// 合并单元格后的标题行使用默认标题样式
writer.merge(row1.size() - 1, "一班成绩单");
writer.merge(MAP_DATA_1.size() - 1, "一班成绩单");
// 一次性写出内容使用默认样式强制输出标题
writer.write(rows, true);
@ -230,21 +218,8 @@ public class ExcelWriteTest {
@Test
@Ignore
public void writeMapTest() {
Map<String, Object> row1 = new LinkedHashMap<>();
row1.put("姓名", "张三");
row1.put("年龄", 23);
row1.put("成绩", 88.32);
row1.put("是否合格", true);
row1.put("考试日期", DateUtil.date());
Map<String, Object> row2 = new LinkedHashMap<>();
row2.put("姓名", "李四");
row2.put("年龄", 33);
row2.put("成绩", 59.50);
row2.put("是否合格", false);
row2.put("考试日期", DateUtil.date());
ArrayList<Map<String, Object>> rows = CollUtil.newArrayList(row1, row2);
ArrayList<Map<String, Object>> rows = CollUtil.newArrayList(MAP_DATA_1, MAP_DATA_2);
// 通过工具类创建writer
ExcelWriter writer = ExcelUtil.getWriter("e:/excel/writeMapTest.xlsx");
@ -257,7 +232,7 @@ public class ExcelWriteTest {
writer.getStyleSet().setFont(font, true);
// 合并单元格后的标题行使用默认标题样式
writer.merge(row1.size() - 1, "一班成绩单");
writer.merge(MAP_DATA_1.size() - 1, "一班成绩单");
// 一次性写出内容使用默认样式
writer.write(rows, true);
// 关闭writer释放内存
@ -267,18 +242,12 @@ public class ExcelWriteTest {
@Test
@Ignore
public void writeMapTest2() {
Map<String, Object> row1 = MapUtil.newHashMap(true);
row1.put("姓名", "张三");
row1.put("年龄", 23);
row1.put("成绩", 88.32);
row1.put("是否合格", true);
row1.put("考试日期", DateUtil.date());
// 通过工具类创建writer
ExcelWriter writer = ExcelUtil.getWriter("e:/writeMapTest2.xlsx");
ExcelWriter writer = ExcelUtil.getWriter("d:/writeMapTest2.xlsx");
// 一次性写出内容使用默认样式
writer.writeRow(row1, true);
writer.writeRow(MAP_DATA_1, true);
// 关闭writer释放内存
writer.close();
}
@ -286,12 +255,6 @@ public class ExcelWriteTest {
@Test
@Ignore
public void writeMapWithStyleTest() {
Map<String, Object> row1 = MapUtil.newHashMap(true);
row1.put("姓名", "张三");
row1.put("年龄", 23);
row1.put("成绩", 88.32);
row1.put("是否合格", true);
row1.put("考试日期", DateUtil.date());
// 通过工具类创建writer
String path = "f:/test/writeMapWithStyleTest.xlsx";
@ -300,7 +263,7 @@ public class ExcelWriteTest {
writer.setStyleSet(null);
// 一次性写出内容使用默认样式
writer.writeRow(row1, true);
writer.writeRow(MAP_DATA_1, true);
// 设置某个单元格样式
CellStyle orCreateRowStyle = writer.getOrCreateCellStyle(0, 1);
@ -335,8 +298,8 @@ public class ExcelWriteTest {
writer.addHeaderAlias("name", "姓名");
writer.addHeaderAlias("age", "年龄");
writer.addHeaderAlias("score", "分数");
writer.addHeaderAlias("isPass", "是否通过");
writer.addHeaderAlias("examDate", "考试时间");
writer.addHeaderAlias("isPass", "是否合格");
writer.addHeaderAlias("examDate", "考试日期");
// 合并单元格后的标题行使用默认标题样式
writer.merge(4, "一班成绩单");
// 一次性写出内容使用默认样式
@ -445,21 +408,21 @@ public class ExcelWriteTest {
@Test
@Ignore
public void writeBeanTest() {
cn.hutool.poi.excel.TestBean bean1 = new cn.hutool.poi.excel.TestBean();
TestBean bean1 = new TestBean();
bean1.setName("张三");
bean1.setAge(22);
bean1.setPass(true);
bean1.setScore(66.30);
bean1.setExamDate(DateUtil.date());
cn.hutool.poi.excel.TestBean bean2 = new cn.hutool.poi.excel.TestBean();
TestBean bean2 = new TestBean();
bean2.setName("李四");
bean2.setAge(28);
bean2.setPass(false);
bean2.setScore(38.50);
bean2.setExamDate(DateUtil.date());
List<cn.hutool.poi.excel.TestBean> rows = CollUtil.newArrayList(bean1, bean2);
List<TestBean> rows = CollUtil.newArrayList(bean1, bean2);
// 通过工具类创建writer
String file = "e:/writeBeanTest.xlsx";
FileUtil.del(file);
@ -481,17 +444,17 @@ public class ExcelWriteTest {
@Test
@Ignore
public void writeBeanTest2() {
cn.hutool.poi.excel.OrderExcel order1 = new cn.hutool.poi.excel.OrderExcel();
OrderExcel order1 = new OrderExcel();
order1.setId("1");
order1.setNum("123");
order1.setBody("body1");
cn.hutool.poi.excel.OrderExcel order2 = new cn.hutool.poi.excel.OrderExcel();
OrderExcel order2 = new OrderExcel();
order1.setId("2");
order1.setNum("456");
order1.setBody("body2");
List<cn.hutool.poi.excel.OrderExcel> rows = CollUtil.newArrayList(order1, order2);
List<OrderExcel> rows = CollUtil.newArrayList(order1, order2);
// 通过工具类创建writer
String file = "f:/test/writeBeanTest2.xlsx";
FileUtil.del(file);
@ -916,4 +879,29 @@ public class ExcelWriteTest {
writer.close();
}
@Test
@Ignore
public void autoSizeColumnTest() {
Map<String, Object> map = new LinkedHashMap<>(MAP_DATA_1);
map.put("中文长度测试(符号)", "abc");
String file1 = "d:/autoSizeColumnTest.xlsx";
String file2 = "d:/autoSizeColumnTest2.xlsx";
FileUtil.del(file1);
FileUtil.del(file2);
try (ExcelWriter writer = new ExcelWriter(file1)) {
writer.writeRow(map, true);
writer.autoSizeColumnAll(2f);
}
try (BigExcelWriter writer = new BigExcelWriter(file2)) {
writer.writeRow(map, true);
writer.autoSizeColumnAll(2f);
}
}
}