Sunday, October 31, 2010

A Fast CSV Reader

To Download Sources Go to this Link

Introduction

One would imagine that parsing CSV files is a straightforward and boring task. I was thinking that too, until I had to parse several CSV files of a couple GB each. After trying to use the OLEDB JET driver and various Regular Expressions, I still ran into serious performance problems. At this point, I decided I would try the custom class option. I scoured the net for existing code, but finding a correct, fast, and efficient CSV parser and reader is not so simple, whatever platform/language you fancy.

I say correct in the sense that many implementations merely use some splitting method like String.Split(). This will, obviously, not handle field values with commas. Better implementations may care about escaped quotes, trimming spaces before and after fields, etc., but none I found were doing it all, and more importantly, in a fast and efficient manner.

And, this led to the CSV reader class I present in this article. Its design is based on the System.IO.StreamReader class, and so is a non-cached, forward-only reader (similar to what is sometimes called a fire-hose cursor).

Benchmarking it against both OLEDB and regex methods, it performs about 15 times faster, and yet its memory usage is very low.

To give more down-to-earth numbers, with a 45 MB CSV file containing 145 fields and 50,000 records, the reader was processing about 30 MB/sec. So all in all, it took 1.5 seconds! The machine specs were P4 3.0 GHz, 1024 MB.

Supported Features

This reader supports fields spanning multiple lines. The only restriction is that they must be quoted, otherwise it would not be possible to distinguish between malformed data and multi-line values.

Basic data-binding is possible via the System.Data.IDataReader interface implemented by the reader.

You can specify custom values for these parameters:

  • Default missing field action;
  • Default malformed CSV action;
  • Buffer size;
  • Field headers option;
  • Trimming spaces option;
  • Field delimiter character;
  • Quote character;
  • Escape character (can be the same as the quote character);
  • Commented line character.

If the CSV contains field headers, they can be used to access a specific field.

When the CSV data appears to be malformed, the reader will fail fast and throw a meaningful exception stating where the error occurred and providing the current content of the buffer.

A cache of the field values is kept for the current record only, but if you need dynamic access, I also included a cached version of the reader,CachedCsvReader, which internally stores records as they are read from the stream. Of course, using a cache this way makes the memory requirements way higher, as the full set of data is held in memory.

Latest Updates (3.7 Release)

  • Breaking: Added more field value trimming options.

Benchmark and Profiling

You can find the code for these benchmarks in the demo project. I tried to be fair and follow the same pattern for each parsing method. The regex used comes from Jeffrey Friedl's book, and can be found at page 271. It doesn't handle trimming and multi-line fields.

The test file contains 145 fields, and is about 45 MB (included in the demo project as a RAR archive).

I also included the raw data from the benchmark program and from the CLR Profiler for .NET 2.0.

Using the Code

The class design follows System.IO.StreamReader as much as possible. The parsing mechanism introduced in version 2.0 is a bit trickier because we handle the buffering and the new line parsing ourselves. Nonetheless, because the task logic is clearly encapsulated, the flow is easier to understand. All the code is well documented and structured, but if you have any questions, simply post a comment.

Basic Usage Scenario

Collapse
using System.IO; using LumenWorks.Framework.IO.Csv;  void ReadCsv() {     // open the file "data.csv" which is a CSV file with headers     using (CsvReader csv =            new CsvReader(new StreamReader("data.csv"), true))     {         int fieldCount = csv.FieldCount;         string[] headers = csv.GetFieldHeaders();          while (csv.ReadNextRecord())         {             for (int i = 0; i < class="code-keyword" style="color: blue; ">string.Format("{0} = {1};",                               headers[i], csv[i]));              Console.WriteLine();         }     } }

Simple Data-Binding Scenario (ASP.NET)

Collapse
using System.IO; using LumenWorks.Framework.IO.Csv;  void ReadCsv() {     // open the file "data.csv" which is a CSV file with headers     using (CsvReader csv = new CsvReader(                            new StreamReader("data.csv"), true))     {         myDataRepeater.DataSource = csv;         myDataRepeater.DataBind();     } }

Complex Data-Binding Scenario (ASP.NET)

Due to the way both the System.Web.UI.WebControls.DataGrid and System.Web.UI.WebControls.GridView handleSystem.ComponentModel.ITypedList, complex binding in ASP.NET is not possible. The only way around this limitation would be to wrap each field in a container implementing System.ComponentModel.ICustomTypeDescriptor.

Anyway, even if it was possible, using the simple data-binding method is much more efficient.

For the curious amongst you, the bug comes from the fact that the two grid controls completely ignore the property descriptors returned bySystem.ComponentModel.ITypedList, and relies instead on System.ComponentModel.TypeDescriptor.GetProperties(...), which obviously returns the properties of the string array and not our custom properties. SeeSystem.Web.UI.WebControls.BoundColumn.OnDataBindColumn(...) in a disassembler.

Complex Data-Binding Scenario (Windows Forms)

Collapse
using System.IO; using LumenWorks.Framework.IO.Csv;  void ReadCsv() {     // open the file "data.csv" which is a CSV file with headers     using (CachedCsvReader csv = new            CachedCsvReader(new StreamReader("data.csv"), true))     {         // Field headers will automatically be used as column names         myDataGrid.DataSource = csv;     } }

Custom Error Handling Scenario

Collapse
using System.IO; using LumenWorks.Framework.IO.Csv;  void ReadCsv() {     // open the file "data.csv" which is a CSV file with headers     using (CsvReader csv = new CsvReader(            new StreamReader("data.csv"), true))     {         // missing fields will not throw an exception,         // but will instead be treated as if there was a null value         csv.MissingFieldAction = MissingFieldAction.ReplaceByNull;          // to replace by "" instead, then use the following action:         //csv.MissingFieldAction = MissingFieldAction.ReplaceByEmpty;          int fieldCount = csv.FieldCount;         string[] headers = csv.GetFieldHeaders();          while (csv.ReadNextRecord())         {             for (int i = 0; i < class="code-keyword" style="color: blue; ">string.Format("{0} = {1};",                               headers[i],                               csv[i] == null ? "MISSING" : csv[i]));              Console.WriteLine();         }     } }

Custom Error Handling Using Events Scenario

Collapse
using System.IO; using LumenWorks.Framework.IO.Csv;  void ReadCsv() {     // open the file "data.csv" which is a CSV file with headers     using (CsvReader csv = new CsvReader(            new StreamReader("data.csv"), true))     {         // missing fields will not throw an exception,         // but will instead be treated as if there was a null value         csv.DefaultParseErrorAction = ParseErrorAction.RaiseEvent;         csv.ParseError += new ParseErrorEventHandler(csv_ParseError);          int fieldCount = csv.FieldCount;         string[] headers = csv.GetFieldHeaders();          while (csv.ReadNextRecord())         {             for (int i = 0; i < class="code-keyword" style="color: blue; ">string.Format("{0} = {1};",                               headers[i], csv[i]));              Console.WriteLine();         }     } }  void csv_ParseError(object sender, ParseErrorEventArgs e) {     // if the error is that a field is missing, then skip to next line     if (e.Error is MissingFieldException)     {         Console.Write("--MISSING FIELD ERROR OCCURRED");         e.Action = eErrorAction.AdvanceToNextLine;     } }

No comments:

Post a Comment