C#导出数据到CSV文件的通用类实例

前端技术 2023/09/08 C#

本文实例讲述了C#导出数据到csv文件的通用类。分享给大家供大家参考。具体如下:

通过这个类可以很简单的定义数据格式,并导出到csv文件

//这里写了一个通用的类
using System;
using System.Data;
using System.Configuration;
using System.Collections.Generic;
using System.Web;
using System.Web.Security;
using System.Web.UI;
using System.Web.UI.WebControls;
using System.Web.UI.WebControls.WebParts;
using System.Web.UI.HtmlControls;
using System.Text;
using System.Reflection;
using System.IO;
using System.Data.Odbc;
namespace Com.DRPENG.SDXY.UI.Common
{
 public class CSVHelper
 {
  #region Fields
  string _fileName;
  DataTable _dataSource;//数据源
  string[] _titles = null;//列标题
  string[] _fields = null;//字段名
  #endregion
  #region .ctor
  /// <summary> 
  /// 构造函数 
  /// </summary> 
  /// <param name=\"dataSource\">数据源</param> 
  public CSVHelper()
  {
  }
  /// <summary> 
  /// 构造函数 
  /// </summary> 
  /// <param name=\"titles\">要输出到 Excel 的列标题的数组</param> 
  /// <param name=\"fields\">要输出到 Excel 的字段名称数组</param> 
  /// <param name=\"dataSource\">数据源</param> 
  public CSVHelper(string[] titles, string[] fields, DataTable dataSource)
   : this(titles, dataSource)
  {
   if (fields == null || fields.Length == 0)
    throw new ArgumentNullException(\"fields\");
   if (titles.Length != fields.Length)
    throw new ArgumentException(\"titles.Length != fields.Length\", \"fields\");
   _fields = fields;
  }
  /// <summary> 
  /// 构造函数 
  /// </summary> 
  /// <param name=\"titles\">要输出到 Excel 的列标题的数组</param> 
  /// <param name=\"dataSource\">数据源</param> 
  public CSVHelper(string[] titles, DataTable dataSource)
   : this(dataSource)
  {
   if (titles == null || titles.Length == 0)
    throw new ArgumentNullException(\"titles\");
   _titles = titles;
  }
  /// <summary> 
  /// 构造函数 
  /// </summary> 
  /// <param name=\"dataSource\">数据源</param> 
  public CSVHelper(DataTable dataSource)
  {
   if (dataSource == null)
    throw new ArgumentNullException(\"dataSource\");
   // maybe more checks needed here (IEnumerable, IList, IListSource, ) ??? 
   // 很难判断,先简单的使用 DataTable 
   _dataSource = dataSource;
  }
  #endregion
  #region public Methods
  #region 导出到CSV文件并且提示下载
  /// <summary>
  /// 导出到CSV文件并且提示下载
  /// </summary>
  /// <param name=\"fileName\"></param>
  public void DataToCSV(string fileName)
  {
   // 确保有一个合法的输出文件名 
   //if (fileName == null || fileName == string.Empty || !(fileName.ToLower().EndsWith(\".csv\")))
   // fileName = GetRandomFileName();
   string data = ExportCSV();
   HttpContext.Current.Response.ClearHeaders();
   HttpContext.Current.Response.Clear();
   HttpContext.Current.Response.Expires = 0;
   HttpContext.Current.Response.BufferOutput = true;
   HttpContext.Current.Response.Charset = \"GB2312\";
   HttpContext.Current.Response.ContentEncoding = System.Text.Encoding.GetEncoding(\"GB2312\");
   HttpContext.Current.Response.AppendHeader(\"Content-Disposition\", string.Format(\"attachment;filename={0}.csv\", System.Web.HttpUtility.UrlEncode(fileName, System.Text.Encoding.UTF8)));
   HttpContext.Current.Response.ContentType = \"text/h323;charset=gbk\";
   HttpContext.Current.Response.Write(data);
   HttpContext.Current.Response.End();
  }
  #endregion
  /// <summary>
  /// 获取CSV导入的数据
  /// </summary>
  /// <param name=\"filePath\">文件路径</param>
  /// <param name=\"fileName\">文件名称(.csv不用加)</param>
  /// <returns></returns>
  public DataTable GetCsvData(string filePath,string fileName)
  {
   string path = Path.Combine(filePath, fileName + \".csv\");
   string connString = @\"Driver={Microsoft Text Driver (*.txt; *.csv)};Dbq=\" + filePath + \";Extensions=asc,csv,tab,txt;\";
   try
   {
    using (OdbcConnection odbcConn = new OdbcConnection(connString))
    {
     odbcConn.Open();
     OdbcCommand oleComm = new OdbcCommand();
     oleComm.Connection = odbcConn;
     oleComm.CommandText = \"select * from [\" + fileName + \"#csv]\";
     OdbcDataAdapter adapter = new OdbcDataAdapter(oleComm);
     DataSet ds = new DataSet();
     adapter.Fill(ds, fileName);
     return ds.Tables[0];
     odbcConn.Close();
    }
    if (File.Exists(path))
    {
     File.Delete(path);
    }
   }
   catch (Exception ex)
   {
    if (File.Exists(path))
    {
     File.Delete(path);
    }
    throw ex;
   }
  }
  #endregion
  #region 返回写入CSV的字符串
  /// <summary>
  /// 返回写入CSV的字符串
  /// </summary>
  /// <returns></returns>
  private string ExportCSV()
  { 
   if(_dataSource==null)
    throw new ArgumentNullException(\"dataSource\");
   StringBuilder strbData = new StringBuilder();
   if (_titles == null)
   {
    //添加列名
    foreach (DataColumn column in _dataSource.Columns)
    {
     strbData.Append(column.ColumnName + \",\");
    }
    strbData.Append(\"\\n\");
    foreach (DataRow dr in _dataSource.Rows)
    {
     for (int i = 0; i < _dataSource.Columns.Count; i++)
     {
      strbData.Append(dr[i].ToString() + \",\");
     }
     strbData.Append(\"\\n\");
    }
    return strbData.ToString();
   }
   else
   {
    foreach (string columnName in _titles)
    {
     strbData.Append(columnName + \",\");
    }
    strbData.Append(\"\\n\");
    if (_fields == null)
    {
     foreach (DataRow dr in _dataSource.Rows)
     {
      for (int i = 0; i < _dataSource.Columns.Count; i++)
      {
       strbData.Append(dr[i].ToString() + \",\");
      }
      strbData.Append(\"\\n\");
     }
     return strbData.ToString();
    }
    else
    {
     foreach (DataRow dr in _dataSource.Rows)
     {
      for (int i = 0; i < _fields.Length; i++)
      {
       strbData.Append(_fields[i].ToString() + \",\");
      }
      strbData.Append(\"\\n\");
     }
     return strbData.ToString();
    }
   }
  }
  #endregion
  #region 得到一个随意的文件名
  /// <summary> 
  /// 得到一个随意的文件名 
  /// </summary> 
  /// <returns></returns> 
  private string GetRandomFileName()
  {
   Random rnd = new Random((int)(DateTime.Now.Ticks));
   string s = rnd.Next(Int32.MaxValue).ToString();
   return DateTime.Now.ToShortDateString() + \"_\" + s + \".csv\";
  }
  #endregion
 }
}

希望本文所述对大家的C#程序设计有所帮助。

本文地址:https://www.stayed.cn/item/21901

转载请注明出处。

本站部分内容来源于网络,如侵犯到您的权益,请 联系我

我的博客

人生若只如初见,何事秋风悲画扇。