NPOI Excel DataTable操作类

/// <summary>
/// ExcelConfig
/// 描 述：Excel导入导出设置
/// </summary>
public class ExcelConfig
{
    /// <summary>
    /// 文件名
    /// </summary>
    public string FileName { get; set; }
    /// <summary>
    /// 标题
    /// </summary>
    public string Title { get; set; }
    /// <summary>
    /// 前景色
    /// </summary>
    public Color ForeColor { get; set; }
    /// <summary>
    /// 背景色
    /// </summary>
    public Color Background { get; set; }
    private short _titlepoint;
    /// <summary>
    /// 标题字号
    /// </summary>
    public short TitlePoint
    {
        get
        {
            if (_titlepoint == 0)
            {
                return 20;
            }
            else
            {
                return _titlepoint;
            }
        }
        set { _titlepoint = value; }
    }
    private short _headpoint;
    /// <summary>
    /// 列头字号
    /// </summary>
    public short HeadPoint
    {
        get
        {
            if (_headpoint == 0)
            {
                return 10;
            }
            else
            {
                return _headpoint;
            }
        }
        set { _headpoint = value; }
    }
    /// <summary>
    /// 标题高度
    /// </summary>
    public short TitleHeight { get; set; }
    /// <summary>
    /// 列标题高度
    /// </summary>
    public short HeadHeight { get; set; }
    private string _titlefont;
    /// <summary>
    /// 标题字体
    /// </summary>
    public string TitleFont
    {
        get
        {
            if (_titlefont == null)
            {
                return "微软雅黑";
            }
            else
            {
                return _titlefont;
            }
        }
        set { _titlefont = value; }
    }
    private string _headfont;
    /// <summary>
    /// 列头字体
    /// </summary>
    public string HeadFont
    {
        get
        {
            if (_headfont == null)
            {
                return "微软雅黑";
            }
            else
            {
                return _headfont;
            }
        }
        set { _headfont = value; }
    }
    /// <summary>
    /// 是否按内容长度来适应表格宽度
    /// </summary>
    public bool IsAllSizeColumn { get; set; }
    /// <summary>
    /// 列设置
    /// </summary>
    public List<ColumnModel> ColumnEntity { get; set; }

}

/// <summary>
/// ExcelHelper
/// 描 述：NPOI Excel DataTable操作类
/// </summary>
public class ExcelHelper
{
    #region Excel导出方法 ExcelDownload
    /// <summary>
    /// Excel导出下载
    /// </summary>
    /// <param name="dtSource">DataTable数据源</param>
    /// <param name="excelConfig">导出设置包含文件名、标题、列设置</param>
    public static void ExcelDownload(DataTable dtSource, ExcelConfig excelConfig)
    {
        HttpContext curContext = HttpContext.Current;
        // 设置编码和附件格式
        curContext.Response.ContentType = "application/ms-excel";
        curContext.Response.ContentEncoding = Encoding.UTF8;
        curContext.Response.Charset = "";
        curContext.Response.AppendHeader("Content-Disposition",
            "attachment;filename=" + HttpUtility.UrlEncode(excelConfig.FileName, Encoding.UTF8));
        //调用导出具体方法Export()
        curContext.Response.BinaryWrite(ExportMemoryStream(dtSource, excelConfig).GetBuffer());
        curContext.Response.End();
    }
    /// <summary>
    /// Excel导出下载
    /// </summary>
    /// <param name="list">数据源</param>
    /// <param name="templdateName">模板文件名</param>
    /// <param name="newFileName">文件名</param>
    public static void ExcelDownload(List<TemplateDataModel> list, string templdateName, string newFileName)
    {
        HttpResponse response = System.Web.HttpContext.Current.Response;
        response.Clear();
        response.Charset = "UTF-8";
        response.ContentType = "application/vnd-excel";//"application/vnd.ms-excel";
        System.Web.HttpContext.Current.Response.AddHeader("Content-Disposition", string.Format("attachment; filename=" + newFileName));
        System.Web.HttpContext.Current.Response.BinaryWrite(ExportListByTempale(list, templdateName).ToArray());
    }
    #endregion

