diff --git a/BE/src/asset/asset.entity.ts b/BE/src/asset/asset.entity.ts index 2e010f4..eab36b9 100644 --- a/BE/src/asset/asset.entity.ts +++ b/BE/src/asset/asset.entity.ts @@ -1,8 +1,9 @@ -import { Column, Entity, PrimaryGeneratedColumn } from 'typeorm'; +import { Column, Entity, PrimaryGeneratedColumn, Unique } from 'typeorm'; const INIT_ASSET = 10000000; @Entity('assets') +@Unique(['user_id']) export class Asset { @PrimaryGeneratedColumn() id: number; diff --git a/BE/src/stock/bookmark/stock-bookmark.entity.ts b/BE/src/stock/bookmark/stock-bookmark.entity.ts index 107a957..9e04e5c 100644 --- a/BE/src/stock/bookmark/stock-bookmark.entity.ts +++ b/BE/src/stock/bookmark/stock-bookmark.entity.ts @@ -1,6 +1,7 @@ -import { Column, Entity, PrimaryGeneratedColumn } from 'typeorm'; +import { Column, Entity, PrimaryGeneratedColumn, Unique } from 'typeorm'; @Entity('bookmarks') +@Unique(['user_id', 'stock_code']) export class Bookmark { @PrimaryGeneratedColumn() id: number; diff --git a/BE/src/stock/order/stock-order.entity.ts b/BE/src/stock/order/stock-order.entity.ts index 7498b72..1c84d7b 100644 --- a/BE/src/stock/order/stock-order.entity.ts +++ b/BE/src/stock/order/stock-order.entity.ts @@ -2,19 +2,23 @@ import { Column, CreateDateColumn, Entity, + Index, PrimaryGeneratedColumn, } from 'typeorm'; import { TradeType } from './enum/trade-type'; import { StatusType } from './enum/status-type'; @Entity('orders') +@Index(['user_id', 'stock_code']) export class Order { @PrimaryGeneratedColumn() id: number; + @Index() @Column({ nullable: false }) user_id: number; + @Index() @Column({ nullable: false }) stock_code: string;