WorkerController.js 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  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. //TODO: удалить бесполезную getStateFinish
  25. async getStateFinish(req, res) {
  26. const request = req.body;
  27. let error = '';
  28. try {
  29. if (!request.workerId)
  30. throw new Error(`key 'workerId' is wrong`);
  31. res.writeHead(200, {
  32. 'Content-Type': 'text/json; charset=utf-8',
  33. });
  34. const splitter = '-- aod2t5hDXU32bUFyqlFE next status --';
  35. const refreshPause = 200;
  36. let i = 0;
  37. let prevProgress = -1;
  38. let prevState = '';
  39. let state;
  40. while (1) {// eslint-disable-line no-constant-condition
  41. state = this.workerState.getState(request.workerId);
  42. if (!state) break;
  43. res.write(splitter + JSON.stringify(state));
  44. res.flush();
  45. if (state.state != 'finish' && state.state != 'error')
  46. await utils.sleep(refreshPause);
  47. else
  48. break;
  49. i++;
  50. if (i > 2*60*1000/refreshPause) {//2 мин ждем телодвижений воркера
  51. res.write(splitter + JSON.stringify({state: 'error', error: 'Слишком долгое время ожидания'}));
  52. break;
  53. }
  54. i = (prevProgress != state.progress || prevState != state.state ? 1 : i);
  55. prevProgress = state.progress;
  56. prevState = state.state;
  57. }
  58. if (!state) {
  59. res.write(splitter + JSON.stringify({}));
  60. }
  61. res.end();
  62. return false;
  63. } catch (e) {
  64. error = e.message;
  65. }
  66. //bad request
  67. res.status(400).send({error});
  68. return false;
  69. }
  70. }
  71. module.exports = WorkerController;