从DataSet导出Txt(续)
在从DataSet导出Txt一篇中，我们还有一个问题没有解决：每个Txt的总金额不能超过10万，单笔金额不能超过5万（上一篇中为5000）。
按照上一篇中我提到的解决方案，首先在服务器中生成Txt，再打包下载到客户端。
首先，我们需要两个方法：一个是创建文件的方法，一个是压缩文件的方法，下面我给出具体的代码。
创建Txt文件的方法：
/// <summary> /// 创建Txt文档 /// </summary> /// <param name="str">写入Txt的内容</param> /// <param name="list">生成的Txt的路径列表</param> private static void CreateTxtFile(string str, List<string> list) { if (str.Length == 0) { return; } string fileName = Guid.NewGuid() + ".txt"; string path = @"F:\\Txt\\" + fileName; list.Add(path); StreamWriter sr = File.CreateText(path); sr.Write(str); sr.Close(); }
压缩文件的方法：
需要引用ICSharpCode.SharpZipLib.dll
public static void CompressFile(string savePath, string txtName, List<string> fileNames) { ZipOutputStream stream = new ZipOutputStream(File.Create(savePath)); //设置压缩级别(0-9)，值越大压缩越厉害 stream.SetLevel(8); //遍历所有要压缩的文件 for (int i = 0; i < fileNames.Count; i++) { if (File.Exists(fileNames[i])) { FileStream fs = File.OpenRead(fileNames[i]); byte[] buffer = new byte[fs.Length];//设置缓冲区大小 int size = fs.Read(buffer, 0, buffer.Length); //读入缓冲区中的字节数 //string entryName = fileName.Substring(fileName.LastIndexOf("\\") + 1); string entryName = txtName + "(" + (i + 1) + ").txt"; ZipEntry entry = new ZipEntry(entryName); stream.PutNextEntry(entry); stream.Write(buffer, 0, size); try { //如果读入缓冲区中的字节数没有所请求的总字节数那么多 while (size < fs.Length) { int sizeRead = fs.Read(buffer, 0, buffer.Length); stream.Write(buffer, 0, sizeRead); size += sizeRead; } } catch (Exception ex) { throw ex; } finally { fs.Close(); } } //删除文件 File.Delete(fileNames[i]); } stream.Finish(); stream.Close(); }
下面来写导出Txt的方法
在此之前，先定义一个Salary类
public class Salary { /// <summary> /// 员工编号 /// </summary> public string EmpNo { get; set; } /// <summary> /// 员工姓名 /// </summary> public string EmpName { get; set; } /// <summary> /// 部门 /// </summary> public string Dept { get; set; } /// <summary> /// 职务 /// </summary> public string Post { get; set; } /// <summary> /// 工资 /// </summary> public decimal Salary { get; set; } /// <summary> /// 开户行 /// </summary> public string Bank { get; set; } /// <summary> /// 银行账号 /// </summary> public string BankNo { get; set; } }
再写一个生成单个Txt文档字符串的方法
public static string GenTxtString(List<Salary> list) { StringBuilder sb = new StringBuilder(); for (int i = 0; i < list.Count; i++) { sb.Append(i + 1).Append(","); sb.Append(list[i].EmpNo).Append(","); sb.Append(list[i].EmpName).Append(","); sb.Append(list[i].Dept).Append(","); sb.Append(list[i].Post).Append(","); sb.Append(list[i].Salary).Append(","); sb.Append(list[i].Bank).Append(","); sb.Append(list[i].BankNo).Append(","); sb.AppendLine(); } return sb.ToString(); }
导出Txt的方法
public static void ToTxt(DataSet ds, string fileName) { DataTable table = ds.Tables[0]; DataRowCollection rows = table.Rows; DataColumnCollection columns = table.Columns; //行数 int rowCount = rows.Count; //列数 int columnCount = columns.Count; //金额总和 decimal totalSalary = 0; //Txt文档路径列表 List<string> fileList = new List<string>(); //Salary列表 List<Salary> salaryList = new List<Salary>(); //遍历所有行 for (int i = 0; i < rows.Count; i++) { //工资 decimal salary = Convert.ToDecimal(rows[i]["Salary"]); //固定值5万 decimal c = 50000.00m; //工资由几个5万组成 int a = (int)(salary / c); //工资小于等于10万 if (salary <= 100000) { totalSalary += salary; if (totalSalary > 100000) { //如果总额大于10万，新创建一个Txt CreateTxtFile(GenTxtString(salaryList), fileList); totalSalary = salary; salaryList.Clear(); } if (a > 0) { for (int j = 0; j < a; j++) { salaryList.Add(new Salary() { EmpNo = rows[i]["EmpNo"].ToString(), EmpName = rows[i]["EmpName"].ToString(), Dept = rows[i]["Dept"].ToString(), Post = rows[i]["Post"].ToString(), FactSalary = c, Bank = rows[i]["Bank"].ToString(), BankNo = rows[i]["BankNo"].ToString() }); } //余数不为0 if (salary % c != 0) { salaryList.Add(new Salary() { EmpNo = rows[i]["EmpNo"].ToString(), EmpName = rows[i]["EmpName"].ToString(), Dept = rows[i]["Dept"].ToString(), Post = rows[i]["Post"].ToString(), FactSalary = salary % c, Bank = rows[i]["Bank"].ToString(), BankNo = rows[i]["BankNo"].ToString() }); } } else { salaryList.Add(new Salary() { EmpNo = rows[i]["EmpNo"].ToString(), EmpName = rows[i]["EmpName"].ToString(), Dept = rows[i]["Dept"].ToString(), Post = rows[i]["Post"].ToString(), FactSalary = salary, Bank = rows[i]["Bank"].ToString(), BankNo = rows[i]["BankNo"].ToString() }); } } //工资大于10万 else { for (int j = 0; j < a; j++) { totalSalary += c; if (totalSalary > 100000) { CreateTxtFile(GenTxtString(salaryList), fileList); totalSalary = c; salaryList.Clear(); } salaryList.Add(new Salary() { EmpNo = rows[i]["EmpNo"].ToString(), EmpName = rows[i]["EmpName"].ToString(), Dept = rows[i]["Dept"].ToString(), Post = rows[i]["Post"].ToString(), FactSalary = c, Bank = rows[i]["Bank"].ToString(), BankNo = rows[i]["BankNo"].ToString() }); } if (salary % c != 0) { totalSalary += salary % c; if (totalSalary > 100000) { CreateTxtFile(GenTxtString(salaryList), fileList); totalSalary = salary % c; salaryList.Clear(); } salaryList.Add(new Salary() { EmpNo = rows[i]["EmpNo"].ToString(), EmpName = rows[i]["EmpName"].ToString(), Dept = rows[i]["Dept"].ToString(), Post = rows[i]["Post"].ToString(), FactSalary = salary % c, Bank = rows[i]["Bank"].ToString(), BankNo = rows[i]["BankNo"].ToString() }); } } } if (totalSalary <= 100000) { CreateTxtFile(GenTxtString(salaryList), fileList); } CompressFile(@"F:\Txt\" + fileName + ".zip", fileName, fileList); }
用以下数据构造DataSet。
| EmpNo | EmpName | Dept | Post | Salary | Bank | BankNo |
| 1000 | 刘德华 | 办公室 | 总经理 | 110000 | 交通银行 | 55784756584762 |
| 1001 | 周星驰 | 办公室 | 副总 | 105000 | 建设银行 | 58476594865764 |
| 1002 | 王力宏 | 财务 | 经理 | 90000 | 农业银行 | 95486437546534 |
| 1003 | 谢霆锋 | 财务 | 出纳 | 6000 | 工商银行 | 59583457547336 |
| 1004 | 吴彦祖 | 软件部 | 程序员 | 10300 | 建设银行 | 59445473365484 |
| 1005 | 林志颖 | 软件部 | 工程师 | 24000 | 中国银行 | 95584744956852 |
| 1006 | 何润东 | 软件部 | 工程师 | 30000 | 农业银行 | 95483574775483 |
| 1007 | 古天乐 | 软件部 | 实习生 | 3000 | 农业银行 | 95483474454499 |
| 1008 | 郑伊健 | 软件部 | 实习生 | 5000 | 农业银行 | 43945748374934 |
以上数据是我随便填的，可以使用任意数据替换。
在Main方法中调用ToTxt方法。
Txt.ToTxt(ds, "导出Txt测试"); Console.WriteLine("导出成功");
运行后，在F:\Txt下会生成压缩文件
双击“导出Txt测试.zip”，会看到生成的Txt文件，如下图所示：
这样生成的每个Txt里都不会超过10万。有兴趣的朋友可以帮我测试测试，看有没有什么地方不对！
下面我附上源码：下载
运行程序之前先在F盘下新建Txt文件夹，也可以修改程序里的Path。
