private void WriteHeaderIfRequired()
{
if (this.isHeaderWritten)
{
return;
}
ProtoWriter.WriteFieldHeader(1, WireType.StartGroup, this.writer);
this.context.StartSubItem(this.resultIndex);
this.context.Columns = ProtoDataColumnFactory.GetColumns(this.reader, this.options);
ColumnsWriter.WriteColumns(this.context);
this.isHeaderWritten = true;
}
private void WriteHeaderIfRequired()
{
if (this.isHeaderWritten)
{
return;
}
ProtoWriter.WriteFieldHeader(1, WireType.StartGroup, this.writer);
this.currentResultToken = ProtoWriter.StartSubItem(this.resultIndex, this.writer);
IList <ProtoDataColumn> columns = ProtoDataColumnFactory.GetColumns(this.reader, this.options);
new HeaderWriter(this.writer).WriteHeader(columns);
this.rowWriter = new RowWriter(this.writer, columns, this.options);
this.isHeaderWritten = true;
}
/// <summary>
/// Serialize an <see cref="System.Data.IDataReader"/> to a binary stream using protocol-buffers.
/// </summary>
/// <param name="stream">The <see cref="System.IO.Stream"/> to write to.</param>
/// <param name="reader">The <see cref="System.Data.IDataReader"/>who's contents to serialize.</param>
/// <param name="options"><see cref="ProtoDataWriterOptions"/> specifying any custom serialization options.</param>
public void Serialize(Stream stream, IDataReader reader, ProtoDataWriterOptions options)
{
Throw.IfNull(stream, nameof(stream));
Throw.IfNull(reader, nameof(reader));
options = options ?? new ProtoDataWriterOptions();
var resultIndex = 0;
using (var writer = new ProtoWriter(stream, null, null))
{
var context = new ProtoWriterContext(writer, options);
do
{
ProtoWriter.WriteFieldHeader(1, WireType.StartGroup, writer);
context.StartSubItem(resultIndex);
context.Columns = ProtoDataColumnFactory.GetColumns(reader, options);
ColumnsWriter.WriteColumns(context);
var recordIndex = 0;
while (reader.Read())
{
RecordWriter.WriteRecord(context, recordIndex, reader);
recordIndex++;
}
context.EndSubItem();
resultIndex++;
}while (reader.NextResult());
}
}
/// <summary>
/// Initializes a new instance of the <see cref="ProtoDataStream"/> class.
/// </summary>
/// <param name="reader">The <see cref="IDataReader"/>who's contents to serialize.</param>
/// <param name="options"><see cref="ProtoDataWriterOptions"/> specifying any custom serialization options.</param>
/// <param name="bufferSize">Buffer size to use when serializing rows.
/// You should not need to change this unless you have exceptionally
/// large rows or an exceptionally high number of columns.</param>
public ProtoDataStream(
IDataReader reader,
ProtoDataWriterOptions options,
int bufferSize = DefaultBufferSize)
{
if (reader == null)
{
throw new ArgumentNullException("reader");
}
if (options == null)
{
throw new ArgumentNullException("options");
}
this.reader = reader;
this.options = options;
resultIndex = 0;
columnFactory = new ProtoDataColumnFactory();
bufferStream = new CircularStream(bufferSize);
writer = new ProtoWriter(bufferStream, null, null);
}
/// <summary>
/// Serialize an <see cref="System.Data.IDataReader"/> to a binary stream using protocol-buffers.
/// </summary>
/// <param name="stream">The <see cref="System.IO.Stream"/> to write to.</param>
/// <param name="reader">The <see cref="System.Data.IDataReader"/>who's contents to serialize.</param>
/// <param name="options"><see cref="ProtoDataWriterOptions"/> specifying any custom serialization options.</param>
public void Serialize(Stream stream, IDataReader reader, ProtoDataWriterOptions options)
{
if (stream == null)
{
throw new ArgumentNullException("stream");
}
if (reader == null)
{
throw new ArgumentNullException("reader");
}
// Null options are permitted to be passed in.
options = options ?? new ProtoDataWriterOptions();
// For a (minor) performance improvement, Serialize() has been left
// as a single long method with functions manually inlined.
var resultIndex = 0;
using (var writer = new ProtoWriter(stream, null, null))
{
do
{
// This is the underlying protocol buffers structure we use:
//
// <1 StartGroup> each DataTable
// <SubItem>
// <2 StartGroup> each DataColumn
// <SubItem>
// <1 String> Column Name
// <2 Variant> Column ProtoDataType (enum casted to int)
// </SubItem>
// <3 StartGroup> each DataRow
// <SubItem>
// <(# Column Index) (corresponding type)> Field Value
// </SubItem>
// </SubItem>
//
// NB if Field Value is a DataTable, the whole DataTable is
// write the table
ProtoWriter.WriteFieldHeader(1, WireType.StartGroup, writer);
SubItemToken resultToken = ProtoWriter.StartSubItem(resultIndex, writer);
var columns = new ProtoDataColumnFactory().GetColumns(reader, options);
new HeaderWriter(writer).WriteHeader(columns);
var rowWriter = new RowWriter(writer, columns, options);
// write the rows
while (reader.Read())
{
rowWriter.WriteRow(reader);
}
ProtoWriter.EndSubItem(resultToken, writer);
resultIndex++;
}while (reader.NextResult());
}
}
