Create an Entity
In this document, you’ll learn how you can create an Entity.
Create the Entity
To create an entity, create a TypeScript file in src/models
. For example, here’s a Post
entity defined in the file src/models/post.ts
:
import { BeforeInsert, Column, Entity, PrimaryColumn } from "typeorm"
import { BaseEntity } from "@medusajs/medusa"
import { generateEntityId } from "@medusajs/medusa/dist/utils"
@Entity()
export class Post extends BaseEntity {
@Column({ type: "varchar" })
title: string | null
@BeforeInsert()
private beforeInsert(): void {
this.id = generateEntityId(this.id, "post")
}
}
This entity has one column title
defined. However, since it extends BaseEntity
it will also have the id
, created_at
, and updated_at
columns.
Medusa’s core entities all have the following format for IDs: <PREFIX>_<RANDOM>
. For example, an order might have the ID order_01G35WVGY4D1JCA4TPGVXPGCQM
.
To generate an ID for your entity that matches the IDs generated for Medusa’s core entities, you should add a BeforeInsert
event handler. Then, inside that handler use Medusa’s utility function generateEntityId
to generate the ID. It accepts the ID as a first parameter and the prefix as a second parameter. The Post
entity IDs will be of the format post_<RANDOM>
.
If you want the entity to also be soft deletable then it should extend SoftDeletableEntity
instead:
import { SoftDeletableEntity } from "@medusajs/medusa"
@Entity()
export class Post extends SoftDeletableEntity {
// ...
}
You can learn more about what decorators and column types you can use in Typeorm’s documentation.
Create a Migration
Additionally, you must create a migration for your entity. Migrations are used to update the database schema with new tables or changes to existing tables.
You can learn more about Migrations, how to create them, and how to run them in the Migration documentation.
Create a Repository
Entities data can be easily accessed and modified using Typeorm Repositories. To create a repository, create a file in src/repositories
. For example, here’s a repository PostRepository
created in src/repositories/post.ts
:
import { EntityRepository, Repository } from "typeorm"
import { Post } from "../models/post"
@EntityRepository(Post)
export class PostRepository extends Repository<Post> { }
This repository is created for the Post
and that is indicated using the decorator @EntityRepository
.
Be careful with your file names as it can cause unclear errors in Typeorm. Make sure all your file names are small letters for both entities and repositories to avoid any issues with file names.
Access a Custom Entity
Before trying this step make sure that you’ve created and run your migrations. You also need to re-build your code using:
- npm
- Yarn
npm run build
yarn run build
You can access your custom entity data in the database in services or subscribers using the repository. For example, here’s a service that lists all posts:
import { TransactionBaseService } from "@medusajs/medusa"
class PostService extends TransactionBaseService {
constructor({ postRepository, manager }) {
super({ postRepository, manager })
this.postRepository = postRepository
this.manager_ = manager
}
async list() {
const postRepository = this.manager_
.getCustomRepository(this.postRepository)
return await postRepository.find()
}
}
export default PostService
In the constructor, you can use dependency injection to get access to instances of services and repositories. Here, you initialize class fields postRepository
and manager
. The manager
is a Typeorm Entity Manager.
Then, in the method list
, you can obtain an instance of the PostRepository
using this.manager_.getCustomRepository
passing it this.postRepository
as a parameter. This lets you use Custom Repositories with Typeorm to create custom methods in your repository that work with the data in your database.
After getting an instance of the repository, you can then use Typeorm’s Repository methods to perform Create, Read, Update, and Delete (CRUD) operations on your entity.
If you need access to your entity in endpoints, you can then use the methods you define in the service.
This same usage of repositories can be done in subscribers as well.
Delete a Soft-Deletable Entity
To delete soft-deletable entities that extend the SoftDeletableEntity
class, you can use the repository method softDelete
method:
await postRepository.softDelete(post.id)