WorkerController.js 2.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. const BaseController = require('./BaseController');
  2. const WorkerState = require('../core/WorkerState');//singleton
  3. const utils = require('../core/utils');
  4. class WorkerController extends BaseController {
  5. constructor(config) {
  6. super(config);
  7. this.workerState = new WorkerState();
  8. }
  9. async getState(req, res) {
  10. const request = req.body;
  11. let error = '';
  12. try {
  13. if (!request.workerId)
  14. throw new Error(`key 'workerId' is wrong`);
  15. const state = this.workerState.getState(request.workerId);
  16. return (state ? state : {});
  17. } catch (e) {
  18. error = e.message;
  19. }
  20. //bad request
  21. res.status(400).send({error});
  22. return false;
  23. }
  24. async getStateFinish(req, res) {
  25. const request = req.body;
  26. let error = '';
  27. try {
  28. if (!request.workerId)
  29. throw new Error(`key 'workerId' is wrong`);
  30. res.writeHead(200, {
  31. 'Content-Type': 'text/json; charset=utf-8',
  32. });
  33. const splitter = '-- aod2t5hDXU32bUFyqlFE next status --';
  34. const refreshPause = 300;
  35. let i = 0;
  36. let prevProgress = -1;
  37. let prevState = '';
  38. let state;
  39. while (1) {// eslint-disable-line no-constant-condition
  40. state = this.workerState.getState(request.workerId);
  41. if (!state) break;
  42. res.write(splitter + JSON.stringify(state));
  43. res.flush();
  44. if (state.state != 'finish')
  45. await utils.sleep(refreshPause);
  46. else
  47. break;
  48. i++;
  49. if (i > 2*60*1000/refreshPause) {//2 мин ждем телодвижений воркера
  50. res.write(splitter + JSON.stringify({state: 'error', error: 'Слишком долгое время ожидания'}));
  51. break;
  52. }
  53. i = (prevProgress != state.progress || prevState != state.state ? 1 : i);
  54. prevProgress = state.progress;
  55. prevState = state.state;
  56. }
  57. if (!state) {
  58. res.write(splitter + JSON.stringify({}));
  59. }
  60. res.end();
  61. return false;
  62. } catch (e) {
  63. error = e.message;
  64. }
  65. //bad request
  66. res.status(400).send({error});
  67. return false;
  68. }
  69. }
  70. module.exports = WorkerController;