Click here to Skip to main content
15,867,488 members
Articles / Programming Languages / C#
Article

Enhancing C# Enums

Rate me:
Please Sign up or sign in to vote.
3.60/5 (41 votes)
11 Oct 20073 min read 158.7K   466   41   32
How to make a class look like an enum with added functionality

Introduction

An enum in C# is a value type that is used to represent a set of fixed distinct values so that it can be used later in program logic. A very straightforward example of a classic enum declaration is given below:

C#
public enum Rating
{
    Awful = 1,
    Bad = 2,
    Medium = 3,
    Good = 4,
    Awesome = 5
}

The enum above describes a set of possible values that a variable of type Rating can assume. This way, we can use it in our code to implement any kind of logic in a type-safe way. For instance, if we have a Rating variable called userRating, we could use it like this:

C#
if(userRating == Rating.Awesome)
{
    //Do something
}

If we try to cast an int that was not defined in the enum list, I would expect the runtime to throw an exception. However, the funny thing is that the line of code bellow will create a variable rating with the value 6.

C#
Rating rating = (Rating)6; //does not throw an exception

The way I see it, it is not a good thing, since our enum could be used improperly and we would only be able to validate it in another class. Besides that, sometimes it's useful to add some functionality to enums. The problem is that they have some limitations. Since it's not a class, it cannot be extended and its defined values are limited to a small set of primitive types (byte, sbyte, short, ushort, int, uint, long, ulong). For instance, sometimes it is very convenient to get the list of enum values and display it in a combobox. Although there are some reflection techniques that let you define and retrieve descriptions for the enum items, you are always limited to a key and a value.

Solution

To overcome these limitations, we can use a regular class to represent an enum while keeping the same basic characteristics. Here's the code of the base class that we will use to implement the enhanced enums.

C#
using System.Collections.ObjectModel;
public abstract class EnumBaseType<T> where T : EnumBaseType<T>
{
    protected static List<T> enumValues = new List<T>();

    public readonly int Key;
    public readonly string Value;

    public EnumBaseType(int key, string value)
    {
        Key = key;
        Value = value;
        enumValues.Add((T)this);
    }

    protected static ReadOnlyCollection<T> GetBaseValues()
    {
        return enumValues.AsReadOnly();
    }

    protected static T GetBaseByKey(int key)
    {
        foreach (T t in enumValues)
        {
            if(t.Key == key) return t;
        }
        return null;
    }
    
    public override string ToString()
    {
    return Value;
    }
}

Note that in this class we define a key and a value field. This corresponds to the classical enum implementation, but now we are not limited to a key and a value. If required, we can extend this class and add new fields to the structure. Here is the actual implementation of our enhanced Rating enum:

C#
public class Rating : EnumBaseType<Rating>
{
    public static readonly Rating Awful = new Rating(1, "Awful");
    public static readonly Rating Bad = new Rating(2, "Bad");
    public static readonly Rating Medium = new Rating(3, "Medium");
    public static readonly Rating Good = new Rating(4, "Good");
    public static readonly Rating Awesome = new Rating(5, "Awesome");

    public Rating(int key, string value) : base(key, value)
    {
    }

public static ReadOnlyCollection<Rating> GetValues()
    {
        return GetBaseValues();
    }

    public static Rating GetByKey(int key)
    {
        return GetBaseByKey(key);
    }
}

Now it can be used as a regular enum in an if statement and can also iterate through the values defined in our class. Notice that, because they are static, we still need to explicitly implement the methods GetValues() and GetByKey() if we want to use them. Otherwise, the enumValues collection would be empty.

C#
foreach (Rating rating in Rating.GetValues())
{
    Console.Out.WriteLine("Key:{0} Value:{1}", rating.Key, rating.Value);

    if (rating == Rating.Awesome)
    {
        //Do something
    }
}

Now let's pretend we are using our Rating enum to rate movies on a website and we want to associate a "Must See" field to the rating, so that Good and Awesome movies display the "Must See" icon besides it. To do that, we have to add a new field to our Rating class. So now the Rating implementation is going to look like this:

C#
public class Rating : EnumBaseType<Rating>
{
    public static readonly Rating Awful = new Rating(1, "Awful", false);
    public static readonly Rating Bad = new Rating(2, "Bad", false);
    public static readonly Rating Medium = new Rating(3, "Medium", false);
    public static readonly Rating Good = new Rating(4, "Good", true);
    public static readonly Rating Awesome = new Rating(5, "Awesome", true);

    public readonly bool MustSee;

    public Rating(int key, string value, bool mustSee) : base(key, value)
    {
        this.MustSee = mustSee;
    }

    public static ReadOnlyCollection<Rating> GetValues()
    {
        return GetBaseValues();
    }

    public static Rating GetByKey(int key)
    {
        return GetBaseByKey(key);
    }
}

Although we added new information to the Rating class, the clients using this enhanced enum will not be affected by this change. In addition, they will have the new MustSee field available.

