yujunjun/strapi
1
0
Fork 0
mirror of https://github.com/strapi/strapi.git synced 2025-09-25 16:29:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

split code into files

Browse Source
This commit is contained in:
Pierre Noël 2021-03-02 17:09:35 +01:00
parent a662db9691
commit 54512c6049
4 changed files with 112 additions and 97 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
packages/strapi-plugin-i18n/config/functions/migrations/field/index.js Normal file
View File

@ -0,0 +1,42 @@
'use strict';
const { difference, orderBy, intersection } = require('lodash/fp');
const { getService } = require('../../../../utils');
const migrateForMongoose = require('./migrateForMongoose');
const migrateForBookshelf = require('./migrateForBookshelf');
// Migration when i18n is disabled on a field of a content-type that have i18n enabled
const after = async ({ model, definition, previousDefinition, ORM }) => {
const ctService = getService('content-types');
const localeService = getService('locales');
if (!ctService.isLocalized(model)) {
return;
}
const localizedAttributes = ctService.getLocalizedFields(definition);
const prevLocalizedAttributes = ctService.getLocalizedFields(previousDefinition);
const attributesDisabled = difference(prevLocalizedAttributes, localizedAttributes);
const attributesToMigrate = intersection(Object.keys(definition.attributes), attributesDisabled);
if (attributesToMigrate.length === 0) {
return;
}
let locales = await localeService.find();
locales = await localeService.setIsDefault(locales);
locales = orderBy(['isDefault', 'code'], ['desc', 'asc'])(locales); // Put default locale first
if (model.orm === 'bookshelf') {
await migrateForBookshelf({ ORM, model, attributesToMigrate, locales });
} else if (model.orm === 'mongoose') {
await migrateForMongoose({ model, attributesToMigrate, locales });
}
};
const before = () => {};
module.exports = {
before,
after,
};

99
packages/strapi-plugin-i18n/config/functions/migrations/field.js → packages/strapi-plugin-i18n/config/functions/migrations/field/migrateForBookshelf.js
View File

@ -1,33 +1,9 @@
'use strict';
const { difference, pick, orderBy, prop, intersection } = require('lodash/fp');
const { getService } = require('../../../utils');
const { shouldBeProcesseed, getUpdatesInfo } = require('./utils');
const BATCH_SIZE = 1000;
// Common functions
const shouldBeProcesseed = processedLocaleCodes => entry => {
return (
entry.localizations.length > 1 &&
intersection(entry.localizations.map(prop('locale')), processedLocaleCodes).length === 0
);
};
const getUpdatesInfo = ({ entriesToProcess, locale, attributesToMigrate }) => {
const updates = [];
for (const entry of entriesToProcess) {
const attributesValues = pick(attributesToMigrate, entry);
const entriesIdsToUpdate = entry.localizations
.filter(related => related.locale !== locale.code)
.map(prop('id'));
updates.push({ entriesIdsToUpdate, attributesValues });
}
return updates;
};
// Bookshelf
const TMP_TABLE_NAME = '__tmp__i18n_field_migration';
const batchInsertInTmpTable = async (updatesInfo, trx) => {
@ -146,75 +122,4 @@ const migrateForBookshelf = async ({ ORM, model, attributesToMigrate, locales })
}
};
// Mongoose
const migrateForMongoose = async ({ model, attributesToMigrate, locales }) => {
const processedLocaleCodes = [];
for (const locale of locales) {
let batchCount = BATCH_SIZE;
let lastId;
while (batchCount === BATCH_SIZE) {
const findParams = { locale: locale.code };
if (lastId) {
findParams._id = { $gt: lastId };
}
const batch = await model
.find(findParams, [...attributesToMigrate, 'locale', 'localizations'])
.sort({ _id: 1 })
.limit(BATCH_SIZE);
if (batch.length > 0) {
lastId = batch[batch.length - 1]._id;
}
batchCount = batch.length;
const entriesToProcess = batch.filter(shouldBeProcesseed);
const updatesInfo = getUpdatesInfo({ entriesToProcess, locale, attributesToMigrate });
const updates = updatesInfo.map(({ entriesIdsToUpdate, attributesValues }) => ({
updateMany: { filter: { _id: { $in: entriesIdsToUpdate } }, update: attributesValues },
}));
await model.bulkWrite(updates);
}
processedLocaleCodes.push(locale.code);
}
};
// Migration when i18n is disabled on a field of a content-type that have i18n enabled
const after = async ({ model, definition, previousDefinition, ORM }) => {
const ctService = getService('content-types');
const localeService = getService('locales');
if (!ctService.isLocalized(model)) {
return;
}
const localizedAttributes = ctService.getLocalizedAttributes(definition);
const prevLocalizedAttributes = ctService.getLocalizedAttributes(previousDefinition);
const attributesDisabled = difference(prevLocalizedAttributes, localizedAttributes);
const attributesToMigrate = intersection(Object.keys(definition.attributes), attributesDisabled);
if (attributesToMigrate.length === 0) {
return;
}
let locales = await localeService.find();
locales = await localeService.setIsDefault(locales);
locales = orderBy(['isDefault', 'code'], ['desc', 'asc'])(locales); // Put default locale first
if (model.orm === 'bookshelf') {
await migrateForBookshelf({ ORM, model, attributesToMigrate, locales });
} else if (model.orm === 'mongoose') {
await migrateForMongoose({ model, attributesToMigrate, locales });
}
throw new Error('Done');
};
const before = () => {};
module.exports = {
before,
after,
};
module.exports = migrateForBookshelf;

