Support pointer in aggregate query (#4493)

This commit is contained in:
Diamond Lewis
2018-01-08 20:42:08 -06:00
committed by GitHub
parent cb8f038ca8
commit 8ec07b83d0
3 changed files with 45 additions and 10 deletions

View File

@@ -515,12 +515,26 @@ export class MongoStorageAdapter implements StorageAdapter {
}
aggregate(className: string, schema: any, pipeline: any, readPreference: ?string) {
let isPointerField = false;
pipeline = pipeline.map((stage) => {
if (stage.$group && stage.$group._id) {
const field = stage.$group._id.substring(1);
if (schema.fields[field] && schema.fields[field].type === 'Pointer') {
isPointerField = true;
stage.$group._id = `$_p_${field}`;
}
}
return stage;
});
readPreference = this._parseReadPreference(readPreference);
return this._adaptiveCollection(className)
.then(collection => collection.aggregate(pipeline, { readPreference, maxTimeMS: this._maxTimeMS }))
.then(results => {
results.forEach(result => {
if (result.hasOwnProperty('_id')) {
if (isPointerField && result._id) {
result._id = result._id.split('$')[1];
}
result.objectId = result._id;
delete result._id;
}

View File

@@ -754,13 +754,13 @@ export class PostgresStorageAdapter implements StorageAdapter {
debug('schemaUpgrade', { className, schema });
conn = conn || this._client;
const self = this;
return conn.tx('schema-upgrade', function * (t) {
const columns = yield t.map('SELECT column_name FROM information_schema.columns WHERE table_name = $<className>', { className }, a => a.column_name);
const newColumns = Object.keys(schema.fields)
.filter(item => columns.indexOf(item) === -1)
.map(fieldName => self.addFieldIfNotExists(className, fieldName, schema.fields[fieldName], t));
yield t.batch(newColumns);
});
}
@@ -896,7 +896,7 @@ export class PostgresStorageAdapter implements StorageAdapter {
return this._client.any('SELECT * FROM "_SCHEMA" WHERE "className"=$<className>', { className })
.then(result => {
if (result.length !== 1) {
throw undefined;
throw undefined;
}
return result[0].schema;
})
@@ -1267,12 +1267,12 @@ export class PostgresStorageAdapter implements StorageAdapter {
const createValue = Object.assign({}, query, update);
return this.createObject(className, schema, createValue)
.catch(error => {
// ignore duplicate value errors as it's upsert
if (error.code !== Parse.Error.DUPLICATE_VALUE) {
throw error;
}
return this.findOneAndUpdate(className, schema, query, update);
});
// ignore duplicate value errors as it's upsert
if (error.code !== Parse.Error.DUPLICATE_VALUE) {
throw error;
}
return this.findOneAndUpdate(className, schema, query, update);
});
}
find(className: string, schema: SchemaType, query: QueryType, { skip, limit, sort, keys }: QueryOptions) {
@@ -1452,7 +1452,7 @@ export class PostgresStorageAdapter implements StorageAdapter {
throw error;
}
return 0;
});
});
}
distinct(className: string, schema: SchemaType, query: QueryType, fieldName: string) {