reader.js 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. import axios from 'axios';
  2. import {sleep} from '../share/utils';
  3. const api = axios.create({
  4. baseURL: '/api/reader'
  5. });
  6. const workerApi = axios.create({
  7. baseURL: '/api/worker'
  8. });
  9. class Reader {
  10. async loadBook(url, callback) {
  11. const refreshPause = 200;
  12. if (!callback) callback = () => {};
  13. let response = await api.post('/load-book', {type: 'url', url});
  14. const workerId = response.data.workerId;
  15. if (!workerId)
  16. throw new Error('Неверный ответ api');
  17. callback({totalSteps: 4});
  18. let i = 0;
  19. while (1) {// eslint-disable-line no-constant-condition
  20. callback(response.data);
  21. if (response.data.state == 'finish') {//воркер закончил работу, можно скачивать кешированный на сервере файл
  22. callback({step: 4});
  23. const book = await this.loadCachedBook(response.data.path, callback);
  24. return Object.assign({}, response.data, {data: book.data});
  25. }
  26. if (response.data.state == 'error') {
  27. let errMes = response.data.error;
  28. if (errMes.indexOf('getaddrinfo') >= 0 ||
  29. errMes.indexOf('ECONNRESET') >= 0 ||
  30. errMes.indexOf('EINVAL') >= 0 ||
  31. errMes.indexOf('404') >= 0)
  32. errMes = `Ресурс не найден по адресу: ${response.data.url}`;
  33. throw new Error(errMes);
  34. }
  35. if (i > 0)
  36. await sleep(refreshPause);
  37. i++;
  38. if (i > 30*1000/refreshPause) {//30 сек ждем телодвижений воркера
  39. throw new Error('Слишком долгое время ожидания');
  40. }
  41. //проверка воркера
  42. const prevProgress = response.data.progress;
  43. response = await workerApi.post('/get-state', {workerId});
  44. i = (prevProgress != response.data.progress ? 1 : i);
  45. }
  46. }
  47. async loadCachedBook(url, callback){
  48. const options = {
  49. onDownloadProgress: progress => {
  50. const total = (progress.total ? progress.total : progress.loaded + 200000);
  51. if (callback)
  52. callback({state: 'loading', progress: Math.round((progress.loaded*100)/total)});
  53. }
  54. }
  55. //загрузка
  56. return await axios.get(url, options);
  57. }
  58. }
  59. export default new Reader();