    #region DataTable导出到Excel文件excelConfig中FileName设置为全路径
    /// <summary>
    /// DataTable导出到Excel文件 Export()
    /// </summary>
    /// <param name="dtSource">DataTable数据源</param>
    /// <param name="excelConfig">导出设置包含文件名、标题、列设置</param>
    public static void ExcelExport(DataTable dtSource, ExcelConfig excelConfig)
    {
        using (MemoryStream ms = ExportMemoryStream(dtSource, excelConfig))
        {
            using (FileStream fs = new FileStream(excelConfig.FileName, FileMode.Create, FileAccess.Write))
            {
                byte[] data = ms.ToArray();
                fs.Write(data, 0, data.Length);
                fs.Flush();
            }
        }
    }
    #endregion

    #region DataTable导出到Excel的MemoryStream
    /// <summary>
    /// DataTable导出到Excel的MemoryStream Export()
    /// </summary>
    /// <param name="dtSource">DataTable数据源</param>
    /// <param name="excelConfig">导出设置包含文件名、标题、列设置</param>
    public static MemoryStream ExportMemoryStream(DataTable dtSource, ExcelConfig excelConfig)
    {
        //int colint = 0;
        //for (int i = 0; i < dtSource.Columns.Count;)
        //{
        //    DataColumn column = dtSource.Columns[i];
        //    if (excelConfig.ColumnEntity[colint].Column != column.ColumnName)
        //    {
        //        dtSource.Columns.Remove(column.ColumnName);
        //    }
        //    else
        //    {
        //        i++;
        //        if (colint < excelConfig.ColumnEntity.Count - 1)
        //        {
        //            colint++;
        //        }
        //    }
        //}
        DataTable dt = dtSource.Clone();
        var columns = dt.Columns;
        foreach (DataColumn column in columns)
        {
            bool v = excelConfig.ColumnEntity.Exists(t => t.Column == column.ColumnName);
            if (!v)
            {
                dtSource.Columns.Remove(column.ColumnName);
            }
        }

        HSSFWorkbook workbook = new HSSFWorkbook();
        ISheet sheet = workbook.CreateSheet();
        #region 右击文件 属性信息
        {
            DocumentSummaryInformation dsi = PropertySetFactory.CreateDocumentSummaryInformation();
            dsi.Company = "NPOI";
            workbook.DocumentSummaryInformation = dsi;

            SummaryInformation si = PropertySetFactory.CreateSummaryInformation();
            si.Author = "刘晓雷"; //填加xls文件作者信息
            si.ApplicationName = "力软信息"; //填加xls文件创建程序信息
            si.LastAuthor = "刘晓雷"; //填加xls文件最后保存者信息
            si.Comments = "刘晓雷"; //填加xls文件作者信息
            si.Title = "标题信息"; //填加xls文件标题信息
            si.Subject = "主题信息";//填加文件主题信息
            si.CreateDateTime = System.DateTime.Now;
            workbook.SummaryInformation = si;
        }
        #endregion

        #region 设置标题样式
        ICellStyle headStyle = workbook.CreateCellStyle();
        int[] arrColWidth = new int[dtSource.Columns.Count];
        string[] arrColName = new string[dtSource.Columns.Count];//列名
        ICellStyle[] arryColumStyle = new ICellStyle[dtSource.Columns.Count];//样式表
        headStyle.Alignment = HorizontalAlignment.Center; // ------------------
        if (excelConfig.Background != new Color())
        {
            if (excelConfig.Background != new Color())
            {
                headStyle.FillPattern = FillPattern.SolidForeground;
                headStyle.FillForegroundColor = GetXLColour(workbook, excelConfig.Background);
            }
        }
        IFont font = workbook.CreateFont();
        font.FontHeightInPoints = excelConfig.TitlePoint;
        if (excelConfig.ForeColor != new Color())
        {
            font.Color = GetXLColour(workbook, excelConfig.ForeColor);
        }
        font.Boldweight = 700;
        headStyle.SetFont(font);
        #endregion

        #region 列头及样式
        ICellStyle cHeadStyle = workbook.CreateCellStyle();
        cHeadStyle.Alignment = HorizontalAlignment.Center; // ------------------
        IFont cfont = workbook.CreateFont();
        cfont.FontHeightInPoints = excelConfig.HeadPoint;
        cHeadStyle.SetFont(cfont);
        #endregion

        #region 设置内容单元格样式
        foreach (DataColumn item in dtSource.Columns)
        {
            ICellStyle columnStyle = workbook.CreateCellStyle();
            columnStyle.Alignment = HorizontalAlignment.Center;
            arrColWidth[item.Ordinal] = Encoding.GetEncoding(936).GetBytes(item.ColumnName.ToString()).Length;
            arrColName[item.Ordinal] = item.ColumnName.ToString();
            if (excelConfig.ColumnEntity != null)
            {
                ColumnModel columnentity = excelConfig.ColumnEntity.Find(t => t.Column == item.ColumnName);
                if (columnentity != null)
                {
                    arrColName[item.Ordinal] = columnentity.ExcelColumn;
                    if (columnentity.Width != 0)
                    {
                        arrColWidth[item.Ordinal] = columnentity.Width;
                    }
                    if (columnentity.Background != new Color())
                    {
                        if (columnentity.Background != new Color())
                        {
                            columnStyle.FillPattern = FillPattern.SolidForeground;
                            columnStyle.FillForegroundColor = GetXLColour(workbook, columnentity.Background);
                        }
                    }
                    if (columnentity.Font != null || columnentity.Point != 0 || columnentity.ForeColor != new Color())
                    {
                        IFont columnFont = workbook.CreateFont();
                        columnFont.FontHeightInPoints = 10;
                        if (columnentity.Font != null)
                        {
                            columnFont.FontName = columnentity.Font;
                        }
                        if (columnentity.Point != 0)
                        {
                            columnFont.FontHeightInPoints = columnentity.Point;
                        }
                        if (columnentity.ForeColor != new Color())
                        {
                            columnFont.Color = GetXLColour(workbook, columnentity.ForeColor);
                        }
                        columnStyle.SetFont(font);
                    }
                    columnStyle.Alignment = getAlignment(columnentity.Alignment);
                }
            }
            arryColumStyle[item.Ordinal] = columnStyle;
        }
        if (excelConfig.IsAllSizeColumn)
        {
            #region 根据列中最长列的长度取得列宽
            for (int i = 0; i < dtSource.Rows.Count; i++)
            {
                for (int j = 0; j < dtSource.Columns.Count; j++)
                {
                    if (arrColWidth[j] != 0)
                    {
                        int intTemp = Encoding.GetEncoding(936).GetBytes(dtSource.Rows[i][j].ToString()).Length;
                        if (intTemp > arrColWidth[j])
                        {
                            arrColWidth[j] = intTemp;
                        }
                    }

                }
            }
            #endregion
        }
        #endregion

        int rowIndex = 0;

        #region 表头及样式
        if (excelConfig.Title != null)
        {
            IRow headerRow = sheet.CreateRow(rowIndex);
            rowIndex++;
            if (excelConfig.TitleHeight != 0)
            {
                headerRow.Height = (short)(excelConfig.TitleHeight * 20);
            }
            
            headerRow.HeightInPoints = 25;
            headerRow.CreateCell(0).SetCellValue(excelConfig.Title);
            headerRow.GetCell(0).CellStyle = headStyle;
            sheet.AddMergedRegion(new NPOI.SS.Util.CellRangeAddress(0, 0, 0, dtSource.Columns.Count - 1)); // ------------------
        }
        #endregion

        #region 列头及样式
        {
            IRow headerRow = sheet.CreateRow(rowIndex);
            rowIndex++;
            #region 如果设置了列标题就按列标题定义列头，没定义直接按字段名输出
            foreach (DataColumn column in dtSource.Columns)
            {
                headerRow.CreateCell(column.Ordinal).SetCellValue(arrColName[column.Ordinal]);
                headerRow.GetCell(column.Ordinal).CellStyle = cHeadStyle;
                //设置列宽
                //sheet.SetColumnWidth(column.Ordinal, (arrColWidth[column.Ordinal] + 1) * 256);

                int colWidth = (arrColWidth[column.Ordinal] + 1) * 256;
                if (colWidth < 255 * 256)
                {
                    sheet.SetColumnWidth(column.Ordinal, colWidth < 5000 ? 5000 : colWidth);
                }
                else
                {
                    sheet.SetColumnWidth(column.Ordinal, 6000);
                }
            }
            #endregion
        }
        #endregion

        ICellStyle dateStyle = workbook.CreateCellStyle();
        IDataFormat format = workbook.CreateDataFormat();
        dateStyle.DataFormat = format.GetFormat("yyyy-mm-dd");
        bool isStyle = false;
        ICellStyle rowStyle = workbook.CreateCellStyle();
        foreach (DataRow row in dtSource.Rows)
        {
            #region 新建表，填充表头，填充列头，样式
            if (rowIndex == 65535)
            {
                sheet = workbook.CreateSheet();
                rowIndex = 0;
                #region 表头及样式
                {
                    if (excelConfig.Title != null)
                    {
                        IRow headerRow = sheet.CreateRow(rowIndex);
                        rowIndex++;
                        if (excelConfig.TitleHeight != 0)
                        {
                            headerRow.Height = (short)(excelConfig.TitleHeight * 20);
                        }
                        headerRow.HeightInPoints = 25;
                        headerRow.CreateCell(0).SetCellValue(excelConfig.Title);
                        sheet.AddMergedRegion(new NPOI.SS.Util.CellRangeAddress(0, 0, 0, dtSource.Columns.Count - 1)); // ------------------
                    }

                }
                #endregion

                #region 列头及样式
                {
                    IRow headerRow = sheet.CreateRow(rowIndex);
                    rowIndex++;
                    #region 如果设置了列标题就按列标题定义列头，没定义直接按字段名输出
                    foreach (DataColumn column in dtSource.Columns)
                    {
                        headerRow.CreateCell(column.Ordinal).SetCellValue(arrColName[column.Ordinal]);
                        headerRow.GetCell(column.Ordinal).CellStyle = cHeadStyle;
                        //设置列宽
                        sheet.SetColumnWidth(column.Ordinal, (arrColWidth[column.Ordinal] + 1) * 256);
                    }
                    #endregion
                }
                #endregion
            }
            #endregion

            #region 填充内容
            IRow dataRow = sheet.CreateRow(rowIndex);
            if (dtSource.Columns.Contains("重复物料提示") && row["重复物料提示"].ToString() == "更新前数据")
            {
                rowStyle.FillPattern = FillPattern.SolidForeground;
                rowStyle.FillForegroundColor = GetXLColour(workbook, Color.Yellow);
                dataRow.RowStyle = rowStyle;
                isStyle = true;
            }
            foreach (DataColumn column in dtSource.Columns)
            {

                ICell newCell = dataRow.CreateCell(column.Ordinal);
                newCell.CellStyle = isStyle?rowStyle: arryColumStyle[column.Ordinal];
                string drValue = row[column].ToString();
                SetCell(newCell, dateStyle, column.DataType, drValue);
            }
           
            #endregion
            rowIndex++;
            isStyle=false;
        }
        using (MemoryStream ms = new MemoryStream())
        {
            workbook.Write(ms);
            ms.Flush();
            ms.Position = 0;
            return ms;
        }
    }
    #endregion

