refactor: get tips api uses TipRepository

payw-org · Jun 18, 2020 · 3cc88c6 · 3cc88c6
1 parent 9050acf
commit 3cc88c6
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 25 deletions.
diff --git a/src/api/one/scheme/tip/get-tips.action/function.ts b/src/api/one/scheme/tip/get-tips.action/function.ts
@@ -2,6 +2,7 @@ import { TipAttrs, TipListResponse } from '@/database/models/tip'
 
 import { Action } from './interface'
 import { OneApiFunc } from '@/api/one/types'
+import { TipRepository } from '@/database/repository/tip-repository'
 import { oneApiResponse } from '@/api/one/utils'
 import prisma from '@/modules/prisma'
 
@@ -12,32 +13,9 @@ const func: OneApiFunc<Action> = async (data) => {
 
   const totalCount = await prisma.tip.count()
   if (topic === null) {
-    tipList = await prisma.tip.findMany({
-      where: {
-        isRemoved: false,
-      },
-      include: {
-        tipLikes: true,
-        tipBookmarks: true,
-      },
-      take: -pageSize,
-      skip: (page - 1) * pageSize,
-      orderBy: { createdAt: 'desc' },
-    })
+    tipList = await TipRepository.findAll(pageSize, page)
   } else {
-    tipList = await prisma.tip.findMany({
-      where: {
-        topic: topic,
-        isRemoved: false,
-      },
-      include: {
-        tipLikes: true,
-        tipBookmarks: true,
-      },
-      take: -pageSize,
-      skip: (page - 1) * pageSize,
-      orderBy: { createdAt: 'desc' },
-    })
+    tipList = await TipRepository.findByTopic(topic, pageSize, page)
   }
 
   const tips = tipList.map((item) => {

diff --git a/src/database/repository/tip-repository.ts b/src/database/repository/tip-repository.ts
@@ -1,5 +1,6 @@
 import { TipAttrs, TipResponse } from '../models/tip'
 
+import { TipTopic } from '@prisma/client'
 import prisma from '@/modules/prisma'
 
 export class TipRepository {
@@ -15,4 +16,41 @@ export class TipRepository {
     })
     return tip
   }
+
+  static async findByTopic(
+    topic: TipTopic,
+    pageSize: number,
+    page: number
+  ): Promise<TipAttrs[]> {
+    const tipList = await prisma.tip.findMany({
+      where: {
+        isRemoved: false,
+        topic: topic,
+      },
+      include: {
+        tipLikes: true,
+        tipBookmarks: true,
+      },
+      take: -pageSize,
+      skip: (page - 1) * pageSize,
+      orderBy: { createdAt: 'desc' },
+    })
+    return tipList
+  }
+
+  static async findAll(pageSize: number, page: number): Promise<TipAttrs[]> {
+    const tipList = await prisma.tip.findMany({
+      where: {
+        isRemoved: false,
+      },
+      include: {
+        tipLikes: true,
+        tipBookmarks: true,
+      },
+      take: -pageSize,
+      skip: (page - 1) * pageSize,
+      orderBy: { createdAt: 'desc' },
+    })
+    return tipList
+  }
 }