2 minute read

This is a neat trick I discovered recently

enums are a very useful concept in programming and allow us to define constant values which can easily hold grouped data and even declare members

I use them heavily in my automation code to define constant values and they work very well when we have the define different branches in when clause in kotlin to maybe perform different sort of actions

A common use case

One common use case for me is when I want to do enum constant lookup using its value. For example, I might declare different configurations for my tests and based on these values, I might want to take different actions, trigger some flows or perform some assertions

Let’s understand this with an example

In the above example, we have defined an enum class called Mouseswhich has all different mouses and a short brand name for each.

If we want to get the enum constant value which has the value = “logi” then we can write a simple foreach loop in kotlin to iterate over all the values and if the desired value is found (Ln 17 – 21) then assign it to the temporary enum variable (Ln 15)

Fair and simple enough right.

However, we would have to write this simple logic again and again for all the different enum constants that we declare.

A slightly better way using companion objects

The above code gets the job done, however, this could be written in a slightly more elegant way.

Below is the updated code

Let’s understand this:

Ln 10 – 13, we have declared a <a href="https://kotlinlang.org/docs/tutorials/kotlin-for-py/objects-and-companion-objects.html">companion object</a> block to tie a member and a function to the enum class itself rather than its instance

Ln 11: We have a private immutable variable called mapping which has the mapping of all the enum constants with their values. This is done by using associateBy() which is called upon the values() array

if we peek into the code for this function, we can see that it returns a Map<K, T>, where K is the value of the enum constant and T is the actual constant itself

Finally, we have a fromValue() method which accepts the enum constant value and returns the associated constant from the mapping variable. If for some reason the value does not match an existing constant, then we raise an exception like below (Ln 19) with the name of the class

java.lang.IllegalStateException: Look up failed for class experiments.MousesV1

Now we can easily call this fromValue method on the enum constant and get the associated constant (Ln 18)

And that’s it. This has already made my code look much more concise and made it easier for me to use enums. Hopefully, you found this useful.

Until next time. Happy coding.

Categories:

Updated:

Comments