onAfterCall support

moleculerjs · Mar 11, 2021 · ad11c5a · ad11c5a
1 parent 1af1c37
commit ad11c5a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/index.d.ts b/index.d.ts
@@ -81,7 +81,7 @@ declare module "moleculer-apollo-server" {
 			fallbackResponse?: any;
 		};
 		onBeforeCall?: (ctx: Context, route: any, req: any, res: any) => Promise<any>;
-		onAfterCall?: (ctx: Context, route: any, req: any, res: any) => Promise<any>;
+		onAfterCall?: (ctx: Context, route: any, req: any, res: any, data: any) => Promise<any>;
 	}
 
 	export interface ApolloServiceOptions {

diff --git a/src/ApolloServer.js b/src/ApolloServer.js
@@ -6,14 +6,20 @@ const { renderPlaygroundPage } = require("@apollographql/graphql-playground-html
 const accept = require("@hapi/accept");
 const moleculerApollo = require("./moleculerApollo");
 
-function send(req, res, statusCode, data, responseType = "application/json") {
+async function send(req, res, statusCode, data, responseType = "application/json") {
 	res.statusCode = statusCode;
 
 	const ctx = res.$ctx;
 	if (!ctx.meta.$responseType) {
 		ctx.meta.$responseType = responseType;
 	}
 
+	const route = res.$route;
+	if (route.onAfterCall) {
+		data = await route.onAfterCall.call(this, ctx, route, req, res, data);
+	}
+
+
 	const service = res.$service;
 	service.sendResponse(req, res, data);
 }