Files
kami-parse-server/src/SchemaMigrations/DefinedSchemas.js
Daniel 99fcf45e55 feat: Asynchronous initialization of Parse Server (#8232)
BREAKING CHANGE: This release introduces the asynchronous initialization of Parse Server to prevent mounting Parse Server before being ready to receive request; it changes how Parse Server is imported, initialized and started; it also removes the callback `serverStartComplete`; see the [Parse Server 6 migration guide](https://github.com/parse-community/parse-server/blob/alpha/6.0.0.md) for more details (#8232)
2022-12-21 15:30:13 +01:00

445 lines
15 KiB
JavaScript

// @flow
// @flow-disable-next Cannot resolve module `parse/node`.
const Parse = require('parse/node');
import { logger } from '../logger';
import Config from '../Config';
import { internalCreateSchema, internalUpdateSchema } from '../Routers/SchemasRouter';
import { defaultColumns, systemClasses } from '../Controllers/SchemaController';
import { ParseServerOptions } from '../Options';
import * as Migrations from './Migrations';
import Auth from '../Auth';
import rest from '../rest';
export class DefinedSchemas {
config: ParseServerOptions;
schemaOptions: Migrations.SchemaOptions;
localSchemas: Migrations.JSONSchema[];
retries: number;
maxRetries: number;
allCloudSchemas: Parse.Schema[];
constructor(schemaOptions: Migrations.SchemaOptions, config: ParseServerOptions) {
this.localSchemas = [];
this.config = Config.get(config.appId);
this.schemaOptions = schemaOptions;
if (schemaOptions && schemaOptions.definitions) {
if (!Array.isArray(schemaOptions.definitions)) {
throw `"schema.definitions" must be an array of schemas`;
}
this.localSchemas = schemaOptions.definitions;
}
this.retries = 0;
this.maxRetries = 3;
}
async saveSchemaToDB(schema: Parse.Schema): Promise<void> {
const payload = {
className: schema.className,
fields: schema._fields,
indexes: schema._indexes,
classLevelPermissions: schema._clp,
};
await internalCreateSchema(schema.className, payload, this.config);
this.resetSchemaOps(schema);
}
resetSchemaOps(schema: Parse.Schema) {
// Reset ops like SDK
schema._fields = {};
schema._indexes = {};
}
// Simulate update like the SDK
// We cannot use SDK since routes are disabled
async updateSchemaToDB(schema: Parse.Schema) {
const payload = {
className: schema.className,
fields: schema._fields,
indexes: schema._indexes,
classLevelPermissions: schema._clp,
};
await internalUpdateSchema(schema.className, payload, this.config);
this.resetSchemaOps(schema);
}
async execute() {
try {
logger.info('Running Migrations');
if (this.schemaOptions && this.schemaOptions.beforeMigration) {
await Promise.resolve(this.schemaOptions.beforeMigration());
}
await this.executeMigrations();
if (this.schemaOptions && this.schemaOptions.afterMigration) {
await Promise.resolve(this.schemaOptions.afterMigration());
}
logger.info('Running Migrations Completed');
} catch (e) {
logger.error(`Failed to run migrations: ${e}`);
if (process.env.NODE_ENV === 'production') process.exit(1);
}
}
async executeMigrations() {
let timeout = null;
try {
// Set up a time out in production
// if we fail to get schema
// pm2 or K8s and many other process managers will try to restart the process
// after the exit
if (process.env.NODE_ENV === 'production') {
timeout = setTimeout(() => {
logger.error('Timeout occurred during execution of migrations. Exiting...');
process.exit(1);
}, 20000);
}
await this.createDeleteSession();
// @flow-disable-next-line
const schemaController = await this.config.database.loadSchema();
this.allCloudSchemas = await schemaController.getAllClasses();
clearTimeout(timeout);
await Promise.all(this.localSchemas.map(async localSchema => this.saveOrUpdate(localSchema)));
this.checkForMissingSchemas();
await this.enforceCLPForNonProvidedClass();
} catch (e) {
if (timeout) clearTimeout(timeout);
if (this.retries < this.maxRetries) {
this.retries++;
// first retry 1sec, 2sec, 3sec total 6sec retry sequence
// retry will only happen in case of deploying multi parse server instance
// at the same time. Modern systems like k8 avoid this by doing rolling updates
await this.wait(1000 * this.retries);
await this.executeMigrations();
} else {
logger.error(`Failed to run migrations: ${e}`);
if (process.env.NODE_ENV === 'production') process.exit(1);
}
}
}
checkForMissingSchemas() {
if (this.schemaOptions.strict !== true) {
return;
}
const cloudSchemas = this.allCloudSchemas.map(s => s.className);
const localSchemas = this.localSchemas.map(s => s.className);
const missingSchemas = cloudSchemas.filter(
c => !localSchemas.includes(c) && !systemClasses.includes(c)
);
if (new Set(localSchemas).size !== localSchemas.length) {
logger.error(
`The list of schemas provided contains duplicated "className" "${localSchemas.join(
'","'
)}"`
);
process.exit(1);
}
if (this.schemaOptions.strict && missingSchemas.length) {
logger.warn(
`The following schemas are currently present in the database, but not explicitly defined in a schema: "${missingSchemas.join(
'", "'
)}"`
);
}
}
// Required for testing purpose
wait(time: number) {
return new Promise<void>(resolve => setTimeout(resolve, time));
}
async enforceCLPForNonProvidedClass(): Promise<void> {
const nonProvidedClasses = this.allCloudSchemas.filter(
cloudSchema =>
!this.localSchemas.some(localSchema => localSchema.className === cloudSchema.className)
);
await Promise.all(
nonProvidedClasses.map(async schema => {
const parseSchema = new Parse.Schema(schema.className);
this.handleCLP(schema, parseSchema);
await this.updateSchemaToDB(parseSchema);
})
);
}
// Create a fake session since Parse do not create the _Session until
// a session is created
async createDeleteSession() {
const { response } = await rest.create(this.config, Auth.master(this.config), '_Session', {});
await rest.del(this.config, Auth.master(this.config), '_Session', response.objectId);
}
async saveOrUpdate(localSchema: Migrations.JSONSchema) {
const cloudSchema = this.allCloudSchemas.find(sc => sc.className === localSchema.className);
if (cloudSchema) {
try {
await this.updateSchema(localSchema, cloudSchema);
} catch (e) {
throw `Error during update of schema for type ${cloudSchema.className}: ${e}`;
}
} else {
try {
await this.saveSchema(localSchema);
} catch (e) {
throw `Error while saving Schema for type ${localSchema.className}: ${e}`;
}
}
}
async saveSchema(localSchema: Migrations.JSONSchema) {
const newLocalSchema = new Parse.Schema(localSchema.className);
if (localSchema.fields) {
// Handle fields
Object.keys(localSchema.fields)
.filter(fieldName => !this.isProtectedFields(localSchema.className, fieldName))
.forEach(fieldName => {
if (localSchema.fields) {
const field = localSchema.fields[fieldName];
this.handleFields(newLocalSchema, fieldName, field);
}
});
}
// Handle indexes
if (localSchema.indexes) {
Object.keys(localSchema.indexes).forEach(indexName => {
if (localSchema.indexes && !this.isProtectedIndex(localSchema.className, indexName)) {
newLocalSchema.addIndex(indexName, localSchema.indexes[indexName]);
}
});
}
this.handleCLP(localSchema, newLocalSchema);
return await this.saveSchemaToDB(newLocalSchema);
}
async updateSchema(localSchema: Migrations.JSONSchema, cloudSchema: Parse.Schema) {
const newLocalSchema = new Parse.Schema(localSchema.className);
// Handle fields
// Check addition
if (localSchema.fields) {
Object.keys(localSchema.fields)
.filter(fieldName => !this.isProtectedFields(localSchema.className, fieldName))
.forEach(fieldName => {
// @flow-disable-next
const field = localSchema.fields[fieldName];
if (!cloudSchema.fields[fieldName]) {
this.handleFields(newLocalSchema, fieldName, field);
}
});
}
const fieldsToDelete: string[] = [];
const fieldsToRecreate: {
fieldName: string,
from: { type: string, targetClass?: string },
to: { type: string, targetClass?: string },
}[] = [];
const fieldsWithChangedParams: string[] = [];
// Check deletion
Object.keys(cloudSchema.fields)
.filter(fieldName => !this.isProtectedFields(localSchema.className, fieldName))
.forEach(fieldName => {
const field = cloudSchema.fields[fieldName];
if (!localSchema.fields || !localSchema.fields[fieldName]) {
fieldsToDelete.push(fieldName);
return;
}
const localField = localSchema.fields[fieldName];
// Check if field has a changed type
if (
!this.paramsAreEquals(
{ type: field.type, targetClass: field.targetClass },
{ type: localField.type, targetClass: localField.targetClass }
)
) {
fieldsToRecreate.push({
fieldName,
from: { type: field.type, targetClass: field.targetClass },
to: { type: localField.type, targetClass: localField.targetClass },
});
return;
}
// Check if something changed other than the type (like required, defaultValue)
if (!this.paramsAreEquals(field, localField)) {
fieldsWithChangedParams.push(fieldName);
}
});
if (this.schemaOptions.deleteExtraFields === true) {
fieldsToDelete.forEach(fieldName => {
newLocalSchema.deleteField(fieldName);
});
// Delete fields from the schema then apply changes
await this.updateSchemaToDB(newLocalSchema);
} else if (this.schemaOptions.strict === true && fieldsToDelete.length) {
logger.warn(
`The following fields exist in the database for "${
localSchema.className
}", but are missing in the schema : "${fieldsToDelete.join('" ,"')}"`
);
}
if (this.schemaOptions.recreateModifiedFields === true) {
fieldsToRecreate.forEach(field => {
newLocalSchema.deleteField(field.fieldName);
});
// Delete fields from the schema then apply changes
await this.updateSchemaToDB(newLocalSchema);
fieldsToRecreate.forEach(fieldInfo => {
if (localSchema.fields) {
const field = localSchema.fields[fieldInfo.fieldName];
this.handleFields(newLocalSchema, fieldInfo.fieldName, field);
}
});
} else if (this.schemaOptions.strict === true && fieldsToRecreate.length) {
fieldsToRecreate.forEach(field => {
const from =
field.from.type + (field.from.targetClass ? ` (${field.from.targetClass})` : '');
const to = field.to.type + (field.to.targetClass ? ` (${field.to.targetClass})` : '');
logger.warn(
`The field "${field.fieldName}" type differ between the schema and the database for "${localSchema.className}"; Schema is defined as "${to}" and current database type is "${from}"`
);
});
}
fieldsWithChangedParams.forEach(fieldName => {
if (localSchema.fields) {
const field = localSchema.fields[fieldName];
this.handleFields(newLocalSchema, fieldName, field);
}
});
// Handle Indexes
// Check addition
if (localSchema.indexes) {
Object.keys(localSchema.indexes).forEach(indexName => {
if (
(!cloudSchema.indexes || !cloudSchema.indexes[indexName]) &&
!this.isProtectedIndex(localSchema.className, indexName)
) {
if (localSchema.indexes) {
newLocalSchema.addIndex(indexName, localSchema.indexes[indexName]);
}
}
});
}
const indexesToAdd = [];
// Check deletion
if (cloudSchema.indexes) {
Object.keys(cloudSchema.indexes).forEach(indexName => {
if (!this.isProtectedIndex(localSchema.className, indexName)) {
if (!localSchema.indexes || !localSchema.indexes[indexName]) {
newLocalSchema.deleteIndex(indexName);
} else if (
!this.paramsAreEquals(localSchema.indexes[indexName], cloudSchema.indexes[indexName])
) {
newLocalSchema.deleteIndex(indexName);
if (localSchema.indexes) {
indexesToAdd.push({
indexName,
index: localSchema.indexes[indexName],
});
}
}
}
});
}
this.handleCLP(localSchema, newLocalSchema, cloudSchema);
// Apply changes
await this.updateSchemaToDB(newLocalSchema);
// Apply new/changed indexes
if (indexesToAdd.length) {
logger.debug(
`Updating indexes for "${newLocalSchema.className}" : ${indexesToAdd.join(' ,')}`
);
indexesToAdd.forEach(o => newLocalSchema.addIndex(o.indexName, o.index));
await this.updateSchemaToDB(newLocalSchema);
}
}
handleCLP(
localSchema: Migrations.JSONSchema,
newLocalSchema: Parse.Schema,
cloudSchema: Parse.Schema
) {
if (!localSchema.classLevelPermissions && !cloudSchema) {
logger.warn(`classLevelPermissions not provided for ${localSchema.className}.`);
}
// Use spread to avoid read only issue (encountered by Moumouls using directAccess)
const clp = ({ ...localSchema.classLevelPermissions } || {}: Parse.CLP.PermissionsMap);
// To avoid inconsistency we need to remove all rights on addField
clp.addField = {};
newLocalSchema.setCLP(clp);
}
isProtectedFields(className: string, fieldName: string) {
return (
!!defaultColumns._Default[fieldName] ||
!!(defaultColumns[className] && defaultColumns[className][fieldName])
);
}
isProtectedIndex(className: string, indexName: string) {
const indexes = ['_id_'];
switch (className) {
case '_User':
indexes.push(
'case_insensitive_username',
'case_insensitive_email',
'username_1',
'email_1'
);
break;
case '_Role':
indexes.push('name_1');
break;
case '_Idempotency':
indexes.push('reqId_1');
break;
}
return indexes.indexOf(indexName) !== -1;
}
paramsAreEquals<T: { [key: string]: any }>(objA: T, objB: T) {
const keysA: string[] = Object.keys(objA);
const keysB: string[] = Object.keys(objB);
// Check key name
if (keysA.length !== keysB.length) return false;
return keysA.every(k => objA[k] === objB[k]);
}
handleFields(newLocalSchema: Parse.Schema, fieldName: string, field: Migrations.FieldType) {
if (field.type === 'Relation') {
newLocalSchema.addRelation(fieldName, field.targetClass);
} else if (field.type === 'Pointer') {
newLocalSchema.addPointer(fieldName, field.targetClass, field);
} else {
newLocalSchema.addField(fieldName, field.type, field);
}
}
}