Nest.js Tutorial

Relationships with PostgreSQL and MikroORM

Marcin Wanago
NestJSSQL

A significant advantage of SQL databases is handling relationships between various tables. Since, in web applications, entities often relate to each other, designing relationships is a big part of working with SQL databases. In this article, we continue learning MikroORM and use it to form relationships.

You can find the code from this article in this repository.

One-To-One

With a one-to-one relationship, a row from the first table has just one matching row from the second table and the other way around. An elementary example of that is creating an address entity.

address.entity.ts
1import { Entity, Property, PrimaryKey } from '@mikro-orm/core';
2 
3@Entity()
4class Address {
5  @PrimaryKey()
6  id: number;
7 
8  @Property()
9  street: string;
10 
11  @Property()
12  city: string;
13 
14  @Property()
15  country: string;
16}
17 
18export default Address;

Once we do that, we can use the @OneToOne() decorator to declare the one-to-one relationship.

user.entity.ts
1import { Entity, Property, PrimaryKey, OneToOne } from '@mikro-orm/core';
2import Address from './address.entity';
3 
4@Entity()
5class User {
6  @PrimaryKey()
7  id: number;
8 
9  @Property({ unique: true })
10  email: string;
11 
12  @Property()
13  name: string;
14 
15  @Property({ hidden: true })
16  password: string;
17 
18  @OneToOne({ nullable: true })
19  address?: Address;
20}
21 
22export default User;

In our application, we assign just one user to a particular address. Because of that, it is a fitting example of a one-to-one relationship.

A thing worth noting is that above, we use the serialization built into MikroORM to hide the password property using the hidden: true option. Instead, we would use the serialization built into NestJS, but unfortunately, that would not work correctly with MikroORM.

When creating the user, we don’t need to create the address explicitly. Instead, MikroORM does that for us by default.

users.service.ts
1import { Injectable } from '@nestjs/common';
2import { CreateUserDto } from './dto/createUser.dto';
3import { InjectRepository } from '@mikro-orm/nestjs';
4import { EntityRepository } from '@mikro-orm/core';
5import User from './user.entity';
6 
7@Injectable()
8class UsersService {
9  constructor(
10    @InjectRepository(User)
11    private readonly userRepository: EntityRepository<User>,
12  ) {}
13 
14  async create(user: CreateUserDto) {
15    const newUser = await this.userRepository.create(user);
16    await this.userRepository.persistAndFlush(newUser);
17    return newUser;
18  }
19  
20  // ...
21}
22 
23export default UsersService;
The application we build in this article uses authentication. If you wanto to know more about it, check out API with NestJS #3. Authenticating users with bcrypt, Passport, JWT, and cookies

The migration

Doing all of the above and running npx mikro-orm migration:create gives us the following migration:

Migration20220529210655.ts
1import { Migration } from '@mikro-orm/migrations';
2 
3export class Migration20220529210655 extends Migration {
4 
5  async up(): Promise<void> {
6    this.addSql('create table "address" ("id" serial primary key, "street" varchar(255) not null, "city" varchar(255) not null, "country" varchar(255) not null);');
7 
8    this.addSql('alter table "user" add column "address_id" int null;');
9    this.addSql('alter table "user" add constraint "user_address_id_foreign" foreign key ("address_id") references "address" ("id") on update cascade on delete set null;');
10    this.addSql('alter table "user" add constraint "user_address_id_unique" unique ("address_id");');
11  }
12 
13  async down(): Promise<void> {
14    this.addSql('alter table "user" drop constraint "user_address_id_foreign";');
15 
16    this.addSql('drop table if exists "address" cascade;');
17 
18    this.addSql('alter table "user" drop constraint "user_address_id_unique";');
19    this.addSql('alter table "user" drop column "address_id";');
20  }
21 
22}
If you want to know more about generating migrations with MikroORM, check out API with NestJS #62. Introduction to MikroORM with PostgreSQL

The crucial thing to notice is that MikroORM adds the address_id column to the user table and makes sure it references the address table.

MikroORM also adds a unique constraint to the address_id column. Doing that makes sure the relationship is one-to-one because only one user can refer to a particular address. Trying to relate a second user to the same address would result in an error.

Fetching the related entities

We need to explicitly tell MikroORM to fetch the address of a given user if we want to do that.

