feat: units pagination

Chia-Network · Dec 23, 2021 · 9132891 · 9132891
1 parent 679c197
commit 9132891
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/controllers/units.controller.js b/src/controllers/units.controller.js
@@ -1,5 +1,8 @@
+'use strict';
+
 import { uuid as uuidv4 } from 'uuidv4';
 import { Staging, UnitMock, Unit, Qualification, Vintage } from '../models';
+import { paginationParams } from './helpers';
 
 export const create = async (req, res) => {
   try {
@@ -26,8 +29,10 @@ export const create = async (req, res) => {
 };
 
 export const findAll = async (req, res) => {
+  const { page, limit } = req.query;
+
   if (req.query.useMock) {
-    res.json(UnitMock.findAll());
+    res.json(UnitMock.findAll({ ...paginationParams(page, limit) }));
     return;
   }
 
@@ -40,6 +45,7 @@ export const findAll = async (req, res) => {
         },
         Vintage,
       ],
+      ...paginationParams(page, limit),
     }),
   );
 };

diff --git a/src/models/units/units.mock.js b/src/models/units/units.mock.js
@@ -1,7 +1,7 @@
 import stub from './units.stub.json';
 
 export const UnitMock = {
-  findAll: () => stub,
+  findAll: ({ limit, offset }) => stub.slice(offset * limit, (offset + 1) * limit),
   findOne: (id) => {
     return stub.find((record) => record.id == id);
   },