"example/05_conv2d_fwd/README.md" did not exist on "acbd7bd7c5efd17b7061157a5868e28acc04d33e"
restServer.ts 3.33 KB
Newer Older
liuzhe-lz's avatar
liuzhe-lz committed
1
2
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT license.
Deshui Yu's avatar
Deshui Yu committed
3
4
5

'use strict';

6
import * as assert from 'assert';
Deshui Yu's avatar
Deshui Yu committed
7
8
9
import * as express from 'express';
import * as http from 'http';
import { Deferred } from 'ts-deferred';
10
import { getLogger, Logger } from './log';
11
import { getBasePort } from './experimentStartupInfo';
Deshui Yu's avatar
Deshui Yu committed
12

SparkSnail's avatar
SparkSnail committed
13

14
15
/**
 * Abstraction class to create a RestServer
16
 * The module who wants to use a RestServer could <b>extends</b> this abstract class
17
18
19
 * And implement its own registerRestHandler() function to register routers
 */
export abstract class RestServer {
Deshui Yu's avatar
Deshui Yu committed
20
21
22
23
    private startTask!: Deferred<void>;
    private stopTask!: Deferred<void>;
    private server!: http.Server;

24
25
26
27
28
    /** The fields can be inherited by subclass */
    protected hostName: string = '0.0.0.0';
    protected port?: number;
    protected app: express.Application = express();
    protected log: Logger = getLogger();
29
    protected basePort?: number;
30

31
32
33
34
35
    constructor() {
        this.port = getBasePort();
        assert(this.port && this.port > 1024);
    }

Deshui Yu's avatar
Deshui Yu committed
36
37
38
39
40
    get endPoint(): string {
        // tslint:disable-next-line:no-http-string
        return `http://${this.hostName}:${this.port}`;
    }

41
42
    public start(hostName?: string): Promise<void> {
        this.log.info(`RestServer start`);
Deshui Yu's avatar
Deshui Yu committed
43
44
45
46
47
48
49
50
51
52
        if (this.startTask !== undefined) {
            return this.startTask.promise;
        }
        this.startTask = new Deferred<void>();

        this.registerRestHandler();

        if (hostName) {
            this.hostName = hostName;
        }
53
54

        this.log.info(`RestServer base port is ${this.port}`);
Deshui Yu's avatar
Deshui Yu committed
55

56
        this.server = this.app.listen(this.port as number, this.hostName).on('listening', () => {
Deshui Yu's avatar
Deshui Yu committed
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
            this.startTask.resolve();
        }).on('error', (e: Error) => {
            this.startTask.reject(e);
        });

        return this.startTask.promise;
    }

    public stop(): Promise<void> {
        if (this.stopTask !== undefined) {
            return this.stopTask.promise;
        }
        this.stopTask = new Deferred<void>();

        if (this.startTask === undefined) {
            this.stopTask.resolve();

            return this.stopTask.promise;
        } else {
            this.startTask.promise.then(
                () => { // Started
78
79
80
                    //Stops the server from accepting new connections and keeps existing connections.
                    //This function is asynchronous, the server is finally closed when all connections
                    //are ended and the server emits a 'close' event.
SparkSnail's avatar
SparkSnail committed
81
                    //Refer https://nodejs.org/docs/latest/api/net.html#net_server_close_callback
Deshui Yu's avatar
Deshui Yu committed
82
83
84
85
86
87
88
89
90
91
92
93
94
                    this.server.close().on('close', () => {
                        this.log.info('Rest server stopped.');
                        this.stopTask.resolve();
                    }).on('error', (e: Error) => {
                        this.log.error(`Error occurred stopping Rest server: ${e.message}`);
                        this.stopTask.reject();
                    });
                },
                () => { // Start task rejected
                    this.stopTask.resolve();
                }
            );
        }
SparkSnail's avatar
SparkSnail committed
95
        this.stopTask.resolve()
Deshui Yu's avatar
Deshui Yu committed
96
97
98
        return this.stopTask.promise;
    }

99
100
101
102
    /**
     * Register REST handler, which is left for subclass to implement
     */
    protected abstract registerRestHandler(): void;
Deshui Yu's avatar
Deshui Yu committed
103
}