users.service.ts
1import { Injectable } from '@nestjs/common';
2import { UserNotFoundException } from './exceptions/userNotFound.exception';
3import { InjectRepository } from '@mikro-orm/nestjs';
4import { EntityRepository } from '@mikro-orm/core';
5import User from './user.entity';
6 
7@Injectable()
8class UsersService {
9  constructor(
10    @InjectRepository(User)
11    private readonly userRepository: EntityRepository<User>,
12  ) {}
13 
14  async getByEmail(email: string) {
15    const user = await this.userRepository.findOne(
16      {
17        email,
18      },
19      {
20        populate: ['address'],
21      },
22    );
23    if (!user) {
24      throw new UserNotFoundException();
25    }
26    return user;
27  }
28  
29  // ...
30}
31 
32export default UsersService;

Thanks to doing the above, whenever we call the getByEmail function, we get both the user and the address.

One-To-Many and Many-To-One

We need the one-to-many relationship when we want a row from the first table to link to multiple rows in the second table. With this approach, the rows from the second table can be linked to just one row from the first table, though.

A great example is a post entity and an author. The user can be an author of multiple posts, but a particular post can only have one author. We need to use the ManyToOne() decorator to create a relationship like that.

post.entity.ts
1import { Entity, Property, PrimaryKey, ManyToOne } from '@mikro-orm/core';
2import User from '../users/user.entity';
3 
4@Entity()
5class PostEntity {
6  @PrimaryKey()
7  id: number;
8 
9  @Property()
10  title: string;
11 
12  @Property()
13  content: string;
14 
15  @ManyToOne()
16  author: User;
17}
18 
19export default PostEntity;

To define an author of a post, we need to make sure that we pass it to our PostsService.

posts.controller.ts
1import {
2  Body,
3  Controller,
4  Post,
5  UseGuards,
6  Req,
7} from '@nestjs/common';
8import { PostsService } from './posts.service';
9import CreatePostDto from './dto/createPost.dto';
10import JwtAuthenticationGuard from '../authentication/jwt-authentication.guard';
11import RequestWithUser from '../authentication/requestWithUser.interface';
12 
13@Controller('posts')
14export default class PostsController {
15  constructor(private readonly postsService: PostsService) {}
16  
17  @Post()
18  @UseGuards(JwtAuthenticationGuard)
19  async createPost(
20    @Body() post: CreatePostDto,
21    @Req() request: RequestWithUser,
22  ) {
23    return this.postsService.createPost(post, request.user);
24  }
25 
26  // ...
27}

We also need to create the data for the post entity properly and add the information about the author.

posts.service.ts
1import { Injectable } from '@nestjs/common';
2import { InjectRepository } from '@mikro-orm/nestjs';
3import { EntityRepository } from '@mikro-orm/core';
4import PostEntity from './post.entity';
5import CreatePostDto from './dto/createPost.dto';
6import User from '../users/user.entity';
7 
8@Injectable()
9export class PostsService {
10  constructor(
11    @InjectRepository(PostEntity)
12    private readonly postRepository: EntityRepository<PostEntity>,
13  ) {}
14 
15  async createPost(post: CreatePostDto, user: User) {
16    const postData = {
17      ...post,
18      author: user,
19    };
20    const newPost = await this.postRepository.create(postData);
21    await this.postRepository.persistAndFlush(newPost);
22    return newPost;
23  }
24 
25  // ...
26}

Thanks to the above, whenever an authenticated user creates a post, we store the information about the user.

The migration

Thanks to creating the above entities, MikroORM generates the following migration:

Migration20220529225632.ts
1import { Migration } from '@mikro-orm/migrations';
2 
3export class Migration20220529225632 extends Migration {
4 
5  async up(): Promise<void> {
6    this.addSql('alter table "post_entity" add column "author_id" int not null;');
7    this.addSql('alter table "post_entity" add constraint "post_entity_author_id_foreign" foreign key ("author_id") references "user" ("id") on update cascade;');
8  }
9 
10  async down(): Promise<void> {
11    this.addSql('alter table "post_entity" drop constraint "post_entity_author_id_foreign";');
12 
13    this.addSql('alter table "post_entity" drop column "author_id";');
14  }
15 
16}

Like before, MikroORM added the author_id column to the post_entity table that references the user.

