Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Get the reference of a collection given a type

Tags:

c#

I have several dictionaries:

Dictionary<int, Type1> Type1Dictionary { get; set; }
Dictionary<int, Type2> Type2Dictionary { get; set; }
Dictionary<int, Type3> Type3Dictionary { get; set; }
Dictionary<int, Type4> Type4Dictionary { get; set; }

Where Typei (i = 1..4) are derived from the same base class (BaseType). I want a method that returns the reference of a dictionary given a type. Later, I will perform some operations like Add or Remove on that dictionary:

Type1 example = new Type1(); 
var dic = GetDictionary(example);
dic.Add(example.ID, example);

Notes: I don't want to set my dictionaries as Dictionary<int, BaseType>

I could write something like this but that wouldn't return a reference to the dictionary:

Dictionary<int, BaseType> GetDictionary(BaseType myObject)
{
    var dic = new Dictionary<int, BaseType>();
    if(myObject is Type1)
    {
    //ideally I would return my Type1Dictionary here but I can't due type incompatibility
       foreach(var x in Type1Dictionary)
       {
            dic.Add(x.Key, x.Value);
       }
       return dic;
    }
    if(myObject is Type2) { /*...*/ }
    if(myObject is Type3) { /*...*/ }
    if(myObject is Type4) { /*...*/ }
}

EDIT:

What I really want is to avoid the following structure:

AddObject(BaseType x)
{
    Type1 x1 = x as Type1;
    if(x1 != null) { Type1Dictionary.Add(x1.ID, x1); }

    Type2 x2 = x as Type2;
    if(x2 != null) { Type2Dictionary.Add(x2.ID, x2); }

    Type3 x3 = x as Type3;
    if(x3 != null) { Type3Dictionary.Add(x3.ID, x3); }

    Type4 x4 = x as Type4;
    if(x4 != null) { Type4Dictionary.Add(x4.ID, x4); }
}

RemoveObject(BaseType x)
{
    Type1 x1 = x as Type1;
    if(x1 != null) { Type1Dictionary.Remove(x1.ID); }

    Type2 x2 = x as Type2;
    if(x2 != null) { Type2Dictionary.Remove(x2.ID); }

    Type3 x3 = x as Type3;
    if(x3 != null) { Type3Dictionary.Remove(x3.ID); }

    Type4 x4 = x as Type4;
    if(x4 != null) { Type4Dictionary.Remove(x4.ID); }
}

But instead:

AddObject(BaseType x)
{
    var dic = GetDictionary(x);
    dic.Add(x.ID, x);
}

RemoveObject(BaseType x)
{
    var dic = GetDictionary(x);
    dic.Remove(x.ID);
}
like image 795
Sturm Avatar asked Nov 11 '15 08:11

Sturm


1 Answers

This can be polished in the terms of thead safety, etc. But you should be able to get the basic idea:

public interface IEntity
{
  int ID { get; }
}

public class Superset<T> where T : IEntity
{
  public Dictionary<Type, Dictionary<int, T>> m_Map = 
    new Dictionary<Type, Dictionary<int, T>>();

  private Dictionary<int, T> GetDictionary(Type t)
  {
    Dictionary<int, T> result = null;
    if (!m_Map.TryGetValue(t, out result))
    {
      result = new Dictionary<int, T>();
      m_Map.Add(t, result);
    }
    return result;
  }

  public void Add<K>(K item) where K : T
  {
    GetDictionary(typeof(K)).Add(item.ID, item);
  }

  public bool Remove<K>(K item) where K : T
  {
    return GetDictionary(typeof(K)).Remove(item.ID);
  }
}
like image 88
Ondrej Svejdar Avatar answered Oct 15 '22 12:10

Ondrej Svejdar