    #region ListExcel导出(加载模板)
    /// <summary>
    /// List根据模板导出ExcelMemoryStream
    /// </summary>
    /// <param name="list"></param>
    /// <param name="templdateName"></param>
    public static MemoryStream ExportListByTempale(List<TemplateDataModel> list, string templdateName)
    {
        try
        {

            string templatePath = HttpContext.Current.Server.MapPath("/") + "/Resource/ExcelTemplate/";
            string templdateName1 = string.Format("{0}{1}", templatePath, templdateName);

            FileStream fileStream = new FileStream(templdateName1, FileMode.Open, FileAccess.Read);
            ISheet sheet = null;
            if (templdateName.IndexOf(".xlsx") == -1)//2003
            {
                HSSFWorkbook hssfworkbook = new HSSFWorkbook(fileStream);
                sheet = hssfworkbook.GetSheetAt(0);
                SetPurchaseOrder(sheet, list);
                sheet.ForceFormulaRecalculation = true;
                using (MemoryStream ms = new MemoryStream())
                {
                    hssfworkbook.Write(ms);
                    ms.Flush();
                    return ms;
                }
            }
            else//2007
            {
                XSSFWorkbook xssfworkbook = new XSSFWorkbook(fileStream);
                sheet = xssfworkbook.GetSheetAt(0);
                SetPurchaseOrder(sheet, list);
                sheet.ForceFormulaRecalculation = true;
                using (MemoryStream ms = new MemoryStream())
                {
                    xssfworkbook.Write(ms);
                    ms.Flush();
                    return ms;
                }
            }

        }
        catch (Exception)
        {
            throw;
        }
    }
    /// <summary>
    /// 赋值单元格
    /// </summary>
    /// <param name="sheet"></param>
    /// <param name="list"></param>
    private static void SetPurchaseOrder(ISheet sheet, List<TemplateDataModel> list)
    {
        try
        {
            foreach (var item in list)
            {
                IRow row = null;
                ICell cell = null;
                row = sheet.GetRow(item.row);
                if (row == null)
                {
                    row = sheet.CreateRow(item.row);
                }
                cell = row.GetCell(item.cell);
                if (cell == null)
                {
                    cell = row.CreateCell(item.cell);
                }
                cell.SetCellValue(item.value);
            }
        }
        catch (Exception)
        {
            throw;
        }
    }
    #endregion

