Как предотвратить «UnhandledPromiseRejection» при неправильном вводе пользователем? - PullRequest
Новая
       7

Как предотвратить «UnhandledPromiseRejection» при неправильном вводе пользователем?

1 голос
/ 15 апреля 2019

У меня есть 2 файла, main.js и input.js. В файле input.js я предлагаю пользователю указать интервал в формате «1-10» (строка). Позже я обрезаю эту строку и получаю из нее 2 числа и проверяю, верны ли числа: 

let getInput = () => {
  return new Promise(function (resolve, reject) {
    readline.question(`Give me interval (e.g.: 1-10).\n`, (input) => {
      let fields = input.split('-');
      let startNumber = Number(fields[0]);
      let endNumber = Number(fields[1]);
      if ((startNumber) && (endNumber) && (startNumber > 0) && (endNumber >= startNumber)) {
        console.log(`ok`);
        readline.close()
        resolve([startNumber, endNumber]);
      } else {
        reject('not ok');
        getInput();
      }
    });
  });
}

В main.js я вызываю эту функцию асинхронно и сохраняю ее результат в переменной: 

let intervalArray = await getInput();
.
.
.
someotherstuff

Моя проблема в том, что я неправильно ввел данные (например, «0-1», «10 -9» или «-10»). Я получил ошибку UnhandledPromise, поэтому код больше не будет выполняться. Моей целью было бы, когда пользователь вводит неверные данные, в консоли должно появиться «не хорошо, дай мне еще один», и программа должна ждать другого ввода. Если эти данные верны, продолжите выполнение. Если нет, программа должна запросить другой ввод.

Как мне этого добиться?

EDIT: Вот полные коды. Input.js: 

const readline = require('readline').createInterface({
  input: process.stdin,
  output: process.stdout
});

let getInput = () => {
  return new Promise(function (resolve, reject) {
    readline.question(`tol-ig formatumban (pl.: 1-10).\n`, (data) => {
      let fields = data.split('-');
      let startNumber = Number(fields[0]);
      let endNumber = Number(fields[1]);
      if ((startNumber) && (endNumber) && (startNumber > 0) && (endNumber >= startNumber)) {
        console.log(`Kereses inditasa ${startNumber}-${endNumber} oldalakon.`);
        readline.close()
        resolve([startNumber, endNumber]);
      } else {
        readline.close();
        reject(new Error('not ok'));
      }
    });
  });
}

module.exports.getInput = getInput;

И main.js: 

const puppeteer = require('puppeteer');
const { getInput } = require('./input');
const { mouseMovements } = require('./mouse');
const { tuneUserAgent } = require('./userAgent');

async function listItems() {
  let intervalArray = null;
  while (intervalArray === null) {
    try {
      let intervalArray = await getInput();
    } catch (err) {
      // write to user not ok
    }
  }
  const browser = await puppeteer.launch({ headless: false });
  const context = await browser.createIncognitoBrowserContext();
  const page = await context.newPage();

  const extractPartners = async url => {
    const page = await context.newPage();
    await tuneUserAgent(page);

    await page.goto(url, { waitUntil: 'load' });
    await page.waitFor(Math.round(Math.random() * 500) + 500);

    await mouseMovements(page);

    const partnersOnPage = await page.evaluate(() =>
      Array.from(document.querySelectorAll("div.compact"))
        .map(compact => (compact.querySelector(".logo a").href.slice(-16))));
    await page.close();

    const nextPageNumber = parseInt(url.match(/page=(\d+)$/)[1], 10) + 1;
    if (nextPageNumber > endPage) {
      console.log(`Terminate recursion on: ${url}`);
      return partnersOnPage;
    } else {
      console.log(`Scraped: ${url}`);
      const nextUrl = `https://marketingplatform.google.com/about/partners/find-a-partner?page=${nextPageNumber}`;
      let randomWait = (Math.round(Math.random() * 2000) + 1000);
      await page.waitFor(randomWait);
      return partnersOnPage.concat(await extractPartners(nextUrl));
    }
  };
  let startPage = intervalArray[0];
  let endPage = intervalArray[1];
  const firstUrl =
    `https://marketingplatform.google.com/about/partners/find-a-partner?page=${startPage}`;
  const partners = await extractPartners(firstUrl);
  await browser.close();
  return Promise.resolve(partners);
};

module.exports.listItems = listItems;

1 Ответ

3 голосов
/ 15 апреля 2019

Вы должны обработать ошибку. Поскольку вы используете await, проще всего использовать try catch. 

try {
  let intervalArray = await getInput();
} catch (err) {
  // write to user not ok
}

Вы можете свернуть его в цикле, чтобы продолжать запрашивать у пользователя новые входные данные. 

let intervalArray = null;
while (intervalArray === null) {
    try {
      let intervalArray = await getInput();
    } catch (err) {
      // write to user not ok
    }
}

^^ Не забудьте удалить getInput(); из else части вашего нового Обещания. Возможно, вам также придется закрыть readline, так как вы откроете его снова. Отклонение также похоже на выдачу ошибки, и вы всегда должны отправлять туда объект, основанный на ошибках. 

  } else {
    readline.close();
    reject(new Error('not ok'));
  }

РЕДАКТИРОВАТЬ: после вашего обновления я создал POC. readline.close() на самом деле не должно быть (выглядит только для однократного использования), но этот POC выглядит нормально:

toBeRequired.js

const readline = require('readline').createInterface({
    input: process.stdin,
    output: process.stdout
});

let getInput = () => {
    return new Promise(function (resolve, reject) {
        readline.question(`tol-ig formatumban (pl.: 1-10).\n`, (data) => {
            let fields = data.split('-');
            let startNumber = Number(fields[0]);
            let endNumber = Number(fields[1]);
            if ((startNumber) && (endNumber) && (startNumber > 0) && (endNumber >= startNumber)) {
                console.log(`Kereses inditasa ${startNumber}-${endNumber} oldalakon.`);
                readline.close()
                resolve([startNumber, endNumber]);
            } else {
                reject(new Error('not ok'));
            }
        });
    });
}

module.exports.getInput = getInput;

server.js

const a = require('./toBeRequired');

async function x() {
  let input = null;
  while (input === null) {
    try {
      input = await a.getInput();
    } catch (err) {
      console.log('nooo');
    }
  }
}

x();

Создайте эти два файла в одной папке и запустите node server.js

...