C#
foreach (Rating rating in Rating.GetValues())
{
    if (rating.MustSee)
    {
        Console.Out.WriteLine("This is a Must See movie!");
    }
}

Drawback

The only drawback of this approach is that the enhanced enum cannot be used in a switch statement, since its values are not constants. However, I still think it's worth using it when you need advanced functionality associated with your enums.

Conclusion

In this article, we saw how to use classes to simulate the behavior of enums while leveraging OOP techniques to enhance their capabilities. I hope, in the future, that the C# team includes some syntax sugar to allow the use of advanced enums without the need to write custom code and without the drawback concerning switch statements.

History

  • 9 October, 2007 -- Original version posted
  • 11 October, 2007 -- First update

License

This article has no explicit license attached to it but may contain usage terms in the article text or the download files themselves. If in doubt please contact the author via the discussion board below.

A list of licenses authors might use can be found here


Written By
Software Developer (Senior) Intelligent Coder
Canada Canada
I've been developing .NET enterprise applications since 2000.

I am originally from Rio de Janeiro and I am currently working at http://www.intelligentcoder.com in Ontario.

I also have my own startup where we offer client intake forms.

Comments and Discussions

 
QuestionAdding another value... Pin
Bass2Four16-Dec-21 4:45
Bass2Four16-Dec-21 4:45 
QuestionNot working on some cases Pin
slelong15-Nov-14 23:44
slelong15-Nov-14 23:44 
AnswerRe: Not working on some cases Pin
tridy4-Mar-15 21:11
tridy4-Mar-15 21:11 
GeneralYeah but ... Pin
thugthug31-Mar-10 9:23
thugthug31-Mar-10 9:23 
GeneralRe: Yeah but ... Pin
charles henington4-Feb-13 5:12
charles henington4-Feb-13 5:12 
GeneralExtend some more... Pin
dancerjohn16-Oct-07 3:08
dancerjohn16-Oct-07 3:08 
GeneralRe: Extend some more... Pin
Cassio Mosqueira16-Oct-07 4:20
Cassio Mosqueira16-Oct-07 4:20 
GeneralSystem.Enum Pin
Rdunzl14-Oct-07 15:17
Rdunzl14-Oct-07 15:17 
GeneralRe: System.Enum Pin
Cassio Mosqueira14-Oct-07 15:36
Cassio Mosqueira14-Oct-07 15:36 
GeneralRe: System.Enum [modified] Pin
Rdunzl14-Oct-07 19:45
Rdunzl14-Oct-07 19:45 
GeneralRe: System.Enum Pin
Cassio Mosqueira15-Oct-07 3:22
Cassio Mosqueira15-Oct-07 3:22 
GeneralRe: System.Enum Pin
Rdunzl15-Oct-07 20:12
Rdunzl15-Oct-07 20:12 
GeneralEnum statics Pin
Thomas Lykke Petersen11-Oct-07 19:31
Thomas Lykke Petersen11-Oct-07 19:31 
GeneralGood idea, but ... Pin
Jakub Mller10-Oct-07 22:04
Jakub Mller10-Oct-07 22:04 
GeneralGood Job Pin
merlin98110-Oct-07 3:49
professionalmerlin98110-Oct-07 3:49 
GeneralRe: Good Job Pin
Cassio Mosqueira10-Oct-07 4:55
Cassio Mosqueira10-Oct-07 4:55 
Thank you.

Regards,
Cassio Alves
Cool | :cool:

Generalbase methods can be published Pin
_SAM_9-Oct-07 20:38
_SAM_9-Oct-07 20:38 
GeneralRe: base methods can be published Pin
Cassio Mosqueira10-Oct-07 4:50
Cassio Mosqueira10-Oct-07 4:50 
GeneralRe: base methods can be published Pin
_SAM_10-Oct-07 5:23
_SAM_10-Oct-07 5:23 
GeneralRe: base methods can be published Pin
Cassio Mosqueira10-Oct-07 13:07
Cassio Mosqueira10-Oct-07 13:07 
GeneralRe: base methods can be published Pin
dancerjohn16-Oct-07 3:20
dancerjohn16-Oct-07 3:20 
GeneralRe: base methods can be published Pin
Tim Schwallie16-Oct-07 16:23
Tim Schwallie16-Oct-07 16:23 
QuestionHow about a Dictionary? Pin
PIEBALDconsult9-Oct-07 16:28
mvePIEBALDconsult9-Oct-07 16:28 
AnswerRe: How about a Dictionary? Pin
Cassio Mosqueira9-Oct-07 18:10
Cassio Mosqueira9-Oct-07 18:10 
GeneralRe: How about a Dictionary? Pin
PIEBALDconsult10-Oct-07 13:33
mvePIEBALDconsult10-Oct-07 13:33 

General General    News News    Suggestion Suggestion    Question Question    Bug Bug    Answer Answer    Joke Joke    Praise Praise    Rant Rant    Admin Admin   

Use Ctrl+Left/Right to switch messages, Ctrl+Up/Down to switch threads, Ctrl+Shift+Left/Right to switch pages.