Types and Roles If Not Exists in PostgreSQL

| databases | programming | sql |

For certain operations such as creating types and roles in PostgreSQL you are not able to use the handy IF NOT EXISTS parameter. This makes creating reusable scripts a bit more challenging, especially if you are going to be running these scripts more than once against the same database. On solution to this problem is to Just Do It™. DO allows you to execute anonymous code blocks. For example, suppose we had a simple task table with a custom type called task_status along with a readerrole that has only select permissions. Using the pg_type and pg_roletables we can write a DO expression that will check to make sure that the elements do not exist before attempting to create them, which will prevent your script from erroring out in the event that they do exist. First we will create the custom type.

DO $$
BEGIN
    IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'task_status') THEN
        CREATE TYPE task_status AS ENUM ('todo', 'doing', 'blocked', 'done');
    END IF;
END
$$;
Then we can create our table.
CREATE TABLE IF NOT EXISTS
tasks (
  id           integer PRIMARY KEY,
  title        varchar(200), 
  status       task_status NOT NULL DEFAULT 'todo',
  created_date timestamp
);
Next, we can create our new role called reader and grant read only rights on the tasks table.
DO $$
BEGIN
    IF NOT EXISTS (SELECT 1 FROM pg_roles WHERE rolname = 'reader') THEN
        CREATE ROLE reader;
    END IF;
END
$$;
GRANT SELECT ON tasks TO reader;
The entire script is shown below.

You can run this script as many times as you wish and it will just work™, which means that if you are running full migrations then you do not have to do any manual overrides or comment anything out. Note that the reason why we do not have to wrap up the GRANT in a check is because if you duplicate a grant, PostgreSQL will return a notice but not an error.

Thank you for reading! Share your thoughts with me on bluesky, mastodon, or via email.

Check out some more stuff to read down below.

Most popular posts this month

Recent Favorite Blog Posts

This is a collection of the last 8 posts that I bookmarked.

Articles from blogs I follow around the net

Pluralistic: Battery rationality (06 Dec 2024)

Today's links Battery rationality: Ringing down the curtain on 9/11's security theater. Hey look at this: Delights to delectate. This day in history: 2009, 2014, 2019, 2023 Upcoming appearances: Where to find me. Recent appearances: Where I've…

via Pluralistic: Daily links from Cory Doctorow December 6, 2024

P&B: Erica Fustero

This is the 67th edition of People and Blogs, the series where I ask interesting people to talk about themselves and their blogs. Today we have Erica Fustero and her blog, ericafustero.com To follow this series subscribe to the newsletter. …

via All the interviews for the People and Blogs series. December 6, 2024

P&B: Erica Fustero

This is the 67th edition of People and Blogs, the series where I ask interesting people to talk about themselves and their blogs. Today we have Erica Fustero and her blog, ericafustero.com To follow this series subscribe to the newsletter. …

via Manuel Moreale RSS Feed December 6, 2024

Generated by openring