Modelling Authorization for a Social Network with Postgres RLS and Drizzle ORM
How to implement secure, user-specific access controls
Postgres’s Row-Level Security feature allows us to set up the access rules for our data. In other words, it can be used to declare the authorization model for an application or a service. As I’ve been trying to better understand RLS and its pros and cons, I decided to work on an RLS-backed authorization model for a social network. This social network’s data model is inspired by Twitter:
- Users
- And user profiles
- Public posts (tweets)
- With public comments (tweet responses)
- Private Chats
- With chat messages and chat participants
This ended up being harder than I expected, so a few of us had to collaborate on the implementation. You can find the full implementation on this GitHub repository here.
If you’re new to RLS, you might want to read some docs or watch a good Youtube video before moving on. Notice also that we’ll be using the open-source pg_session_jwt Postgres extension for the authentication to Postgres (the auth.user_id()
function in the code snippets comes from this extension).
Users and Profiles
Let’s start with users and their profiles. This is what we’ve ended up with:
To simplify the authorization implementation, we’ve separated the users
and the user_profiles
tables. The users
table has no RLS policies, meaning that nobody can read or write to it. So, how does it get populated if this is a real application?
- One option is that this table is just being synchronized from an external auth provider using a FDW or some data synchronization job.
- Another option, if you roll your own auth, is to allow your app to read and write to this table using a Postgres role that has the
BYPASSRLS
attribute.
Then, the user_profiles
table has fairly simple RLS policies:
- A user can update its own profile
- Anyone (authenticated or not) can see anyone’s profile
Posts and Comments
Posts and comments have a very simple schema:
The posts and comments are both simple:
- You can only insert/update a new post or comment if you’re the author of said post or comment
- Anyone (authenticated or not) can view any post or any comment
Chats, chat participants and chat messages
The private chats have more complex requirements around who can read/write what, so let’s break it down table by table.
Chats
The chats
table is a good place to start:
First of all, you’ll notice that anonymous users (i.e., unauthenticated users) cannot see the list of chats. In fact, chats are only visible to authenticated users who happen to be a participant in the chat (as per the my_chats_participants
view – which we’ll get to in a bit).
Chat participants
The chat participants table is more complex. We need a table, as well as a view, since we need a circular RLS policy that allows one to see all other chat participants in a chat if they’re a participant themselves. The best way to handle this using Drizzle is with a table+view.
The view is necessary because RLS does not support rules that filter a table based on its own data in a recursive way. Specifically, RLS cannot handle conditions like: “Show only the chat participants of chats where I am also a participant.” Attempting to enforce this rule directly on the chatParticipants
table leads to a recursion error. Using a view allows us to apply this filtering logic without running into RLS limitations. That way, the view returns the list of chat participants for each chat that the user belongs to.
While using a view can be very effective to solve any “recursion issues” in RLS, it also comes with a catch — by default, the view inherits the permissions of the role that created it. In our case, that would be the database owner, which has full access to the database.
To ensure security, it’s essential to make this view read-only. This approach allows the view to serve as a safe, accessible data layer without granting unintended write permissions. All data modifications can then be handled directly through the chat_participants
table.
PostgreSQL doesn’t provide an explicit option to make a view read-only, but there’s a reliable workaround. By adding certain SQL clauses — such as JOIN
, WITH
, or DISTINCT
— we can turn the view into a “non-simple view”, A non-simple view in PostgreSQL is inherently read-only, preventing any data modification.
Chat messages
Our chat_messages
table gets to reuse our prior my_chat_participants
view to see who can read/insert what, and no one is allowed to modify or delete messages, whether they be their own or others.
We have two policies:
- We use
crudPolicy
to generate a single RLS policy forSELECT
. The policy dictates that you can only see messages in a chat if you are a participant of that chat. (We could have done this withpgPolicy
too, butcrudPolicy
is slightly nicer.) - We use
pgPolicy
to generate a single RLS policy forINSERT
. This policy dictates that you can only insert chat messages of which you are the author, and in chats where you are a participant. - Editing or deleting chat messages is not possible in this model (since we don’t have any policies for the UPDATE or DELETE operations).
Summing up
I have a few takeaways from having worked on this project:
- Writing RLS policies in raw SQL is just a difficult experience with lots of pitfalls. You have to write a lot of long statements and repetitive code. And the same is true with Drizzle’s pgPolicy. Instead, I advocate for using crudPolicy from drizzle-orm/neon — and this will work on any Postgres database, not just Neon!
- In general, if you’re going to expose your database schema directly to your frontend (either with SQL from the client, or with Postgrest), you need to put a lot of time and effort into validating your RLS policies. It’s extremely easy to forget to have all the checks, and that can lead to data leaks.
- This is not the kind of work you can trust an LLM for.
- In the same vein, a database schema that’s exposed via Postgrest or SQL from the frontend needs to be carefully designed besides just the RLS policies.
- For example, we had to make sure that the
userId
column in theuser_profiles
table is tagged withunique
. This is because we want to prevent a user from creating multiple profiles for themselves. - If you have a traditional backend, a lot of this logic can be encapsulated there.
- For example, we had to make sure that the
- Testing RLS policies is hard. I know there are some tools out there that simplify this (and we have some ideas ourselves), but we had to actually try to create an app using Neon Authorize in order to fully test this schema.
- This schema not only has access logic baked into it, as well as some other logical constraints (column uniqueness, cascade deletes and referential integrity via foreign keys). I believe it’s a good idea, most of the time, to push down all this logic to the database layer.
Finally, I’d like to thank some folks at Neon who helped get this example together (Pedro, Bryan, Jakub and others). If you’d like to learn more about Neon Authorize, you can find the docs here. And if you’re curious about the future of Neon Authorize, we’ve also written at length about it here.