How to use enum attributes in Ruby on Rails

honeybadger_staff

Honeybadger Staff

Posted on March 4, 2024

How to use enum attributes in Ruby on Rails

This article was originally written by Jeffery Morhous on the Honeybadger Developer Blog.

Enums, or enumerations, are an incredibly common way to represent options for an attribute of a model in Rails. If a model has an attribute-like status, it is often a string with predefined options. In Rails, you can represent this as an integer and automatically convert it to a string using built-in methods! This makes presenting options to a user as strings while storing these selections as a number straightforward and maintainable.

In this article, we'll discuss why you might want to use enums in your code, how to add an enum attribute to an existing table, how to create a new table with an enum, and how to use enums in your application.

Why enums are useful

Enums in Rails allow us to map a set of symbolic keys to specific integer values in the database. This makes them an efficient way to represent and store a limited set of values for an attribute,
such as a status, without having to use strings or create additional tables mapping strings to integers. Using enums also brings consistency and readability to the code, making it easier for developers to understand the possible values of an attribute without having to dig through the database schema.

In addition to simplifying data representation and storage, enums also come with useful utility methods. For example, enums automatically generate scope methods for each value, making it easy to query and filter records based on the enum attribute. They also generate methods for each enum value, so the code can easily check the current value of an enum attribute in a more readable and expressive manner. Enums are often leveraged in form helpers, making presenting options in a form straightforward.

Adding an enum to an existing table

To add an enum attribute to an existing table in Rails, start by creating a new migration.

For example, if you're adding an attribute status to the model course, run the following:

rails generate migration AddStatusToCourse status:integer
Enter fullscreen mode Exit fullscreen mode

If you want the attribute to have a default value, open the newly created migration in db/migrate and edit the add_column method call to have the default value you want:

add_column :courses, :status, :integer, default: 0
Enter fullscreen mode Exit fullscreen mode

To apply the migration to your database, run the following:

rails db:migrate
Enter fullscreen mode Exit fullscreen mode

Next, edit your model to declare the enum and its mappings of strings to integers.

class Course < ApplicationRecord
  enum status: {
    pending: 0,
    active: 1,
    archived: 2 
  }
end
Enter fullscreen mode Exit fullscreen mode

That's it! You've added an attribute to your model and defined it as an enum; all that's remaining is to actually use the enum.

Creating an enum with a new database table

If you don't already have a model with which you'd like to use an enum, adding an enum attribute is as easy as any other attribute you'll need for that model. If you haven’t generated the
course model, for example, you can begin by generating a new model.

rails generate model Course name:string status:integer
Enter fullscreen mode Exit fullscreen mode

Open up your latest migration, and ensure it looks like this:

class CreateFamilies < ActiveRecord::Migration[7.0]
  def change
    create_table :users do |t|
      t.string  :name
      t.integer :status
    end
  end
end
Enter fullscreen mode Exit fullscreen mode

Finally, run the migration:

rails db:migrate
Enter fullscreen mode Exit fullscreen mode

Finally, add the enum definition to your model, as shown in the
previous section:

class Course < ApplicationRecord
  enum status: {
    pending: 0,
    active: 1,
    archived: 2 
  }
end
Enter fullscreen mode Exit fullscreen mode

Setting an enum value

Setting the value of an enum attribute of a model is readable, even though the data is saved as an integer.

In what is probably the most common method, you can set the value using the symbolic key:

course.status = :active
course.save
Enter fullscreen mode Exit fullscreen mode

You can also set the value using the string representation of the symbolic key:

course.status = "active"
course.save
Enter fullscreen mode Exit fullscreen mode

If you're initializing the object, you can set the value during initialization:

course = Course.new(status: :active)
course.save
Enter fullscreen mode Exit fullscreen mode

You can even use the bang methods for each option generated by the enum:

course = Course.new
course.active!
Enter fullscreen mode Exit fullscreen mode

These are some of the most common ways to set the value of an enum attribute in Rails. It's generally recommended to use the symbolic keys or their string representation rather than the integer directly, as they make your code more readable and expressive.

Checking an enum value

While you can technically check the integer value of an enum, Rails generates predicate methods that make checking the value of an enum easy and readable:

course = Course.new(status: :active)

course.pending?  # returns false
course.active?   # returns true
course.archived? # returns false
Enter fullscreen mode Exit fullscreen mode

You can also compare the enum attribute directly with a symbolic key or even its string representation:

course = Course.new(status: :active)

course.status == :active   # returns true
course.status == "active"  # also returns true!
Enter fullscreen mode Exit fullscreen mode

There are more ways to compare enum values, but these are the most common.

Conclusion

Overall, enums in Ruby on Rails offer a clean and efficient way to represent a finite set of values for a model attribute. Their integration with Rails' built-in features, such as scope and predicate methods, makes them a powerful and convenient choice for managing such attributes. In turn, this leads to improved
code readability, maintainability, and a more enjoyable development experience.

Leaning on Rails conventions when using enums ensures the seamless translation of symbols into stored integer values and affords
straightforward methods for checking those values.

💖 💪 🙅 🚩
honeybadger_staff
Honeybadger Staff

Posted on March 4, 2024

Join Our Newsletter. No Spam, Only the good stuff.

Sign up to receive the latest update from our blog.

Related