source: branches/Persistence Test/HeuristicLab.DataAccess.ADOHelper/3.2/ManyToManyRelationHelper.cs @ 3025

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;

namespace HeuristicLab.DataAccess.ADOHelper {
  public class ManyToManyRelationHelper<AdapterT, RowT>
    where AdapterT : new()
    where RowT : System.Data.DataRow
    {
    ITableAdapterWrapper<AdapterT, ManyToManyRelation, RowT> tableAdapterWrapper;

    private int childIndex;

    public ManyToManyRelationHelper(
      ITableAdapterWrapper<AdapterT, ManyToManyRelation, RowT> tableAdapterWrapper, 
      int childIndex) {
      this.tableAdapterWrapper = tableAdapterWrapper;
      this.childIndex = childIndex;
    }

    public Session Session {
      set {
        tableAdapterWrapper.Session = value;
      }
    }

    public void UpdateRelationships(Guid objectA,
      IList<Guid> relationships) {
      UpdateRelationships(objectA, relationships, null);
    }

    public void UpdateRelationships(Guid objectA, 
      IList<Guid> relationships,
      IList<object> additionalAttributes) {
      //firstly check for created references
      IList<Guid> existing =
        this.GetRelationships(objectA); 

      foreach (Guid relationship in relationships) {
        if (!existing.Contains(relationship)) {
          ManyToManyRelation rel = 
            new ManyToManyRelation();
          rel.Id = objectA;
          rel.Id2 = relationship;
          if(additionalAttributes != null)
            rel.AdditionalAttributes = 
              new List<object>(additionalAttributes);
          
          RowT inserted = 
            tableAdapterWrapper.InsertNewRow(rel);

          tableAdapterWrapper.UpdateRow(inserted);
        }
      }

      //secondly check for deleted references
      ICollection<Guid> deleted =
        new List<Guid>();

      foreach (Guid relationship in existing) {
        if(!relationships.Contains(relationship)) {
          deleted.Add(relationship);
        }
      }

      foreach (Guid relationship in deleted) {
        RowT toDelete =
          FindRow(objectA, relationship);
        if (toDelete != null) {
          toDelete.Delete();
          tableAdapterWrapper.UpdateRow(toDelete);
        }
      }
    }

    public IList<Guid> GetRelationships(Guid objectA) {
      IList<Guid> result =
        new List<Guid>();

      IEnumerable<RowT> rows = 
        tableAdapterWrapper.FindById(objectA);

      foreach(RowT row in rows) {
        result.Add((Guid)row[childIndex]);
      }

      return result;
    }

    private RowT FindRow(Guid objectA, Guid objectB) {
      IEnumerable<RowT> rows =
       tableAdapterWrapper.FindById(objectA);

      RowT found = null;

      foreach (RowT row in rows) {
        if (row[childIndex].Equals(objectB)) {
          found = row;
          break;
        }
      }

      return found;
    }
  }
}