    #region 设置表格内容
    private static void SetCell(ICell newCell, ICellStyle dateStyle, Type dataType, string drValue)
    {
        switch (dataType.ToString())
        {
            case "System.String"://字符串类型
                newCell.SetCellValue(drValue);
                break;
            case "System.DateTime"://日期类型
                System.DateTime dateV;
                if (System.DateTime.TryParse(drValue, out dateV))
                {
                    newCell.SetCellValue(dateV);
                }
                else
                {
                    newCell.SetCellValue("");
                }
                newCell.CellStyle = dateStyle;//格式化显示
                break;
            case "System.Boolean"://布尔型
                bool boolV = false;
                bool.TryParse(drValue, out boolV);
                newCell.SetCellValue(boolV);
                break;
            case "System.Int16"://整型
            case "System.Int32":
            case "System.Int64":
            case "System.Byte":
                int intV = 0;
                int.TryParse(drValue, out intV);
                newCell.SetCellValue(intV);
                break;
            case "System.Decimal"://浮点型
            case "System.Double":
                double doubV = 0;
                double.TryParse(drValue, out doubV);
                newCell.SetCellValue(doubV);
                break;
            case "System.DBNull"://空值处理
                newCell.SetCellValue("");
                break;
            default:
                newCell.SetCellValue("");
                break;
        }
    }
    #endregion

