strapi/packages/core/database/lib/entity-repository.js
Alexandre Bodin 80f38005f1 Fix
2021-07-05 21:20:24 +02:00

96 lines
1.9 KiB
JavaScript

'use strict';
const withDefaultPagination = params => {
const { page = 1, pageSize = 100, ...rest } = params;
return {
page: parseInt(page),
pageSize: parseInt(pageSize),
...rest,
};
};
const createRepository = (uid, db) => {
return {
findOne(params) {
return db.entityManager.findOne(uid, params);
},
findMany(params) {
return db.entityManager.findMany(uid, params);
},
findWithCount(params) {
return Promise.all([
db.entityManager.findMany(uid, params),
db.entityManager.count(uid, params),
]);
},
async findPage(params) {
const { page, pageSize, ...rest } = withDefaultPagination(params);
const offset = Math.max(page - 1, 0) * pageSize;
const limit = pageSize;
const query = {
...rest,
limit,
offset,
};
const [results, total] = await Promise.all([
db.entityManager.findMany(uid, query),
db.entityManager.count(uid, query),
]);
return {
results,
pagination: {
page,
pageSize,
pageCount: Math.ceil(total / pageSize),
total,
},
};
},
create(params) {
return db.entityManager.create(uid, params);
},
createMany(params) {
return db.entityManager.createMany(uid, params);
},
update(params) {
return db.entityManager.update(uid, params);
},
updateMany(params) {
return db.entityManager.updateMany(uid, params);
},
delete(params) {
return db.entityManager.delete(uid, params);
},
deleteMany(params) {
return db.entityManager.deleteMany(uid, params);
},
count(params) {
return db.entityManager.count(uid, params);
},
// TODO: add relation API
populate() {},
load() {},
};
};
module.exports = {
createRepository,
};