hiro/apollo-server
1
0
Fork 0
mirror of https://github.com/vale981/apollo-server synced 2025-03-06 10:11:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactored server handler

Browse source
This commit is contained in:
nnance 2016-07-16 07:31:14 -07:00
parent 6b6fcba2d2
commit f9bc7dec8d
2 changed files with 51 additions and 47 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/integrations/hapiApollo.test.ts
View file

@ -28,6 +28,6 @@ function createApp(options: CreateAppOptions = {}) {
return server.listener;
}
describe.only('integration:HAPI', () => {
describe('integration:HAPI', () => {
testSuite(createApp);
});

96
src/integrations/hapiApollo.ts
View file

@ -10,6 +10,10 @@ export interface IRegister {
attributes?: any;
}
export interface HAPIOptionsFunction {
(req?: hapi.Request): ApolloOptions | Promise<ApolloOptions>;
}
export class ApolloHAPI {
constructor() {
this.register.attributes = {
@ -34,56 +38,14 @@ export class ApolloHAPI {
optionsObject = options;
}
const formatErrorFn = optionsObject.formatError || graphql.formatError;
if (!request.payload) {
reply('POST body missing.').code(500);
return;
}
let b = request.payload;
let isBatch = true;
// TODO: do something different here if the body is an array.
// Throw an error if body isn't either array or object.
if (!Array.isArray(b)) {
isBatch = false;
b = [b];
}
const responses = await processQuery(request.payload, optionsObject);
let responses: Array<graphql.GraphQLResult> = [];
for (let payload of b) {
try {
const operationName = payload.operationName;
let variables = payload.variables;
if (typeof variables === 'string') {
// TODO: catch errors
variables = JSON.parse(variables);
}
let params = {
schema: optionsObject.schema,
query: payload.query,
variables: variables,
rootValue: optionsObject.rootValue,
operationName: operationName,
logFunction: optionsObject.logFunction,
validationRules: optionsObject.validationRules,
formatError: formatErrorFn,
formatResponse: optionsObject.formatResponse,
};
if (optionsObject.formatParams) {
params = optionsObject.formatParams(params);
}
responses.push(await runQuery(params));
} catch (e) {
responses.push({ errors: [formatErrorFn(e)] });
}
}
if (isBatch) {
if (responses.length > 1) {
reply(responses);
} else {
const gqlResponse = responses[0];
@ -131,8 +93,50 @@ export class GraphiQLHAPI {
}
}
export interface HAPIOptionsFunction {
(req?: hapi.Request): ApolloOptions | Promise<ApolloOptions>;
async function processQuery(body, optionsObject) {
const formatErrorFn = optionsObject.formatError || graphql.formatError;
let isBatch = true;
// TODO: do something different here if the body is an array.
// Throw an error if body isn't either array or object.
if (!Array.isArray(body)) {
isBatch = false;
body = [body];
}
let responses: Array<graphql.GraphQLResult> = [];
for (let payload of body) {
try {
const operationName = payload.operationName;
let variables = payload.variables;
if (typeof variables === 'string') {
// TODO: catch errors
variables = JSON.parse(variables);
}
let params = {
schema: optionsObject.schema,
query: payload.query,
variables: variables,
rootValue: optionsObject.rootValue,
operationName: operationName,
logFunction: optionsObject.logFunction,
validationRules: optionsObject.validationRules,
formatError: formatErrorFn,
formatResponse: optionsObject.formatResponse,
};
if (optionsObject.formatParams) {
params = optionsObject.formatParams(params);
}
responses.push(await runQuery(params));
} catch (e) {
responses.push({ errors: [formatErrorFn(e)] });
}
}
return responses;
}
function isOptionsFunction(arg: ApolloOptions | HAPIOptionsFunction): arg is HAPIOptionsFunction {