41
packages/strapi-plugin-i18n/config/functions/migrations/field/migrateForMongoose.js Normal file
View File

@ -0,0 +1,41 @@
'use strict';
const { shouldBeProcesseed, getUpdatesInfo } = require('./utils');
const BATCH_SIZE = 1000;
const migrateForMongoose = async ({ model, attributesToMigrate, locales }) => {
const processedLocaleCodes = [];
for (const locale of locales) {
let batchCount = BATCH_SIZE;
let lastId;
while (batchCount === BATCH_SIZE) {
const findParams = { locale: locale.code };
if (lastId) {
findParams._id = { $gt: lastId };
}
const batch = await model
.find(findParams, [...attributesToMigrate, 'locale', 'localizations'])
.sort({ _id: 1 })
.limit(BATCH_SIZE);
if (batch.length > 0) {
lastId = batch[batch.length - 1]._id;
}
batchCount = batch.length;
const entriesToProcess = batch.filter(shouldBeProcesseed);
const updatesInfo = getUpdatesInfo({ entriesToProcess, locale, attributesToMigrate });
const updates = updatesInfo.map(({ entriesIdsToUpdate, attributesValues }) => ({
updateMany: { filter: { _id: { $in: entriesIdsToUpdate } }, update: attributesValues },
}));
await model.bulkWrite(updates);
}
processedLocaleCodes.push(locale.code);
}
};
module.exports = migrateForMongoose;

27
packages/strapi-plugin-i18n/config/functions/migrations/field/utils.js Normal file
View File

@ -0,0 +1,27 @@
'use strict';
const { pick, prop, intersection } = require('lodash/fp');
const shouldBeProcesseed = processedLocaleCodes => entry => {
return (
entry.localizations.length > 1 &&
intersection(entry.localizations.map(prop('locale')), processedLocaleCodes).length === 0
);
};
const getUpdatesInfo = ({ entriesToProcess, locale, attributesToMigrate }) => {
const updates = [];
for (const entry of entriesToProcess) {
const attributesValues = pick(attributesToMigrate, entry);
const entriesIdsToUpdate = entry.localizations
.filter(related => related.locale !== locale.code)
.map(prop('id'));
updates.push({ entriesIdsToUpdate, attributesValues });
}
return updates;
};
module.exports = {
shouldBeProcesseed,
getUpdatesInfo,
};