It is crucial to notice that the above migration does not make the author_id unique in contrast to the one-to-one relationship. Thanks to that, a particular user can be the author of multiple posts.

Fetching the related entities

MikroORM, by default, does not query the details of the related entities.

We can alter this behavior by telling MikroORM explicitly to populate the additional information.

posts.service.ts
1import { Injectable } from '@nestjs/common';
2import { InjectRepository } from '@mikro-orm/nestjs';
3import { EntityRepository } from '@mikro-orm/core';
4import PostEntity from './post.entity';
5 
6@Injectable()
7export class PostsService {
8  constructor(
9    @InjectRepository(PostEntity)
10    private readonly postRepository: EntityRepository<PostEntity>,
11  ) {}
12 
13  getPosts() {
14    return this.postRepository.findAll({
15      populate: ['author'],
16    });
17  }
18 
19  // ...
20}

We can go even further and populate the nested properties too.

posts.service.ts
1import { Injectable } from '@nestjs/common';
2import { InjectRepository } from '@mikro-orm/nestjs';
3import { EntityRepository } from '@mikro-orm/core';
4import PostEntity from './post.entity';
5 
6@Injectable()
7export class PostsService {
8  constructor(
9    @InjectRepository(PostEntity)
10    private readonly postRepository: EntityRepository<PostEntity>,
11  ) {}
12 
13  getPosts() {
14    return this.postRepository.findAll({
15      populate: ['author', 'author.address'],
16    });
17  }
18  
19  // ...
20}

Fortunately, the populate property is built in a type-safe way, and TypeScript will ensure we don’t make a mistake here.

Many-To-Many

When we define a many-to-many relationship, a row from the first table can relate to multiple rows of the second table and the other way around.

A straightforward example is a post that can belong to multiple categories. On the other hand, a category can be associated with various posts.

category.entity.ts
1import { Entity, Property, PrimaryKey } from '@mikro-orm/core';
2 
3@Entity()
4class Category {
5  @PrimaryKey()
6  public id: number;
7 
8  @Property()
9  public name: string;
10}
11 
12export default Category;
post.entity.ts
1import {
2  Entity,
3  Property,
4  PrimaryKey,
5  ManyToOne,
6  Collection,
7  ManyToMany,
8} from '@mikro-orm/core';
9import User from '../users/user.entity';
10import Category from '../categories/category.entity';
11 
12@Entity()
13class PostEntity {
14  @PrimaryKey()
15  id: number;
16 
17  @Property()
18  title: string;
19 
20  @Property()
21  content: string;
22 
23  @ManyToOne()
24  author: User;
25 
26  @ManyToMany(() => Category)
27  categories: Collection<Category>;
28}
29 
30export default PostEntity;

MikroORM wraps the categories property in the Collection wrapper, which is not a regular array.

The easiest way to let the user assign categories to a post is to expect the POST request to contain an array of category ids.

createPost.dto.ts
1import { IsString, IsNotEmpty, IsNumber } from 'class-validator';
2 
3class CreatePostDto {
4  @IsString()
5  @IsNotEmpty()
6  title: string;
7 
8  @IsString()
9  @IsNotEmpty()
10  content: string;
11 
12  @IsNumber({}, { each: true })
13  categories: number[];
14}
15export default CreatePostDto;

Providing the categories array when calling the this.postRepository.create method assigns categories to the created post.

Fetching the related entities

To ensure that the above works properly, we can fetch information about the related categories.

posts.service.ts
1getPosts() {
2  return this.postRepository.findAll({
3    populate: ['author', 'author.address', 'categories'],
4  });
5}
With the above approach, we send quite a lot of redundant data. Instead, we might want avoid populating the relationship’s data and fetch it separately.

The migration

By creating entities like that and generating a migration, we end up with the following file:

