void RoundTripSingleBool(int numValues)
{
var pocos = new List <SingleBoolPoco>();
var random = new Random(123);
for (int i = 0; i < numValues; i++)
{
pocos.Add(new SingleBoolPoco {
Bool = random.Next() % 2 == 0
});
}
var stream = new MemoryStream();
Footer footer;
StripeStreamHelper.Write(stream, pocos, out footer);
var stripeStreams = StripeStreamHelper.GetStripeStreams(stream, footer);
var boolReader = new BooleanReader(stripeStreams, 1);
var results = boolReader.Read().ToArray();
for (int i = 0; i < numValues; i++)
{
Assert.Equal(pocos[i].Bool, results[i]);
}
}
public void Test_BooleanReader()
{
const string raw = "true false not_boolean";
var stream = new MemoryStream(Encoding.ASCII.GetBytes(raw));
var reader = new BooleanReader(stream, null);
var obj = reader.Read();
Assert.IsNotNull(obj);
Assert.AreEqual(true, obj.Value);
obj = reader.Read();
Assert.IsNotNull(obj);
Assert.AreEqual(false, obj.Value);
obj = reader.Read();
Assert.IsNull(obj);
stream.Dispose();
}
static void Main(string[] args)
{
if (args.Length != 1)
{
Console.WriteLine("Usage: <FILENAME>");
return;
}
var filename = args[0];
var stream = new FileStream(filename, FileMode.Open, FileAccess.Read);
var fileTail = new FileTail(stream);
foreach (var stripe in fileTail.Stripes)
{
Console.WriteLine($"Reading stripe with {stripe.NumRows} rows");
var stripeStreamCollection = stripe.GetStripeStreamCollection();
if (fileTail.Footer.Types[0].Kind != ColumnTypeKind.Struct)
{
throw new InvalidDataException($"The base type must be {nameof(ColumnTypeKind.Struct)}");
}
var names = fileTail.Footer.Types[0].FieldNames;
for (int columnId = 1; columnId < fileTail.Footer.Types.Count; columnId++)
{
var columnType = fileTail.Footer.Types[columnId];
var columnName = names[columnId - 1];
switch (columnType.Kind)
{
case ColumnTypeKind.Long:
case ColumnTypeKind.Int:
case ColumnTypeKind.Short:
{
Console.WriteLine($"Reading longs from column {columnId} ({columnName})");
var reader = new LongReader(stripeStreamCollection, (uint)columnId);
var count = reader.Read().Count();
Console.WriteLine($"Done reading {count} longs");
break;
}
case ColumnTypeKind.Byte:
{
Console.WriteLine($"Reading bytes from column {columnId} ({columnName})");
var reader = new ByteReader(stripeStreamCollection, (uint)columnId);
var count = reader.Read().Count();
Console.WriteLine($"Done reading {count} bytes");
break;
}
case ColumnTypeKind.Boolean:
{
Console.WriteLine($"Reading bools from column {columnId} ({columnName})");
var reader = new BooleanReader(stripeStreamCollection, (uint)columnId);
var count = reader.Read().Count();
Console.WriteLine($"Done reading {count} bools");
break;
}
case ColumnTypeKind.Float:
{
Console.WriteLine($"Reading floats from column {columnId} ({columnName})");
var reader = new FloatReader(stripeStreamCollection, (uint)columnId);
var count = reader.Read().Count();
Console.WriteLine($"Done reading {count} floats");
break;
}
case ColumnTypeKind.Double:
{
Console.WriteLine($"Reading doubles from column {columnId} ({columnName})");
var reader = new DoubleReader(stripeStreamCollection, (uint)columnId);
var count = reader.Read().Count();
Console.WriteLine($"Done reading {count} doubles");
break;
}
case ColumnTypeKind.Binary:
{
Console.WriteLine($"Reading binary from column {columnId} ({columnName})");
var reader = new ColumnTypes.BinaryReader(stripeStreamCollection, (uint)columnId);
var count = reader.Read().Count();
Console.WriteLine($"Done reading {count} binaries");
break;
}
case ColumnTypeKind.Decimal:
{
Console.WriteLine($"Reading decimals from column {columnId} ({columnName})");
var reader = new DecimalReader(stripeStreamCollection, (uint)columnId);
var count = reader.Read().Count();
Console.WriteLine($"Done reading {count} decimals");
break;
}
case ColumnTypeKind.Timestamp:
{
Console.WriteLine($"Reading timestamps from column {columnId} ({columnName})");
var reader = new TimestampReader(stripeStreamCollection, (uint)columnId);
var count = reader.Read().Count();
Console.WriteLine($"Done reading {count} timestamps");
break;
}
case ColumnTypeKind.Date:
{
Console.WriteLine($"Reading dates from column {columnId} ({columnName})");
var reader = new DateReader(stripeStreamCollection, (uint)columnId);
var count = reader.Read().Count();
Console.WriteLine($"Done reading {count} dates");
break;
}
case ColumnTypeKind.String:
{
Console.WriteLine($"Reading strings from column {columnId} ({columnName})");
var reader = new ColumnTypes.StringReader(stripeStreamCollection, (uint)columnId);
var count = reader.Read().Count();
Console.WriteLine($"Done reading {count} strings");
break;
}
default:
throw new NotImplementedException();
}
}
Console.WriteLine("Done reading stripe");
}
}