Use an enum to select which class to instantiate

Question

I have an enum that I am trying to associate to dto's:

 public enum DtoSelection
 {
     dto1,
     dto2,
     dto3,
 }

There are 108 and values in this enum.

I have a dto object for each of these dto's:

 public class dto1 : AbstractDto
 {
       public int Id { get; set; }
       //some stuff specific to this dto
 }

I am trying to make a method (eventually a service) that will return me a new dto object of the type associated to the the dto in question:

 private AbstractDto(int id)
 {
      if (id == DtoSelection.Dto1.ToInt()) //extension method I wrote for enums
            return new Dto1();
      if (id == DtoSelection.Dto2.ToInt())
            return new Dto2();
 }

Obviously I do not want to do this 108 times. For whatever reason my brain is just missing something obvious. What is the best way to handle this.

Answer 1

An elegant way of solving this is by using Attributes and one base class. Let me show you:

1. You must create a base class. In your example, could be AbstractDto, like following:

    public abstract class AbstractDto : Attribute
    {
         //code of AbstractDto       
    }
   

2. Then, we need to create a custom attribute that will be used on every Dto class to determine which enum corresponds to each class.

    public class DtoEnumAttribute : Attribute
    {
        public DtoSelection Enum { get; set; }
   
        public DtoEnumAttribute(DtoSelection enum)
        {
            this.Enum = enum;
        }
     }
   

3. Then we should decorate every child Dto with its proper enum. Let's do an example for Dto1:

    [DtoEnum(DtoSelection.Dto1)]
    public class Dto1 : AbstractDto
    {
         //code of Dto1
    }
   

4. Finally, you can use a method that can receive an specific enum and filter, or whatever logic you need. The following code will instantiate every class that inherit from AbstractDto ordered by the Enum that you have defined. You can use it on a Where clause to return only the instance of the class that matches the enum that you want. Ask me if you need help on this point.

    public void MethodToGetInstances()
    {
           IEnumerable<AbstractDto> dtos = typeof(AbstractDto)
               .Assembly.GetTypes()
               .Where(t => t.IsSubclassOf(typeof(AbstractDto)) && !t.IsAbstract)
               .Select(t => (AbstractDto)Activator.CreateInstance(t))
               .OrderBy(x => ((DtoEnumAttribute)x.GetType().GetCustomAttributes(typeof(DtoEnumAttribute), false).FirstOrDefault()).Enum);
   
           //If you have parameters on you Dto's, you might pass them to CreateInstance(t, params)
   
    }
   

On the dtos list, you will have the instances that you want. Hope it helps!