public void SqlRecordFillTest()
{
SqlMetaData[] metaData = new SqlMetaData[]
{
new SqlMetaData("col1", SqlDbType.Bit),
new SqlMetaData("col2", SqlDbType.TinyInt),
new SqlMetaData("col3", SqlDbType.VarBinary, 1000),
new SqlMetaData("col4", SqlDbType.NVarChar, 1000),
new SqlMetaData("col5", SqlDbType.DateTime),
new SqlMetaData("col6", SqlDbType.Float),
new SqlMetaData("col7", SqlDbType.UniqueIdentifier),
new SqlMetaData("col8", SqlDbType.SmallInt),
new SqlMetaData("col9", SqlDbType.Int),
new SqlMetaData("col10", SqlDbType.BigInt),
new SqlMetaData("col11", SqlDbType.Real),
new SqlMetaData("col12", SqlDbType.Decimal),
new SqlMetaData("col13", SqlDbType.Money),
new SqlMetaData("col14", SqlDbType.Variant)
};
SqlDataRecord record = new SqlDataRecord(metaData);
for (int i = 0; i < record.FieldCount; i++)
{
Assert.Equal($"col{i + 1}", record.GetName(i));
}
record.SetBoolean(0, true);
Assert.Equal(true, record.GetBoolean(0));
record.SetByte(1, 1);
Assert.Equal(1, record.GetByte(1));
byte[] bb1 = new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
byte[] bb2 = new byte[5];
record.SetSqlBinary(2, new SqlBinary(new byte[0]));
record.SetBytes(2, 0, bb1, 0, 3);
record.SetBytes(2, 2, bb1, 6, 3);
// Verify the length of the byte array
Assert.Equal(5, record.GetBytes(2, 0, bb2, 0, 5));
Assert.Equal(5, record.GetBytes(2, 0, null, 0, 0));
byte[] expected = new byte[] { 1, 2, 7, 8, 9 };
Assert.Equal <byte>(expected, bb2);
char[] cb1 = new char[] { 'a', 'b', 'c', 'd', 'e', 'f', 'g' };
char[] cb2 = new char[5];
record.SetChars(3, 0, cb1, 0, 3);
record.SetChars(3, 2, cb1, 4, 3);
char[] expectedValue = new char[] { 'a', 'b', 'e', 'f', 'g' };
Assert.Equal(expectedValue.Length, record.GetChars(3, 0, cb2, 0, 5));
Assert.Equal <char>(expectedValue, new string(cb2, 0, (int)record.GetChars(3, 0, null, 0, 0)));
record.SetString(3, "");
string xyz = "xyz";
record.SetString(3, "xyz");
Assert.Equal(xyz, record.GetString(3));
Assert.Equal(xyz.Length, record.GetChars(3, 0, cb2, 0, 5));
Assert.Equal(xyz, new string(cb2, 0, (int)record.GetChars(3, 0, null, 0, 0)));
record.SetChars(3, 2, cb1, 4, 3);
Assert.Equal(5, record.GetChars(3, 0, cb2, 0, 5));
string interleavedResult = "xyefg";
Assert.Equal(interleavedResult, new string(cb2, 0, (int)record.GetChars(3, 0, null, 0, 0)));
Assert.Equal(interleavedResult, record.GetString(3));
record.SetSqlDateTime(4, SqlDateTime.MaxValue);
Assert.Equal(SqlDateTime.MaxValue, record.GetSqlDateTime(4));
record.SetSqlDouble(5, SqlDouble.MaxValue);
Assert.Equal(SqlDouble.MaxValue, record.GetSqlDouble(5));
SqlGuid guid = new SqlGuid("F9168C5E-CEB2-4faa-B6BF-329BF39FA1E4");
record.SetSqlGuid(6, guid);
Assert.Equal(guid, record.GetSqlGuid(6));
record.SetSqlInt16(7, SqlInt16.MaxValue);
Assert.Equal(SqlInt16.MaxValue, record.GetSqlInt16(7));
record.SetSqlInt32(8, SqlInt32.MaxValue);
Assert.Equal(SqlInt32.MaxValue, record.GetSqlInt32(8));
record.SetSqlInt64(9, SqlInt64.MaxValue);
Assert.Equal(SqlInt64.MaxValue, record.GetSqlInt64(9));
record.SetSqlSingle(10, SqlSingle.MinValue);
Assert.Equal(SqlSingle.MinValue, record.GetSqlSingle(10));
record.SetSqlDecimal(11, SqlDecimal.Null);
record.SetSqlDecimal(11, SqlDecimal.MaxValue);
Assert.Equal(SqlDecimal.MaxValue, record.GetSqlDecimal(11));
record.SetSqlMoney(12, SqlMoney.MaxValue);
Assert.Equal(SqlMoney.MaxValue, record.GetSqlMoney(12));
// Try adding different values to SqlVariant type
for (int i = 0; i < record.FieldCount - 1; ++i)
{
object valueToSet = record.GetSqlValue(i);
record.SetValue(record.FieldCount - 1, valueToSet);
object o = record.GetSqlValue(record.FieldCount - 1);
if (o is SqlBinary)
{
Assert.Equal <byte>(((SqlBinary)valueToSet).Value, ((SqlBinary)o).Value);
}
else
{
Assert.Equal(valueToSet, o);
}
record.SetDBNull(record.FieldCount - 1);
Assert.Equal(DBNull.Value, record.GetSqlValue(record.FieldCount - 1));
record.SetDBNull(i);
Assert.Equal(DBNull.Value, record.GetValue(i));
}
}
/// <summary>
/// Stream data from source into IEnumerable to be applied to SQL datatable
/// </summary>
/// <remarks>
/// Note that this function will execute source reads synchronously (during actual stored procedure execution)
/// </remarks>
private IEnumerable <SqlDataRecord> StreamDataTable(StreamReader rawReader = null, TextFieldParser csvReader = null)
{
// Set up metadata object matching user-defined type in SQL (contains row identifier column and RowData varchar(max) column);
// this could be made configurable, but would require some complex logic below to properly map columns:
var udt_ImportTable_Schema = new SqlMetaData[] {
new SqlMetaData("RowID", SqlDbType.Int),
new SqlMetaData("RowData", SqlDbType.VarChar, SqlMetaData.Max)
};
// Create record to populate (SQL server will receive each row when "yield return" is called, meaning
// we can just update this inputRecord with new data for each input row):
var inputRecord = new SqlDataRecord(udt_ImportTable_Schema);
int rowCount = 1;
// If specialized CSV reader provided, use TextFieldParser to read all available data and pass to SQL in custom-delimited
// string (note this could be made configurable, using a custom input table type with discrete columns rather than a single
// delimited string data column):
if (csvReader != null)
{
var csvFieldString = new StringBuilder();
while (!csvReader.EndOfData)
{
inputRecord.SetInt32(0, rowCount++);
// Read separated columns from input source, and construct string using custom delimiter "|::|". Note that
// SQL will need to be prepared to re-tokenize string using this delimiter, and de-escape "|::" with "|:"
// only AFTER doing so (also note that any fields absent in source data will be absent in input row - i.e.
// number of column/fields could be inconsistent between rows, meaning that SQL must be prepared for
// "ragged right" type data and treat missing columns accordingly):
var csvFields = csvReader.ReadFields();
for (int i = 0; i < csvFields.Length; ++i)
{
csvFieldString.Append(csvFields[i].Replace("|:", "|::"));
csvFieldString.Append("|::|");
}
inputRecord.SetString(1, csvFieldString.ToString());
yield return(inputRecord);
csvFieldString.Clear();
}
}
// Otherwise if raw StreamReader provided, just read all lines as-is:
else if (rawReader != null)
{
while (!rawReader.EndOfStream)
{
inputRecord.SetInt32(0, rowCount++);
inputRecord.SetString(1, rawReader.ReadLine());
// If line filter regex configured, only actually return this line if it matches (note: with regex specified,
// it is possible that no lines will match and IEnumerable will be empty - this will cause an exception to be
// thrown at execution time, requiring a null collection rather than an empty one):
if (importLineFilterRegex == null ? true : importLineFilterRegex.IsMatch(inputRecord.GetString(1)))
{
yield return(inputRecord);
}
}
}
else
{
throw new ArgumentException("No reader object supplied to stream to data table");
}
}
