小酌重构系列[24]——封装集合

概述

当方法返回类型或属性类型为集合时，有些开发者会千篇一律地使用IList<T>集合。然而IList<T>具有集合的所有操作，这意味着调用者不仅可以读取集合信息，还能够修改集合。业务需求本来只是为调用者提供一个可读的集合，例如数据的查询和展示，但当方法返回IList<T>时，无疑隐式地开放了集合可写的权限。此时，我们无法阻止调用者篡改集合元素。

注意：将属性设定为IList<T>类型时，即使声明为只读的，我们仍然无法避免集合元素的篡改。
例如public IList<Person> People{get; private set;}，People属性是一个IList集合，它虽然是只读的，但是People集合里面的元素可以被修改。

这种情况下，我们应该使用IEnumerable<T>来代替IList<T>。
IList<T>和IEnumerable<T>都可以遍历集合的元素，IList<T>拥有集合的所有操作方法，包括集合元素的增加、修改和删除。
而IEnumerable<T>则只有一个GetEnumerator方法（扩展方法除外），它返回一个可用于循环访问集合的IEnumerator<T>对象。

示例

重构前

下面这段代码定义了两个类：Order和OrderLine。

• OrderLines属性是只读的
• Order提供了AddOrderLine()和RemoveOrderLine()方法，用于添加和删除订单明细

当调用者在取到Order实例时，仍然可以通过IList<T>的Add()或Remove()方法修改OrderLines集合。
假设调用者篡改了OrderLines中的元素，并且通过另外的方法回传了Order实例，则可能会产生一些bug。

隐藏代码

/// <summary>
/// 订单
/// </summary>
public class Order
{
    private readonly List<OrderLine> _orderLines = new List<OrderLine>();
    private double _orderTotal;

    public IList<OrderLine> OrderLines
    {
        get { return _orderLines; }
    }

    public void AddOrderLine(OrderLine orderLine)
    {
        _orderTotal += orderLine.Total;
        _orderLines.Add(orderLine);
    }

    public void RemoveOrderLine(OrderLine orderLine)
    {
        orderLine = _orderLines.Find(item => item == orderLine);
        if (orderLine == null)
            return;

        _orderTotal -= orderLine.Total;
        _orderLines.Remove(orderLine);
    }
}

/// <summary>
/// 订单明细
/// </summary>
public class OrderLine
{
    public double Total { get; set; }
}

重构后

重构后，不仅OrderLines集合是只读的，而且也只能通过AddOrderLine()和RemoveOrderLine()方法来增加或删除订单明细。

隐藏代码

/// <summary>
/// 订单
/// </summary>
public class Order
{
    private readonly List<OrderLine> _orderLines;
    private double _orderTotal;

    public Order(List<OrderLine> orderLines)
    {
        _orderLines = orderLines;
    }

    /// <summary>
    /// 订单明细集合是只读的，只能通过AddOrderLine()和RemoveOrderLine()来增加或删除订单明细
    /// </summary>
    public IEnumerable<OrderLine> OrderLines
    {
        get { return _orderLines; }
    }

    public double OrderTotal
    {
        get { return _orderTotal; }
    }

    public void AddOrderLine(OrderLine orderLine)
    {
        _orderTotal += orderLine.Total;
        _orderLines.Add(orderLine);
    }

    public void RemoveOrderLine(OrderLine orderLine)
    {
        orderLine = _orderLines.Find(item => item == orderLine);
        if (orderLine == null)
            return;

        _orderTotal -= orderLine.Total;
        _orderLines.Remove(orderLine);
    }
}

/// <summary>
/// 订单明细
/// </summary>
public class OrderLine
{
    public double Total { get; set; }
}

注意：上述代码有一些瑕疵，OrderLine类没有重写Object类的Equals()、GetHashCode()方法。
这行代码orderLine = _orderLines.Find(item => item == orderLine)会使得orderLine每次都是null。
较为完整的OrderLine如下：

隐藏代码

public class OrderLine
{
    public int Id { get; set; }
    public int OrderId { get; set; }
    public double Total { get; set; }


    /// <summary>
    /// 重写Equals方法
    /// </summary>
    public override bool Equals(object obj)
    {
        if (obj == null || !(obj is OrderLine))
            return false;

        if (ReferenceEquals(this, obj))
            return true;

        var other = (OrderLine) obj;
        if (IsTransient() && other.IsTransient())
            return false;

        var typeOfThis = GetType();
        var typeOfOther = other.GetType();

        if (!typeOfThis.IsAssignableFrom(typeOfOther) && !typeOfOther.IsAssignableFrom(typeOfThis))
        {
            return false;
        }

        return Id.Equals(other.Id);
    }

    /// <summary>
    /// 重写GetHashCode方法
    /// </summary>
    /// <returns></returns>
    public override int GetHashCode()
    {
        return Id.GetHashCode();
    }

    /// <summary>
    /// 是否为瞬时对象
    /// </summary>
    /// <returns></returns>
    public virtual bool IsTransient()
    {
        return EqualityComparer<int>.Default.Equals(Id, default(int));
    }

    /// <summary>
    /// 提供==操作符，可用于对象比较
    /// </summary>
    public static bool operator ==(OrderLine left, OrderLine right)
    {
        if (Equals(left, null))
        {
            return Equals(right, null);
        }

        return left.Equals(right);
    }

    /// <summary>
    /// 提供!=操作用，可用于对象比较
    /// </summary>
    public static bool operator !=(OrderLine left, OrderLine right)
    {
        return !(left == right);
    }
}

小结

当集合作为返回参数时，应使用适合业务需求的集合类型，不宜提供过多的集合操作给调用者。

小酌重构系列目录汇总

关注keepfool