MES-ETL/ConsoleApp2/DataRecord.cs

namespace ConsoleApp2;

public class DataRecord : ICloneable
{
    public static bool TryGetField(DataRecord record, string columnName, out string value)
    {
        value = string.Empty;
        if (record.Headers is null)
            throw new InvalidOperationException("Cannot get field when headers of a record have not been set.");
        var idx = IndexOfIgnoreCase(record.Headers, columnName);
        if (idx == -1)
            return false;
        value = record.Fields[idx];
        return true;
    }

    public static string GetField(DataRecord record, string columnName)
    {
        if (record.Headers is null)
            throw new InvalidOperationException("Headers have not been set.");
        var idx = IndexOfIgnoreCase(record.Headers, columnName);
        if (idx is -1)
            throw new IndexOutOfRangeException(
                $"Column name '{columnName}' not found in this record, table name '{record.TableName}'.");
        return record.Fields[idx];
    }

    private static int IndexOfIgnoreCase(IList<string> list, string value)
    {
        var idx = -1;
        for (var i = 0; i < list.Count; i++)
        {
            if (list[i].Equals(value, StringComparison.OrdinalIgnoreCase)) 
            {
                idx = i;
                break;
            }
        }

        return idx;
    }

    public string? RawField { get; set; }
    public IList<string> Fields { get; }
    public IList<string> Headers { get; }
    public string TableName { get; }
    public string? Database { get; set; }


    public DataRecord(IEnumerable<string> fields, string tableName, IEnumerable<string> headers, string? database = null)
    {
        Fields = fields.ToList();
        TableName = tableName;
        Headers = headers.ToList();
        Database = database;
        
        if (Fields.Count != Headers.Count)
            throw new ArgumentException(
                $"The number of fields does not match the number of headers. Expected: {Headers.Count} Got: {Fields.Count} Fields: {string.Join(',', Fields)}",
                nameof(fields));
    }

    public string this[int index]
    {
        get => Fields[index];
        set => Fields[index] = value;
    }

    public string this[string columnName]
    {
        get => GetField(this, columnName);
        set => SetField(columnName, value);
    }

    public int FieldCount => Fields.Count;

    public bool TryGetField(string columnName, out string value) => TryGetField(this, columnName, out value);

    public bool SetField(string columnName, string value) => SetField(this, columnName, value);

    public bool SetField(DataRecord record, string columnName, string value)
    {
        if (record.Headers is null)
            throw new InvalidOperationException("Headers have not been set.");
        var idx = IndexOfIgnoreCase(record.Headers, columnName);
        if (idx is -1)
            throw new IndexOutOfRangeException(
                $"Column name '{columnName}' not found in this record, table name '{record.TableName}");
        record.Fields[idx] = value;
        return true;
    }
    
    public void AddField(string columnName, string value)
    {
        if (IndexOfIgnoreCase(Headers, columnName) != -1)
            throw new InvalidOperationException($"{TableName}: 列名 '{columnName}' 已存在");
        
        Fields.Add(value);
        Headers.Add(columnName);
    }
    
    public void RemoveField(string columnName)
    {
        var idx = IndexOfIgnoreCase(Headers, columnName);
        if (idx == -1)
            throw new InvalidOperationException($"{TableName}: 列名 '{columnName}' 不存在");
        
        Fields.RemoveAt(idx);
        Headers.Remove(columnName);
    }

    public bool HeaderExists(string columnName) => IndexOfIgnoreCase(Headers, columnName) != -1;
    public object Clone()
    {
        return new DataRecord(new List<string>(Fields), TableName, new List<string>(Headers), Database);
    }
}