    #region 从Excel导入
    /// <summary>
    /// 读取excel ,默认第一行为标头
    /// </summary>
    /// <param name="strFileName">excel文档路径</param>
    /// <returns></returns>
    public static DataTable ExcelImport(string strFileName)
    {
        DataTable dt = new DataTable();

        ISheet sheet = null;
        using (FileStream file = new FileStream(strFileName, FileMode.Open, FileAccess.Read))
        {
            if (strFileName.IndexOf(".xlsx") == -1)//2003
            {
                HSSFWorkbook hssfworkbook = new HSSFWorkbook(file);
                sheet = hssfworkbook.GetSheetAt(0);
            }
            else//2007
            {
                XSSFWorkbook xssfworkbook = new XSSFWorkbook(file);
                sheet = xssfworkbook.GetSheetAt(0);
            }
        }

        System.Collections.IEnumerator rows = sheet.GetRowEnumerator();

        IRow headerRow = sheet.GetRow(0);
        int cellCount = headerRow.LastCellNum;

        for (int j = 0; j < cellCount; j++)
        {
            ICell cell = headerRow.GetCell(j);
            dt.Columns.Add(cell.ToString());
        }

        for (int i = (sheet.FirstRowNum + 1); i <= sheet.LastRowNum; i++)
        {
            IRow row = sheet.GetRow(i);
            if(!row.IsEmpty())  
            {
                DataRow dataRow = dt.NewRow();

                for (int j = row.FirstCellNum; j < cellCount; j++)
                {
                    if (row.GetCell(j) != null)
                        dataRow[j] = row.GetCell(j).ToString();
                }

                dt.Rows.Add(dataRow);
            }
        }
        return dt;
    }
    /// <summary>
    /// 读取excel ,默认第一行为标头 
    /// </summary>
    /// <param name="fileStream">文件数据流</param>
    /// <returns></returns>
    public static DataTable ExcelImport(Stream fileStream, string flieType)
    {
        DataTable dt = new DataTable();
        ISheet sheet = null;
        if (flieType == ".xls")
        {
            HSSFWorkbook hssfworkbook = new HSSFWorkbook(fileStream);
            sheet = hssfworkbook.GetSheetAt(0);
        }
        else
        {
            XSSFWorkbook xssfworkbook = new XSSFWorkbook(fileStream);
            sheet = xssfworkbook.GetSheetAt(0);
        }
        System.Collections.IEnumerator rows = sheet.GetRowEnumerator();
        IRow headerRow = sheet.GetRow(0);
        int cellCount = headerRow.LastCellNum;
        for (int j = 0; j < cellCount; j++)
        {
            ICell cell = headerRow.GetCell(j);
            dt.Columns.Add(cell.ToString());
        }
        for (int i = (sheet.FirstRowNum + 1); i <= sheet.LastRowNum; i++)
        {
            IRow row = sheet.GetRow(i);
            DataRow dataRow = dt.NewRow();

            for (int j = row.FirstCellNum; j < cellCount; j++)
            {
                if (row.GetCell(j) != null)
                    dataRow[j] = row.GetCell(j).ToString();
            }

            dt.Rows.Add(dataRow);
        }
        return dt;
    }
    #endregion

