HaiNan/ReportDataResumeTool/Common/ExcelHelper.cs

using NPOI.HSSF.UserModel;
using NPOI.SS.UserModel;
using NPOI.XSSF.UserModel;
using System;
using System.Collections.Generic;
using System.Data;
using System.IO;

namespace ReportDataResumeTool.Common
{
    /// <summary>
    /// Excel处理服务类 
    /// </summary>
    public class ExcelHelper
    {
        private string fileName = null; //文件名
        private IWorkbook workbook = null;
        private FileStream fs = null;
        private bool disposed;
        /// <summary>
        /// 构造函数，读入文件名
        /// </summary>
        public ExcelHelper(string fileName)
        {
            this.fileName = fileName;
            disposed = false;
        }

        /// <summary>
        /// 将excel中的数据导入到DataTable中 
        /// </summary> 
        /// <param name="sheetName">excel工作薄sheet的名称</param>
        /// <param name="isFirstRowColumn">第一行是否是DataTable的列名</param>
        /// <returns>返回的DataTable</returns>
        public DataTable ExcelToDataTable(string sheetName, bool isFirstRowColumn)
        {
            ISheet sheet = null;
            DataTable data = new DataTable();
            int startRow = 0;
            fs = new FileStream(fileName, FileMode.Open, FileAccess.Read);
            workbook = WorkbookFactory.Create(fs);
            if (sheetName != null)
            {
                sheet = workbook.GetSheet(sheetName);
                //如果没有找到指定的sheetName对应的sheet，则尝试获取第一个sheet
                if (sheet == null)
                {
                    sheet = workbook.GetSheetAt(0);
                }
            }
            else
            {
                sheet = workbook.GetSheetAt(0);
            }
            if (sheet != null)
            {
                IRow firstRow = sheet.GetRow(0);
                int cellCount = firstRow.LastCellNum; //一行最后一个cell的编号，即总的列数
                // int cellCount = ;
                if (isFirstRowColumn)
                {
                    for (int i = firstRow.FirstCellNum; i < cellCount; ++i)
                    {
                        ICell cell = firstRow.GetCell(i);
                        if (cell != null)
                        {
                            string cellValue = cell.StringCellValue;
                            if (cellValue != null)
                            {
                                DataColumn column = new DataColumn(cellValue);
                                data.Columns.Add(column);
                            }
                        }
                    }
                    cellCount = data.Columns.Count;
                    startRow = sheet.FirstRowNum + 1;//得到项标题后
                }
                else
                {
                    startRow = sheet.FirstRowNum;
                }
                //最后一列的标号
                int rowCount = sheet.LastRowNum;
                for (int i = startRow; i <= rowCount; ++i)
                {
                    IRow row = sheet.GetRow(i);
                    if (row == null) continue; //没有数据的行默认是null　　　　　　　

                    DataRow dataRow = data.NewRow();
                    for (int j = row.FirstCellNum; j < cellCount; ++j)
                    {
                        // MessageBox.Show(row.GetCell(j).ToString());
                        if (row.GetCell(j) != null)
                        { //同理，没有数据的单元格都默认是null
                            dataRow[j] = row.GetCell(j).ToString();
                        }
                        else
                        {
                            dataRow[j] = "";
                        }

                    }
                    data.Rows.Add(dataRow);

                }
            }
            return data;
        }