Migration20220530004103.ts
1import { Migration } from '@mikro-orm/migrations';
2 
3export class Migration20220530004103 extends Migration {
4 
5  async up(): Promise<void> {
6    this.addSql('create table "category" ("id" serial primary key, "name" varchar(255) not null);');
7 
8    this.addSql('create table "post_entity_categories" ("post_entity_id" int not null, "category_id" int not null);');
9    this.addSql('alter table "post_entity_categories" add constraint "post_entity_categories_pkey" primary key ("post_entity_id", "category_id");');
10 
11    this.addSql('alter table "post_entity_categories" add constraint "post_entity_categories_post_entity_id_foreign" foreign key ("post_entity_id") references "post_entity" ("id") on update cascade on delete cascade;');
12    this.addSql('alter table "post_entity_categories" add constraint "post_entity_categories_category_id_foreign" foreign key ("category_id") references "category" ("id") on update cascade on delete cascade;');
13  }
14 
15  async down(): Promise<void> {
16    this.addSql('alter table "post_entity_categories" drop constraint "post_entity_categories_category_id_foreign";');
17 
18    this.addSql('drop table if exists "category" cascade;');
19 
20    this.addSql('drop table if exists "post_entity_categories" cascade;');
21  }
22 
23}

When we defined a many-to-many relationship between posts and categories, MikroORM created a post_entity_categories table. To store the information about a many-to-many relation, MikroORM uses the post_entity_id and category_id columns.

Bidirectional relationships

So far, we’ve been creating unidirectional relationships. It means that we’ve defined them only on one side of the relationship.

For example, we’ve used the @ManyToOne() decorator to assign an author to a post.

post.entity.ts
1import {
2  Entity,
3  PrimaryKey,
4  ManyToOne,
5} from '@mikro-orm/core';
6import User from '../users/user.entity';
7 
8@Entity()
9class PostEntity {
10  @PrimaryKey()
11  id: number;
12 
13  @ManyToOne()
14  author: User;
15  
16  // ...
17}
18 
19export default PostEntity;

Doing the above and creating a migration caused our post_entity table to have the author_id column.

Because the post entity contains the author_id, we call it the owning side of the relationship.

We could make the above relationship bidirectional by defining it on the user’s side, using the @OneToMany() decorator.

user.entity.ts
1import { Entity, PrimaryKey, OneToMany, Collection } from '@mikro-orm/core';
2import PostEntity from '../posts/post.entity';
3 
4@Entity()
5class User {
6  @PrimaryKey()
7  id: number;
8 
9  // ...
10 
11  @OneToMany(() => PostEntity, (post: PostEntity) => post.author)
12  posts: Collection<PostEntity>
13}
14 
15export default User;

The crucial thing about defining a bidirectional relationship is that it does not modify the structure of the database. For example, creating a bidirectional relationship does not cause the user table’s rows to contain the posts’ ids.

Thanks to having the posts property in the user’s entity, we can easily populate it.

users.service.ts
1import { Injectable } from '@nestjs/common';
2import { UserNotFoundException } from './exceptions/userNotFound.exception';
3import { InjectRepository } from '@mikro-orm/nestjs';
4import { EntityRepository } from '@mikro-orm/core';
5import User from './user.entity';
6 
7@Injectable()
8class UsersService {
9  constructor(
10    @InjectRepository(User)
11    private readonly userRepository: EntityRepository<User>,
12  ) {}
13 
14  async getByEmail(email: string) {
15    const user = await this.userRepository.findOne(
16      {
17        email,
18      },
19      {
20        populate: ['address', 'posts'],
21      },
22    );
23    if (!user) {
24      throw new UserNotFoundException();
25    }
26    return user;
27  }
28 
29  // ...
30}
31 
32export default UsersService;
We can also create bidirectional relationships using One-To-One and Many-To-Many.

The potential issues with bidirectional relationships

While the above solution gives us an effortless way to fetch additional data, it comes with a cost. Even though it looks straightforward, it contains quite a bit of implicit complexity.

When we fetch the user with a list of posts, MikroORM has to go through all posts in the posts table to find matching entities. So it might hurt our performance at some point if we use this feature carelessly.

Also, in our case, to achieve the bidirectional relationship, the users.entity.ts file imports the post.entity.ts file and vice versa. The above creates a circular dependency, which is usually frowned upon. If you want to know more about circular dependencies and what issues they might cause, check out API with NestJS #61. Dealing with circular dependencies.

Summary

We’ve gone through all the relationship types in this article and implemented them with MikroORM and PostgreSQL. Besides regular relationships, we’ve also defined a bidirectional relationship. As a result, we’ve learned quite a bit about how MikroORM works and how it affects our database structure. Thanks to that, we can better grasp how our application communicates with the database and how it can affect its performance.