1
0
Fork 0
mirror of https://github.com/openstf/stf synced 2025-10-04 10:19:30 +02:00

Modify APIs from Aysnc -> Sync using timeout

This commit is contained in:
Vishal Banthia 2015-12-14 14:02:20 +09:00
parent d67d06a19f
commit a6266931ad
10 changed files with 354 additions and 239 deletions

View file

@ -1,5 +1,6 @@
var http = require('http')
var path = require('path')
var events = require('events')
var express = require('express')
var SwaggerExpress = require('swagger-express-mw')
@ -11,11 +12,13 @@ var logger = require('../../util/logger')
var zmqutil = require('../../util/zmqutil')
var srv = require('../../util/srv')
var lifecycle = require('../../util/lifecycle')
var wireutil = require('../../wire/util')
module.exports = function(options) {
var log = logger.createLogger('api')
, app = express()
, server = http.createServer(app)
, channelRouter = new events.EventEmitter()
var push = zmqutil.socket('push')
Promise.map(options.endpoints.push, function(endpoint) {
@ -27,18 +30,47 @@ module.exports = function(options) {
})
})
})
.catch(function(err) {
.catch(function(err) {w
log.fatal('Unable to connect to push endpoint', err)
lifecycle.fatal()
})
// Input
var sub = zmqutil.socket('sub')
Promise.map(options.endpoints.sub, function(endpoint) {
return srv.resolve(endpoint).then(function(records) {
return srv.attempt(records, function(record) {
log.info('Receiving input from "%s"', record.url)
sub.connect(record.url)
return Promise.resolve(true)
})
})
})
.catch(function(err) {
log.fatal('Unable to connect to sub endpoint', err)
lifecycle.fatal()
})
// Establish always-on channels
;[wireutil.global].forEach(function(channel) {
log.info('Subscribing to permanent channel "%s"', channel)
sub.subscribe(channel)
})
sub.on('message', function(channel, data) {
channelRouter.emit(channel.toString(), channel, data)
})
// Swagger Express Config
var config = {
appRoot: __dirname
, swaggerFile: path.resolve(__dirname, 'swagger', 'api_v1.yaml')
};
SwaggerExpress.create(config, function(err, swaggerExpress) {
if (err) { throw err; }
if (err) {
throw err
}
swaggerExpress.register(app);
})
@ -47,6 +79,8 @@ module.exports = function(options) {
app.use(function(req, res, next) {
var reqOptions = _.merge(options, {
'push': push
, 'sub': sub
, 'channelRouter': channelRouter
})
req.options = reqOptions
@ -59,6 +93,17 @@ module.exports = function(options) {
, keys: [options.secret]
}))
lifecycle.observe(function() {
[push, sub].forEach(function(sock) {
try {
sock.close()
}
catch (err) {
// No-op
}
})
})
server.listen(options.port)
log.info('Listening on port %d', options.port)
}