mirror of
https://github.com/Ratstail91/sineQL.git
synced 2025-11-29 02:34:28 +11:00
Moved to using a real database ORM for testing
This commit is contained in:
119
test/handlers/query-handlers.js
Normal file
119
test/handlers/query-handlers.js
Normal file
@@ -0,0 +1,119 @@
|
||||
const { Op } = require('sequelize');
|
||||
const { books, authors } = require('../database/models');
|
||||
|
||||
//the handler functions return arrays for each type, containing every element that satisfies the queries
|
||||
|
||||
//the "query" argument contains the object built from the sineQL query
|
||||
//the "graph" argument contains the typeGraph
|
||||
|
||||
//the task of the handler functions is to query the database, and return the correct results
|
||||
|
||||
/* possible values for "query" include:
|
||||
|
||||
{
|
||||
id: 1,
|
||||
typeName: 'Author',
|
||||
name: { typeName: 'String', scalar: true, match: 'Kenneth Grahame' },
|
||||
books: { typeName: 'Book', match: {
|
||||
id: 2,
|
||||
typeName: 'Book',
|
||||
title: { typeName: 'String', scalar: true, match: 'The Wind in the Willows' }
|
||||
published: { typeName: 'Date', scalar: true }
|
||||
}
|
||||
}
|
||||
|
||||
*/
|
||||
|
||||
//depth-first search seems to be the best option
|
||||
//Each query shouldn't know if it's a sub-query
|
||||
|
||||
const queryHandlers = {
|
||||
//complex compound
|
||||
Author: async (query, graph) => {
|
||||
//get the fields alone
|
||||
const { typeName, match, ...fields } = query;
|
||||
|
||||
//hack the id into the fields list (if it's not there already)
|
||||
fields['id'] = fields['id'] || { typeName: 'Integer', scalar: true }; //TODO: should this be default?
|
||||
|
||||
//get the names of matched fields (fields to find)
|
||||
const matchedNames = Object.keys(fields).filter(field => fields[field].match);
|
||||
|
||||
//short-circuit if querying everything
|
||||
const where = {};
|
||||
if (matchedNames.length > 0) {
|
||||
//build the "where" object
|
||||
matchedNames.forEach(mn => {
|
||||
if (query[mn].match !== true) { //true means it's a compound type
|
||||
where[mn] = { [Op.eq]: query[mn].match };
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
//these are field names
|
||||
const scalars = Object.keys(fields).filter(field => graph[fields[field].typeName].scalar);
|
||||
const nonScalars = Object.keys(fields).filter(field => !graph[fields[field].typeName].scalar);
|
||||
|
||||
let authorResults = await authors.findAll({
|
||||
attributes: scalars, //fields to find (keys)
|
||||
where: where,
|
||||
raw: true
|
||||
}); //sequelize ORM model
|
||||
|
||||
const promiseArray = nonScalars.map(async nonScalar => {
|
||||
//hack the author ID in, so it can be referenced below
|
||||
fields[nonScalar]['authorId'] = fields[nonScalar]['authorId'] || { typeName: 'Integer', scalar: true };
|
||||
|
||||
//delegate to a deeper part of the tree
|
||||
const nonScalarArray = await queryHandlers[fields[nonScalar].typeName](fields[nonScalar], graph);
|
||||
|
||||
//for each author, update this non-scalar field with the non-scalar's recursed value
|
||||
authorResults = authorResults.map(author => {
|
||||
author[nonScalar] = nonScalarArray.filter(ns => ns['authorId'] == author.id);
|
||||
return author;
|
||||
});
|
||||
|
||||
//prune the authors when matching, but their results are empty
|
||||
authorResults = authorResults.filter(author => {
|
||||
return !(fields[nonScalar].match && author[nonScalar].length == 0);
|
||||
});
|
||||
});
|
||||
|
||||
await Promise.all(promiseArray);
|
||||
|
||||
//finally, return the results
|
||||
return authorResults;
|
||||
},
|
||||
|
||||
//simple compound
|
||||
Book: async (query, graph) => {
|
||||
//get the fields alone
|
||||
const { typeName, match, ...fields } = query;
|
||||
|
||||
//hack the id into the fields list (if it's not there already)
|
||||
fields['id'] = fields['id'] || { typeName: 'Integer', scalar: true }; //TODO: should this be automatic?
|
||||
|
||||
//get the names of matched fields
|
||||
const matchedNames = Object.keys(fields).filter(field => fields[field].match);
|
||||
|
||||
//short-circuit if querying everything
|
||||
const where = {};
|
||||
if (matchedNames.length > 0) {
|
||||
//build the "where" object
|
||||
matchedNames.forEach(mn => {
|
||||
if (query[mn].match !== true) {
|
||||
where[mn] = { [Op.eq]: query[mn].match };
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
//return the result (N+1 bottleneck)
|
||||
return await books.findAll({
|
||||
attributes: Object.keys(fields), //fields to find (everything for simple compounds)
|
||||
where: where,
|
||||
raw: true
|
||||
}); //sequelize ORM model
|
||||
}
|
||||
};
|
||||
|
||||
module.exports = queryHandlers;
|
||||
Reference in New Issue
Block a user