Как получить chat_id из базы данных — Telegram bot?

Question

[Паша]

Привет
Я делаю бота на Telegraf, использую cron для отправки оповещений и mongodb для хранения данных юзеров.
Пытаюсь запустить примерно такой скрипт:

const CronJob = require('cron').CronJob;

const job = new CronJob('*/5 * * * * *', async function() {
    const user = await db.User.findOne({chat_id: ctx.chat.id});
    await bot.telegram.sendMessage(user.chat_id, 'Hello World');
})

bot.command('launch', async (ctx) => {
  job.start()
})

bot.command('stop', async (ctx) => {
  job.stop();
})

Когда я использую db.User.findOne(), получается взять только первого юзера из базы данных.
Также я могу получить нужный chat_id в событии, когда я обращаюсь к ctx.
Вот пример:

bot.command('launch', async (ctx) => {
  const user = await db.User.findOne({chat_id: ctx.chat.id});
  console.log(user.chat_id)
})

Подскажите, пожалуйста, решение проблемы. Уже несколько дней с этим мучаюсь.

Answer 1

[Дмитрий Гололобов]

const job = new CronJob('*/5 * * * * *', async function() {
    const user = await db.User.findOne({chat_id: ctx.chat.id});
    await bot.telegram.sendMessage(user.chat_id, 'Hello World');
})

Эта функция не видит контекст ctx, т.е. в теории chatid в нее не передается

Comments

[Паша]

Спасибо за ответ
В том и дело, что я бы хотел передать chat_id, но не знаю как.

[Дмитрий Гололобов]

Паша, попробуй так:

Тут функцию создания задач в расписании надо обренуть еще в одну, чтобы в нее можно было передать чатID:

const createJob = async function(chatid) {
 return new CronJob('*/5 * * * * *', async function() {
    const user = await db.User.findOne({chat_id: chatid});
    await bot.telegram.sendMessage(user.chat_id, 'Hello World');
});
}



bot.command('launch', async (ctx) => {
  const myNewJob = await createJob(ctx.chat.id)
  myNewJob.start()
})

Но в этом примере есть косяк, остановить такую задачу будет не реально, так как она создана внутри команды launch

Поэтому по хорошему под работающие задачи нужен отдельный объект или массив объектов, в общем некий state, в котором бы хранились задачи.

А лучше расписание задач хранить в какой-нибудь базе, а при старте бота, просто запускать его.

В общем начните с того, что я написал, а дальше постепенно по чуть-чуть продвигайтесь в нужную вам сторону

[Паша]

Дмитрий Гололобов, работает!
Благодарю за помощь!

Answer 2

[Паша]

Дмитрий Гололобов, попробовал ваш вариант, работает.
Пытаюсь сделать так, чтобы бот отправлял определённое количество сообщений из БД:

// беру сообщения из бд
const messages = data.get('messages')
    .map('message')
    .value();

let i=0;

const CronJob = require('cron').CronJob;

const createJob = async function(chatid) {
  return new CronJob('*/15 * * * * *', async function() {
    const user = await db.User.findOne({chat_id: chatid});
    await bot.telegram.sendMessage(user.chat_id, messages[i]);
    console.log('Job running');
    i++;
 });
 }
 
 bot.command('launch', async (ctx) => {
   const myNewJob = await createJob(ctx.chat.id)
   await bot.telegram.sendMessage(ctx.chat.id, `Let's start`);
   myNewJob.start()
 })

Когда юзеры оправляют запрос, то последний, кто начал, подхватывает скрипт, то есть у него приходит следующее сообщение из списка для первого юзера.

Вот ещё один вариант попробовал. Тут плюс, что можно остановить скрипт. Берется список юзеров, которые оправили запрос, и сообщения распределяются между юзерами, то есть если начали одновременно, то из 4 сообщений каждому придет по 2 разных.

const messages = data.get('messages')
    .map('message')
    .value();

const CronJob = require('cron').CronJob;

const usersToNotify = []

let i=0;

const job = new CronJob('*/15 * * * * *', async function() {
  for await (const user of usersToNotify) {
    await bot.telegram.sendMessage(user, messages[i]);
    i++;
    if (messages.length == i) {
      job.stop();
      }
}
})

job.start()

bot.command('launch', async (ctx) => {
  const User = await db.User.findOne({chat_id: ctx.chat.id});
  usersToNotify.push(User.chat_id)
})

bot.command('stop', async (ctx) => {
  const User = await db.User.findOne({chat_id: ctx.chat.id});
  const index = usersToNotify.findIndex(value => value === User.chat_id)
  usersToNotify.splice(index,1)
})

В обоих вариантах кажется проблема в том, что cron работает для всех юзеров глобально.
Подскажите, пожалуйста, можно как-то запускать скрипт отдельно для каждого chat_id?
Благодарю заранее!

Comments

[Дмитрий Гололобов]

Я вот сейчас смотрю на все это и понимаю, что я бы так сильно не заморачивался.

Можно сделать так например:

Пользователь сообщает боту "хочу получать уведомления" (команда launch), бот в ответ на запрос пользователя, просто складывает в базу пометку, что такому-то пользователю будем отправлять сообщения,

Командой stop пользователь сообщает боту, о том, что больше не хочет получать уведомления. И по этой команде отметка в базе меняется на обратную.

А cronjob можно сделать тогда общий. Только выборку сделать по пользователям такую, в которой будет учитываться наличие флага "отправлять уведомления"

Псевдокод:

const job = new CronJob('*/15 * * * * *', async () => {
  const users = await getUsersFromDBwhereNotifyTrue();  //тут написать правильную выборку из БД
  for(const user of users) {
      await bot.telegram.sendMessage(user.telegram_id, 'текст уведомления');
  }
}

bot.command('launch'), async ctx => {
   setThisUserNotifyTrueInDB();  //тут написать правильный запрос в БД
   await ctx.reply('Теперь вы будете получать уведомления раз в 15 минут');
}

bot.command('stop'), async ctx => {
   setThisUserNotifyFalseInDB();  //тут написать правильный запрос в БД
   await ctx.reply('Вы отписались от уведомлений!!!');
}

job.start();
bot.launch();

[Паша]

Дмитрий Гололобов, спасибо за ответ.
Просто я пытаюсь сделать серию сообщений как миникурс, то есть при отправке команды юзер получает эту серию сообщений независимо от других юзеров.