61 lines
1.8 KiB
TypeScript
61 lines
1.8 KiB
TypeScript
|
import { IncomingMessage } from "http";
|
||
|
|
import * as path from "path";
|
||
|
|
import { Counter, HttpConnection, HttpServer } from "tsrpc";
|
||
|
|
import { serviceProto, ServiceType } from "./shared/protocols/serviceProto";
|
||
|
|
|
||
|
|
// Create the Server
|
||
|
|
const server = new HttpServer(serviceProto, {
|
||
|
|
port: 3000,
|
||
|
|
cors: '*'
|
||
|
|
});
|
||
|
|
|
||
|
|
// Init
|
||
|
|
export async function init(context: any, callback: (err: Error | null, data: any) => void) {
|
||
|
|
// Auto implement APIs
|
||
|
|
await server.autoImplementApi(path.resolve(__dirname, 'api'));
|
||
|
|
callback(null, '');
|
||
|
|
}
|
||
|
|
|
||
|
|
const connCounter = new Counter(1);
|
||
|
|
|
||
|
|
// Handler
|
||
|
|
export function handler(httpReq: IncomingMessage, httpRes: {
|
||
|
|
setStatusCode: (code: number) => void,
|
||
|
|
setHeader: (key: string, value: string) => void,
|
||
|
|
deleteHeader: (key: string) => void,
|
||
|
|
send: (body: Buffer | string) => void
|
||
|
|
}, context: any) {
|
||
|
|
httpRes.setStatusCode(200);
|
||
|
|
httpRes.setHeader('X-Powered-By', `TSRPC`);
|
||
|
|
httpRes.setHeader('Access-Control-Allow-Origin', '*');
|
||
|
|
httpRes.setHeader('Access-Control-Allow-Headers', 'Content-Type,*');
|
||
|
|
if (httpReq.method === 'OPTIONS') {
|
||
|
|
httpRes.send('');
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
let chunks: Buffer[] = [];
|
||
|
|
httpReq.on('data', data => {
|
||
|
|
chunks.push(data);
|
||
|
|
});
|
||
|
|
|
||
|
|
let conn: HttpConnection<ServiceType> | undefined;
|
||
|
|
httpReq.on('end', async () => {
|
||
|
|
let isJSON = false;
|
||
|
|
conn = new HttpConnection({
|
||
|
|
server: 'FAAS' as any,
|
||
|
|
id: '' + connCounter.getNext(),
|
||
|
|
ip: '???',
|
||
|
|
httpReq: httpReq,
|
||
|
|
httpRes: { end: httpRes.send.bind(httpRes) } as any,
|
||
|
|
isJSON: isJSON
|
||
|
|
});
|
||
|
|
await server.flows.postConnectFlow.exec(conn, conn.logger);
|
||
|
|
|
||
|
|
let buf = chunks.length === 1 ? chunks[0] : Buffer.concat(chunks);
|
||
|
|
|
||
|
|
server['_onRecvBuffer'](conn, buf);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|