diff --git a/sdk/common/src/main/java/io/dataease/model/excel/AutoAdaptWidthStyleStrategy.java b/sdk/common/src/main/java/io/dataease/model/excel/AutoAdaptWidthStyleStrategy.java index 0b1ac2e9c6..8655ff437e 100644 --- a/sdk/common/src/main/java/io/dataease/model/excel/AutoAdaptWidthStyleStrategy.java +++ b/sdk/common/src/main/java/io/dataease/model/excel/AutoAdaptWidthStyleStrategy.java @@ -4,14 +4,18 @@ import com.alibaba.excel.enums.CellDataTypeEnum; import com.alibaba.excel.metadata.Head; import com.alibaba.excel.metadata.data.CellData; import com.alibaba.excel.metadata.data.WriteCellData; +import com.alibaba.excel.write.handler.context.CellWriteHandlerContext; import com.alibaba.excel.write.metadata.holder.WriteSheetHolder; import com.alibaba.excel.write.style.column.AbstractColumnWidthStyleStrategy; +import io.dataease.i18n.Translator; import org.apache.commons.collections4.CollectionUtils; +import org.apache.commons.lang3.ObjectUtils; import org.apache.poi.ss.usermodel.Cell; import java.util.HashMap; import java.util.List; import java.util.Map; +import java.util.stream.Collectors; public class AutoAdaptWidthStyleStrategy extends AbstractColumnWidthStyleStrategy { @@ -66,4 +70,15 @@ public class AutoAdaptWidthStyleStrategy extends AbstractColumnWidthStyleStrateg } } } + + @Override + public void beforeCellCreate(CellWriteHandlerContext context) { + if (!context.getHead()) return; + Head headData = context.getHeadData(); + List headNameList = null; + if (ObjectUtils.isNotEmpty(headData) && CollectionUtils.isNotEmpty(headNameList = headData.getHeadNameList())) { + List i18nNameList = headNameList.stream().map(Translator::get).collect(Collectors.toList()); + context.getHeadData().setHeadNameList(i18nNameList); + } + } }