WorkerController.js 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  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 = 200;
  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. if (i == 0) {
  43. state = Object.assign({dummy: '0'.repeat(1024)}, state);
  44. }
  45. res.write(splitter + JSON.stringify(state));
  46. res.flush();
  47. if (state.state != 'finish' && state.state != 'error')
  48. await utils.sleep(refreshPause);
  49. else
  50. break;
  51. i++;
  52. if (i > 2*60*1000/refreshPause) {//2 мин ждем телодвижений воркера
  53. res.write(splitter + JSON.stringify({state: 'error', error: 'Слишком долгое время ожидания'}));
  54. break;
  55. }
  56. i = (prevProgress != state.progress || prevState != state.state ? 1 : i);
  57. prevProgress = state.progress;
  58. prevState = state.state;
  59. }
  60. if (!state) {
  61. res.write(splitter + JSON.stringify({}));
  62. }
  63. res.end();
  64. return false;
  65. } catch (e) {
  66. error = e.message;
  67. }
  68. //bad request
  69. res.status(400).send({error});
  70. return false;
  71. }
  72. }
  73. module.exports = WorkerController;