/// <summary>
/// 转换成工作簿
/// </summary>
/// <typeparam name="T">实体</typeparam>
/// <param name="source">数据源</param>
/// <param name="excelFile">文件路径</param>
/// <param name="sheetName">工作表名</param>
/// <returns></returns>
internal static IWorkbook ToWorkbook <T>(this IEnumerable <T> source, string excelFile, string sheetName)
{
// 静态属性还是实例属性
var properties =
typeof(T).GetProperties(BindingFlags.Public | BindingFlags.Instance | BindingFlags.GetProperty);
// 获取Fluent配置
bool fluentConfigEnabled = false;
IFluentConfiguration fluentConfig;
if (Excel.Setting.FluentConfigs.TryGetValue(typeof(T), out fluentConfig))
{
fluentConfigEnabled = true;
}
// 获取单元格设置
var cellConfigs = new CellConfig[properties.Length];
for (var i = 0; i < properties.Length; i++)
{
PropertyConfiguration pc;
var property = properties[i];
// 优先获取配置（高优先级），没有则获取属性
if (fluentConfigEnabled && fluentConfig.PropertyConfigs.TryGetValue(property, out pc))
{
cellConfigs[i] = pc.CellConfig;
}
else
{
var attrs = property.GetCustomAttributes(typeof(NpoiColumnAttribute), true) as NpoiColumnAttribute[];
if (attrs != null && attrs.Length > 0)
{
cellConfigs[i] = attrs[0].CellConfig;
}
else
{
cellConfigs[i] = null;
}
}
}
// 初始化工作簿
var workbook = InitializeWorkbook(excelFile);
// 创建工作表
var sheet = workbook.GetOrCreateSheet(sheetName);
// 缓存单元格样式
var cellStyles = new Dictionary <int, ICellStyle>();
// 标题行单元格样式
var titleStyle = workbook.CreateCellStyle();
titleStyle.Alignment = HorizontalAlignment.Center;
titleStyle.VerticalAlignment = VerticalAlignment.Center;
//titleStyle.FillPattern=FillPattern.Bricks;
titleStyle.FillBackgroundColor = HSSFColor.Grey40Percent.Index;
titleStyle.FillForegroundColor = HSSFColor.White.Index;
var titleRow = sheet.GetOrCreateRow(0);
var rowIndex = 1;
foreach (var item in source)
{
var row = sheet.GetOrCreateRow(rowIndex);
for (int i = 0; i < properties.Length; i++)
{
var property = properties[i];
int index = i;
var config = cellConfigs[i];
if (config != null)
{
if (config.IsIgnored)
{
continue;
}
index = config.Index;
}
// 首次
if (rowIndex == 1)
{
//如果没有标题，则使用属性名称作为标题
var title = property.Name;
if (config != null)
{
if (!string.IsNullOrWhiteSpace(config.Title))
{
title = config.Title;
}
if (!string.IsNullOrWhiteSpace(config.Formatter))
{
try
{
var style = workbook.CreateCellStyle();
var dataFormat = workbook.CreateDataFormat();
style.DataFormat = dataFormat.GetFormat(config.Formatter);
cellStyles[i] = style;
}
catch (Exception ex)
{
System.Diagnostics.Debug.WriteLine(ex.ToString());
}
}
var titleCell = titleRow.GetOrCreateCell(index);
titleCell.CellStyle = titleStyle;
titleCell.SetCellValue(title);
}
}
var value = property.GetValue(item, null);
if (value == null)
{
continue;
}
var cell = row.GetOrCreateCell(index);
ICellStyle cellStyle;
if (cellStyles.TryGetValue(i, out cellStyle))
{
cell.CellStyle = cellStyle;
}
var unwrapType = property.PropertyType.UnwrapNullableType();
cell.SetCellValueExt(value, unwrapType, config.CustomEnum, config.Formatter);
}
rowIndex++;
}
// 合并单元格
var mergableConfigs = cellConfigs.Where(c => c != null && c.AllowMerge).ToList();
if (mergableConfigs.Any())
{
// 合并单元格样式
var vStyle = workbook.CreateCellStyle();
vStyle.VerticalAlignment = VerticalAlignment.Center;
foreach (var config in mergableConfigs)
{
object previous = null;
int rowspan = 0, row = 1;
for (row = 1; row < rowIndex; row++)
{
var value = sheet.GetRow(row).GetCellValue(config.Index);
if (object.Equals(previous, value) && value != null)
{
rowspan++;
}
else
{
if (rowspan > 1)
{
sheet.GetRow(row - rowspan).Cells[config.Index].CellStyle = vStyle;
sheet.AddMergedRegion(new CellRangeAddress(row - rowspan, row - 1, config.Index,
config.Index));
}
rowspan = 1;
previous = value;
}
}
// 什么情况下——>所有行都需要合并
if (rowspan > 1)
{
sheet.GetRow(row - rowspan).Cells[config.Index].CellStyle = vStyle;
sheet.MergeCell(row - rowspan, row - 1, config.Index, config.Index);
}
}
}
if (rowIndex > 1)
{
var statisticsConfigs = new List <StatisticsConfig>();
var filterConfigs = new List <FilterConfig>();
var freezeConfigs = new List <FreezeConfig>();
if (fluentConfigEnabled)
{
statisticsConfigs.AddRange(fluentConfig.StatisticsConfigs);
filterConfigs.AddRange(fluentConfig.FilterConfigs);
freezeConfigs.AddRange(fluentConfig.FreezeConfigs);
}
else
{
var statistics =
typeof(T).GetCustomAttributes(typeof(NpoiStatisticsAttribute), true) as NpoiStatisticsAttribute[];
if (statistics != null && statistics.Length > 0)
{
foreach (var item in statistics)
{
statisticsConfigs.Add(item.StatisticsConfig);
}
}
var freezes =
typeof(T).GetCustomAttributes(typeof(NpoiFreezeAttribute), true) as NpoiFreezeAttribute[];
if (freezes != null && freezes.Length > 0)
{
foreach (var item in freezes)
{
freezeConfigs.Add(item.FreezeConfig);
}
}
var filters =
typeof(T).GetCustomAttributes(typeof(NpoiFilterAttribute), true) as NpoiFilterAttribute[];
if (filters != null && filters.Length > 0)
{
foreach (var item in filters)
{
filterConfigs.Add(item.FilterConfig);
}
}
}
// 统计行
foreach (var item in statisticsConfigs)
{
var lastRow = sheet.CreateRow(rowIndex);
var cell = lastRow.CreateCell(0);
cell.SetCellValue(item.Name);
foreach (var column in item.Columns)
{
cell = lastRow.CreateCell(column);
cell.CellFormula =
$"{item.Formula}({GetCellPosition(1, column)}:{GetCellPosition(rowIndex - 1, column)})";
}
rowIndex++;
}
// 设置冻结窗格
foreach (var item in freezeConfigs)
{
sheet.CreateFreezePane(item.ColSplit, item.RowSplit, item.LeftMostColumn, item.TopRow);
}
// 设置自动筛选
foreach (var item in filterConfigs)
{
sheet.SetAutoFilter(new CellRangeAddress(item.FirstRow, item.LastRow ?? rowIndex, item.FirstCol,
item.LastCol));
}
}
// 自动调整所有列
for (int i = 0; i < properties.Length; i++)
{
sheet.AutoSizeColumn(i);
}
return(workbook);
}
/// <summary>
/// 初始化一个<see cref="NpoiColumnAttribute"/>类型的实例
/// </summary>
public NpoiColumnAttribute()
{
CellConfig = new CellConfig();
}