基于NPOI的Excel导入和导出功能

 
复制代码
  1 using NPOI.HSSF.UserModel;
  2 using NPOI.SS.UserModel;
  3 using System;
  4 using System.Collections.Generic;
  5 using System.IO;
  6 using System.Linq;
  7 using System.Reflection;
  8 using System.ComponentModel;
  9 using System.Collections;
 10 
 11 
 12 namespace ExcelHelper.Operating
 13 {
 14     public abstract class WorkbookBuilder
 15     {
 16         protected WorkbookBuilder()
 17         {
 18             currentWorkbook = CreateWorkbook();
 19 
 20             buildContext = new BuildContext() { WorkbookBuilder = this, Workbook = currentWorkbook };
 21         }
 22 
 23         public delegate void BuildEventHandler(BuildContext context);
 24 
 25         protected abstract IWorkbook CreateWorkbook();
 26 
 27         public IWorkbook currentWorkbook;
 28 
 29         private ICellStyle _centerStyle;
 30 
 31         public ICellStyle CenterStyle
 32         {
 33             get
 34             {
 35                 if (_centerStyle == null)
 36                 {
 37                     _centerStyle = currentWorkbook.CreateCellStyle();
 38 
 39                     _centerStyle.Alignment = HorizontalAlignment.Center;
 40 
 41                     _centerStyle.VerticalAlignment = VerticalAlignment.Center;
 42                 }
 43 
 44                 return _centerStyle;
 45             }
 46         }
 47 
 48         private Int32 StartRow = 0;//起始行
 49 
 50 
 51         private BuildContext buildContext;
 52  
 53         public event BuildEventHandler OnHeadCellSetAfter;
 54  
 55         public event BuildEventHandler OnContentCellSetAfter;
 56 
 57 
 58         #region DataTableToExcel
 59 
 60         public void Insert(ISheetDetail sheetDetail)
 61         {
 62             ISheet sheet;
 63 
 64             if (sheetDetail.IsContinue)
 65             {
 66                 sheet = currentWorkbook.GetSheetAt(currentWorkbook.NumberOfSheets - 1);
 67 
 68                 StartRow = sheet.LastRowNum + 1;
 69             }
 70             else
 71             {
 72                 sheet = currentWorkbook.CreateSheet(sheetDetail.SheetName);
 73             }
 74 
 75             buildContext.Sheet = sheet;
 76 
 77             sheet = DataToSheet(sheetDetail.SheetDetailDataWrappers, sheet);
 78 
 79         }
 80         /// <summary>
 81         /// 这里添加数据，循环添加，主要应对由多个组成的
 82         /// </summary>
 83         /// <param name="sheetDetailDataWrappers"></param>
 84         /// <param name="sheet"></param>
 85         /// <returns></returns>
 86         private ISheet DataToSheet(SheetDataCollection sheetDetailDataWrappers, ISheet sheet)
 87         {
 88             foreach (var sheetDetailDataWrapper in sheetDetailDataWrappers)
 89             {
 90                 if (sheetDetailDataWrapper.Datas == null || sheetDetailDataWrapper.Datas.Count() == 0)
 91                 {
 92                     continue;
 93                 }
 94 
 95                 Type type = sheetDetailDataWrapper.Datas.GetType().GetGenericArguments()[0];
 96 
 97                 if (sheetDetailDataWrapper.HaveTitle)
 98                 {
 99                     sheet = SetTitle(sheet, sheetDetailDataWrapper, type);
100                 }
101 
102                 sheet = AddValue(sheet, sheetDetailDataWrapper, type);
103 
104                 StartRow = StartRow + sheetDetailDataWrapper.EmptyIntervalRow;
105             }
106 
107             return sheet;
108         }
109 
110         #endregion
111 
112         #region 设置值
113 
114         private void SetCellValue(ICell cell, object obj)
115         {
116             if (obj == null)
117             {
118                 cell.SetCellValue(" "); return;
119             }
120   
121             if (obj is String)
122             {
123                 cell.SetCellValue(obj.ToString()); return;
124             }
125 
126             if (obj is Int32 || obj is Double)
127             {
128                 cell.SetCellValue(Math.Round(Double.Parse(obj.ToString()), 2)); return;
129             }
130 
131             if (obj.GetType().IsEnum)
132             {
133                 cell.SetCellValue(EnumService.GetDescription((Enum)obj)); return;
134             }
135 
136             if (obj is DateTime)
137             {
138                 cell.SetCellValue(((DateTime)obj).ToString("yyyy-MM-dd HH:mm:ss")); return;
139             }
140 
141             if (obj is Boolean)
142             {
143                 cell.SetCellValue((Boolean)obj ? "√" : "×"); return;
144             }     
145         }
146 
147         #endregion
148 
149         #region SetTitle
150         private ISheet SetTitle(ISheet sheet, ISheetDataWrapper sheetDetailDataWrapper, Type type)
151         {
152             IRow titleRow = null;
153 
154             ICell titleCell = null;
155 
156             if (!String.IsNullOrEmpty(sheetDetailDataWrapper.DataName))
157             {
158                 titleRow = sheet.CreateRow(StartRow);
159 
160                 buildContext.Row = titleRow;
161  
162                 StartRow++;
163 
164                 titleCell = SetCell(titleRow, 0, sheetDetailDataWrapper.DataName);
165 
166                 if (OnHeadCellSetAfter != null)
167                 {
168                     OnHeadCellSetAfter(buildContext);
169                 }
170             }
171 
172             IRow row = sheet.CreateRow(StartRow);
173 
174             buildContext.Row = row;
175 
176             IList<PropertyInfo> checkPropertyInfos = ExcelModelsPropertyManage.CreatePropertyInfos(type);
177 
178             int i = 0;
179 
180             foreach (PropertyInfo property in checkPropertyInfos)
181             {
182                 DisplayNameAttribute dn = property.GetCustomAttributes(typeof(DisplayNameAttribute), false).SingleOrDefault() as DisplayNameAttribute;
183 
184                 if (dn != null)
185                 {
186                     SetCell(row, i++, dn.DisplayName);
187                     continue;
188                 }
189 
190                 Type t = property.PropertyType;
191 
192                 if (t.IsGenericType)
193                 {
194                     if (sheetDetailDataWrapper.Titles == null || sheetDetailDataWrapper.Titles.Count() == 0)
195                     {
196                         continue;
197                     }
198 
199                     foreach (var item in sheetDetailDataWrapper.Titles)
200                     {
201                         SetCell(row, i++, item.TypeName);
202                     }
203                 }
204             }
205         
206             if (titleCell != null && i > 0)
207             {
208                 titleCell.MergeTo(titleRow.CreateCell(i - 1));
209 
210                 titleCell.CellStyle = this.CenterStyle;
211             }
212 
213             StartRow++;
214 
215             return sheet;
216         }
217         #endregion
218 
219         #region AddValue
220         private ISheet AddValue(ISheet sheet, ISheetDataWrapper sheetDetailDataWrapper, Type type)
221         {
222             IList<PropertyInfo> checkPropertyInfos = ExcelModelsPropertyManage.CreatePropertyInfos(type);
223 
224             Int32 cellCount = 0;
225 
226             foreach (var item in sheetDetailDataWrapper.Datas)
227             {
228                 if (item == null)
229                 {
230                     StartRow++;
231                     continue;
232                 }
233 
234                 IRow newRow = sheet.CreateRow(StartRow);
235 
236                 buildContext.Row = newRow;
237 
238                 foreach (PropertyInfo property in checkPropertyInfos)
239                 {
240                     Object obj = property.GetValue(item, null);
241 
242                     Type t = property.PropertyType;
243 
244                     if (t.IsGenericType && t.GetGenericTypeDefinition() == typeof(IEnumerable<>))
245                     {
246                         var ssd = ((IEnumerable)obj).Cast<IExtendedBase>();
247 
248                         if (ssd == null)
249                         {
250                             continue;
251                         }
252 
253                         foreach (var v in sheetDetailDataWrapper.Titles)
254                         {
255                             IExtendedBase sv = ssd.Where(s => s.TypeId == v.TypeId).SingleOrDefault();
256 
257                             SetCell(newRow, cellCount++, sv.TypeValue);
258                         }
259 
260                         continue;
261                     }
262  
263                     SetCell(newRow, cellCount++, obj);
264                 }
265 
266                 StartRow++;
267                 cellCount = 0;
268             }
269 
270             return sheet;
271         }
272 
273         #endregion
274 
275         #region 设置单元格
276         /// <summary>
277         /// 设置单元格
278         /// </summary>
279         /// <param name="row"></param>
280         /// <param name="index"></param>
281         /// <param name="value"></param>
282         /// <returns></returns>
283         private ICell SetCell(IRow row, int index, object value)
284         {
285             ICell cell = row.CreateCell(index);
286 
287             SetCellValue(cell, value);
288 
289             buildContext.Cell = cell;
290 
291             if (OnContentCellSetAfter != null)
292             {
293                 OnContentCellSetAfter(buildContext);
294             }
295 
296             return cell;
297         } 
298         #endregion
299 
300         #region ExcelToDataTable
301 
302         /// <summary>
303         /// 导入
304         /// </summary>
305         /// <typeparam name="T">具体对象</typeparam>
306         /// <param name="fs"></param>
307         /// <param name="fileName"></param>
308         /// <param name="isFirstRowColumn"></param>
309         /// <returns></returns>
310         public static IEnumerable<T> ExcelToDataTable<T>(Stream fs, bool isFirstRowColumn = false) where T : new()
311         {
312             List<T> ts = new List<T>();
313 
314             Type type = typeof(T);
315 
316             IList<PropertyInfo> checkPropertyInfos = ExcelModelsPropertyManage.CreatePropertyInfos(type);
317 
318             try
319             {
320                 IWorkbook workbook = WorkbookFactory.Create(fs);
321 
322                 fs.Dispose();
323 
324                 ISheet sheet = workbook.GetSheetAt(0);
325 
326                 if (sheet != null)
327                 {
328                     IRow firstRow = sheet.GetRow(0);
329 
330                     int cellCount = firstRow.LastCellNum; //一行最后一个cell的编号 即总的列数
331 
332                     Int32 startRow = isFirstRowColumn ? 1 : 0;
333 
334                     int rowCount = sheet.LastRowNum; //行数
335 
336                     int length = checkPropertyInfos.Count;
337 
338                     length = length > cellCount + 1 ? cellCount + 1 : length;
339 
340                     Boolean haveValue = false;
341 
342                     for (int i = startRow; i <= rowCount; ++i)
343                     {
344                         IRow row = sheet.GetRow(i);
345 
346                         if (row == null) continue; //没有数据的行默认是null　　　　　　　
347 
348                         T t = new T();
349 
350                         for (int f = 0; f < length; f++)
351                         {
352                             ICell cell = row.GetCell(f);
353 
354                             if (cell == null || String.IsNullOrEmpty(cell.ToString()))
355                             {
356                                 continue;
357                             }
358 
359                             object b = cell.ToString();
360 
361                             if (cell.CellType == CellType.Numeric)
362                             {
363                                 //NPOI中数字和日期都是NUMERIC类型的，这里对其进行判断是否是日期类型
364                                 if (HSSFDateUtil.IsCellDateFormatted(cell))//日期类型
365                                 {
366                                     b = cell.DateCellValue;
367                                 }
368                                 else
369                                 {
370                                     b = cell.NumericCellValue;
371                                 }
372                             }
373 
374                             PropertyInfo pinfo = checkPropertyInfos[f];
375 
376                             if (pinfo.PropertyType.Name != b.GetType().Name) //类型不一样的时候，强转
377                             {
378                                 b = System.ComponentModel.TypeDescriptor.GetConverter(pinfo.PropertyType).ConvertFrom(b.ToString());
379                             }
380 
381                             type.GetProperty(pinfo.Name).SetValue(t, b, null);
382 
383                             if (!haveValue)
384                             {
385                                 haveValue = true;
386                             }
387                         }
388                         if (haveValue)
389                         {
390                             ts.Add(t); haveValue = false;
391                         }
392                     }
393                 }
394 
395                 return ts;
396             }
397             catch (Exception ex)
398             {
399                 return null;
400             }
401         }
402 
403         #endregion
404     }
405 
406     public class BuildContext
407     {
408         public WorkbookBuilder WorkbookBuilder { get; set; }
409         
410         public IWorkbook Workbook { get; set; }
411 
412         public ISheet Sheet { get; set; }
413 
414         public IRow Row { get; set; }
415 
416         public ICell Cell { get; set; }
417 
418     }
419 }