public
/
epg-grabber
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update utils.js

This commit is contained in:
Aleksandr Statciuk 2021-10-13 23:10:27 +03:00
parent 940846f52a
commit 10998d969c
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/utils.js
View File

@ -10,6 +10,7 @@ const utc = require('dayjs/plugin/utc')
dayjs.extend(utc) dayjs.extend(utc)
axiosCookieJarSupport(axios) axiosCookieJarSupport(axios)
let timeout
const utils = {} const utils = {}
const defaultUserAgent = const defaultUserAgent =
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3945.130 Safari/537.36 Edg/79.0.309.71' 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3945.130 Safari/537.36 Edg/79.0.309.71'
@ -166,7 +167,7 @@ utils.buildRequest = async function (item, config) {
const CancelToken = axios.CancelToken const CancelToken = axios.CancelToken
const source = CancelToken.source() const source = CancelToken.source()
const request = { ...config.request } const request = { ...config.request }
const timeout = setTimeout(() => { timeout = setTimeout(() => {
source.cancel('Connection timeout') source.cancel('Connection timeout')
}, request.timeout) }, request.timeout)
const headers = await utils.getRequestHeaders(item, config) const headers = await utils.getRequestHeaders(item, config)
@ -183,6 +184,17 @@ utils.buildRequest = async function (item, config) {
} }
utils.fetchData = function (request) { utils.fetchData = function (request) {
axios.interceptors.response.use(
function (response) {
clearTimeout(timeout)
return response
},
function (error) {
clearTimeout(timeout)
return Promise.reject(error)
}
)
return axios(request) return axios(request)
} }