90 lines
2.8 KiB
C#
90 lines
2.8 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
|
|
namespace BismNormalizer.TabularCompare
|
|
{
|
|
/// <summary>
|
|
/// Keeps track of tables for processing, and dynamically increments the rows processed in each partition.
|
|
/// </summary>
|
|
public class ProcessingTable : IComparable<ProcessingTable>
|
|
{
|
|
private string _name;
|
|
private string _id;
|
|
private List<PartitionRowCounter> _partitions = new List<PartitionRowCounter>();
|
|
|
|
/// <summary>
|
|
/// Name of the table being processed.
|
|
/// </summary>
|
|
public string Name => _name;
|
|
|
|
/// <summary>
|
|
/// Id of the table being processed. Can be different for tabular models with multidimensional metadata.
|
|
/// </summary>
|
|
public string Id => _id;
|
|
|
|
/// <summary>
|
|
/// Collection of PartitionRowCounter objects.
|
|
/// </summary>
|
|
public List<PartitionRowCounter> Partitions => _partitions;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the PartitionRowCounter class using a name and id.
|
|
/// </summary>
|
|
public ProcessingTable(string name, string id)
|
|
{
|
|
_name = name;
|
|
_id = id;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Get the total row count for all partitions associated with the table.
|
|
/// </summary>
|
|
/// <returns>Total row count.</returns>
|
|
public long GetRowCount()
|
|
{
|
|
long rowCount = 0;
|
|
foreach (PartitionRowCounter partition in _partitions)
|
|
{
|
|
rowCount += partition.RowCount;
|
|
}
|
|
return rowCount;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Find partition based on its Id.
|
|
/// </summary>
|
|
/// <param name="id"></param>
|
|
/// <returns>PartitionRowCounter object. Null if not found.</returns>
|
|
public PartitionRowCounter FindPartition(string id)
|
|
{
|
|
foreach (PartitionRowCounter partition in _partitions)
|
|
{
|
|
if (partition.Id == id)
|
|
{
|
|
return partition;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Verifies if table contains a partition with the specified Id.
|
|
/// </summary>
|
|
/// <param name="id"></param>
|
|
/// <returns>A Boolean specifying whether the partition is contained by the table.</returns>
|
|
public bool ContainsPartition(string id)
|
|
{
|
|
foreach (PartitionRowCounter partition in _partitions)
|
|
{
|
|
if (partition.Id == id)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public int CompareTo(ProcessingTable other) => string.Compare(this.Name, other.Name);
|
|
}
|
|
}
|