    #region RGB颜色转NPOI颜色
    private static short GetXLColour(HSSFWorkbook workbook, Color SystemColour)
    {
        short s = 0;
        HSSFPalette XlPalette = workbook.GetCustomPalette();
        NPOI.HSSF.Util.HSSFColor XlColour = XlPalette.FindColor(SystemColour.R, SystemColour.G, SystemColour.B);
        if (XlColour == null)
        {
            if (NPOI.HSSF.Record.PaletteRecord.STANDARD_PALETTE_SIZE < 255)
            {
                XlColour = XlPalette.FindSimilarColor(SystemColour.R, SystemColour.G, SystemColour.B);
                s = XlColour.Indexed;
            }

        }
        else
            s = XlColour.Indexed;
        return s;
    }
    #endregion

    #region 设置列的对齐方式
    /// <summary>
    /// 设置对齐方式
    /// </summary>
    /// <param name="style"></param>
    /// <returns></returns>
    private static HorizontalAlignment getAlignment(string style)
    {
        switch (style)
        {
            case "center":
                return HorizontalAlignment.Center;
            case "left":
                return HorizontalAlignment.Left;
            case "right":
                return HorizontalAlignment.Right;
            case "fill":
                return HorizontalAlignment.Fill;
            case "justify":
                return HorizontalAlignment.Justify;
            case "centerselection":
                return HorizontalAlignment.CenterSelection;
            case "distributed":
                return HorizontalAlignment.Distributed;
        }
        return NPOI.SS.UserModel.HorizontalAlignment.General;


    }

    #endregion
}