        /// <summary>
        /// 将DataTable数据导入到excel中
        /// </summary> 
        /// <param name="data">要导入的数据</param>
        /// <param name="sheetName">要导入的excel的sheet的名称</param>
        /// <param name="isColumnWritten">DataTable的列名是否要导入</param>
        /// <returns>导入数据行数(包含列名那一行)</returns>
        public int DataTableToExcel(DataTable data, string sheetName, bool isColumnWritten, IList<string> titleList = null)
        {
            int page = 1;
            int i = 0;
            int j = 0;
            int count = 0;
            ISheet sheet = null;

            fs = new FileStream(fileName, FileMode.OpenOrCreate, FileAccess.ReadWrite);
            if (fileName.IndexOf(".xlsx") > 0) // 2007版本
                workbook = new XSSFWorkbook();
            else if (fileName.IndexOf(".xls") > 0) // 2003版本
                workbook = new HSSFWorkbook();

            try
            {
                if (workbook != null)
                {
                    if (sheetName.Length > 31)
                    {
                        sheetName = sheetName.Substring(0, 31);
                    }
                    sheet = workbook.CreateSheet(sheetName);
                    sheet.FitToPage = false;
                }
                else
                {
                    return -1;
                }

                if (isColumnWritten == true) //写入DataTable的列名
                {
                    IRow row = sheet.CreateRow(0);
                    for (j = 0; j < data.Columns.Count; ++j)
                    {
                        sheet.SetColumnWidth(j, data.Columns[j].ColumnName.Length * 1800);
                        row.CreateCell(j).SetCellValue(data.Columns[j].ColumnName);
                        if (titleList?.Count > 0 && titleList.Contains(data.Columns[j].ColumnName))
                        {
                            row.GetCell(j).CellStyle = SetTitleDataStyle();
                        }
                    }
                    count = 1;
                }
                else
                {
                    count = 0;
                }

                for (i = 0; i < data.Rows.Count; ++i)
                {
                    IRow row = sheet.CreateRow(count);
                    for (j = 0; j < data.Columns.Count; ++j)
                    {
                        row.CreateCell(j).SetCellValue(data.Rows[i][j].ToString());
                        if (titleList?.Count > 0 && titleList.Contains(data.Rows[i][j].ToString()))
                        {
                            row.GetCell(j).CellStyle = SetTitleDataStyle();
                        }
                    }
                    if (i > 5 && i % 10000 == 0)
                    {
                        sheet = workbook.CreateSheet(sheetName + page.ToString());
                        IRow row1 = sheet.CreateRow(0);
                        for (j = 0; j < data.Columns.Count; ++j)
                        {
                            sheet.SetColumnWidth(j, data.Columns[j].ColumnName.Length * 384);
                            row1.CreateCell(j).SetCellValue(data.Columns[j].ColumnName);
                            if (titleList?.Count > 0 && titleList.Contains(data.Columns[j].ColumnName))
                            {
                                row1.GetCell(j).CellStyle = SetTitleDataStyle();
                            }
                        }
                        count = 0;

                        page++;
                    }
                    ++count;
                }

                workbook.Write(fs, false); //写入到excel
                return count + page * 5000;
            }
            catch (Exception ex)
            {
                Console.WriteLine("Exception: " + ex.Message);
                return -1;
            }
        }

        /// <summary>
        /// 设置标题样式
        /// </summary> 
        /// <returns>返回单元格样式</returns>
        private ICellStyle SetTitleDataStyle()
        {
            ICellStyle dataStyle = workbook.CreateCellStyle();
            dataStyle.BorderBottom = BorderStyle.None;
            dataStyle.BorderLeft = BorderStyle.None;
            dataStyle.BorderRight = BorderStyle.None;
            dataStyle.BorderTop = BorderStyle.None;
            dataStyle.VerticalAlignment = VerticalAlignment.Center;
            dataStyle.WrapText = false;
            IFont font = workbook.CreateFont();
            font.FontHeightInPoints = 30;
            font.IsBold = true;
            return dataStyle;
        }

        /// <summary>
        /// 将List数据导入到excel中
        /// </summary> 
        /// <param name="dataList">要导入的数据</param>
        /// <param name="sheetName">要导入的excel的sheet的名称</param>
        /// <param name="isColumnWritten">DataTable的列名是否要导入</param>
        /// <returns>导入数据行数(包含列名那一行)</returns>
        public int ListToExcel<T>(IList<T> dataList, string sheetName, bool isColumnWritten, IList<string> titleList = null)
        {
            if (dataList == null || dataList.Count <= 0)
            {
                return 0;
            }
            var dt = ConvertHelper.ListToDataTable<T>(dataList);
            var res = DataTableToExcel(dt, sheetName, isColumnWritten, titleList);
            Dispose();
            return res;
        }

        /// <summary>
        /// 将excel数据读取到list中
        /// </summary> 
        /// <param name="sheetName">要导入的excel的sheet的名称,例如：Sheet1</param>
        /// <param name="isFirstRowPropertyName">第一行是否对象属性名称</param>
        /// <returns>导入数据行数(包含列名那一行)</returns>
        public IList<T> ReadExcelToList<T>(string sheetName, bool isFirstRowPropertyName) where T : class, new()
        {
            var dt = ExcelToDataTable(sheetName, isFirstRowPropertyName);
            var list = ConvertHelper.DataTableToList<T>(dt);
            Dispose();
            return list;
        }

        /// <summary>
        /// 释放资源
        /// </summary> 
        public void Dispose()
        {
            if (!this.disposed)
            {
                if (fs != null)
                    fs.Close();
                fs = null;
                disposed = true;
            }
            GC.SuppressFinalize(this);
        }
    }

}