Куда и как добавить асинхронность в телеграм бота?

Question

[gelerum]

Здравствуйте, я использую библиотеку telebot вместе с MongoDB

Через GetClient() и GetBot() я общаюсь с одноименными структурами

type client struct {
	client *mongo.Client  // установливаю соединение и не закрываю через другую функцию
}

Соединение с базой создается в самом начале и не закрывается.

func main() {
    telegram.InitBot() // создаю соединение с апи бота 
    storage.InitClient() // создаю соединение с mongodb и добавляю клиент в структуру
    telegram.GetBot().Bot.Handle("/start", telegram.GetBot().HandleStart)
}

Функция ниже выполняется, по сообщению пользователя /start

func (b *bot) HandleStart(message *tb.Message) {
	b.Bot.Send(message.Sender, "Hello")
	storage.GetClient().CreateUserDocument(message.Sender.ID) // создает документ в бд, если еще не существует с таким id пользователя
}

Так я создаю документ с начальной структурой и пользовательским id, если еще с таким id не создано:

func (c client) CreateUserDocument(chatID int) {
	coll := c.client.Database("db").Collection("users")
	count, _ := coll.CountDocuments(context.TODO(), bson.D{{"chatID", chatID}})
	if count != 1 {
		document := bson.D{{"chatID", chatID}, {"expenses", bson.A{}}, {"income", bson.A{}}}
		coll.InsertOne(context.TODO(), document)
	}
}

Как мне оптимизировать этот код и где добавить асинхронность? Это не весь код проэкта, на другие комманды бота делаются другие запросы в бд.

Answer 1

[Евгений Мамонов]

Можно сделать работу бота через webhook'и, тогда каждый запрос будет изначально обрабатываться в отдельной горутине.
Плюс при таком подходе можно горизонтально масштабировать нагрузку, т.е. запустить бот на нескольких серверах и через какой нибудь балансировщик раскидывать на них запросы.

Comments

[gelerum]

Спасибо, так и сделаю

[gelerum]

А мне разве не надо изменить как-то код, чтобы горутины переключались между друг другом, они же сами по себе не будут определять где ожидание?

[Евгений Мамонов]

Если вы делаете обработчик вебхуков на пакете, например, net/http - тогда, когда приходит новый запрос handler вызывается уже в новой горутине. Т.е. новые горутины внутри handler'a уже запускать не нужно.

Вот хороший пример реализации с помощью webhook'ов
https://www.sohamkamani.com/golang/telegram-bot/

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

func (b *bot) HandleStart(message *tb.Message) {
    go func() {
        b.Bot.Send(message.Sender, "Hello")
        storage.GetClient().CreateUserDocument(message.Sender.ID)
  }()
}

или так, в зависимости от того, что нужно

func (b *bot) HandleStart(message *tb.Message) {
  // запускаем "тяжелую" функцию в горутине
  go func() {
      storage.GetClient().CreateUserDocument(message.Sender.ID)
  }()
  // отвечаем быстро, чтобы пользователь не ждал
  b.Bot.Send(message.Sender, "Hello")
}

[gelerum]

Евгений Мамонов, А мне не нужно использовать sync.WaitGroup?
И я правильно понимаю, что при таком подходе: сначала пользователь 1 отправил комманду /start, потом пользователь 2 отправил такую же команду. Для первого пользователя идет создание структуры, и пока выполняется InsertOne(), начинается обработка 2 пользователя, то есть если добавить в начало HandleStart() отправку сообщения, то пока идет запись структуры 1 пользователя, 2 второму отправится сообщение?
Это ведь не будет работать, как я описал, нужно не полностью storage.GetClient().CreateUserDocument(message.Sender.ID) добавлять в горутину, а отдельно строчку с coll.InsertOne()

[Евгений Мамонов]

Если брать эту часть

func (b *bot) HandleStart(message *tb.Message) {
    // тут вы отправляете сообщние конкретному получателю message.Sender, не первому, не второму, а именно тому, кто инициировал этот вызов.
    b.Bot.Send(message.Sender, "Hello")
    // тут тоже вы создаёте для message.Sender.ID
    storage.GetClient().CreateUserDocument(message.Sender.ID)
}

Коллизия может быть только если один и тот же пользователь с разных устройств может одновременно запустить /start, но я не уверен на 100%, не исключено, что ChatID для разных устройств будет отличаться, надо проверить.

Я бы решал это средствами базы, т.к. защита на уровне мьютекса не поможет в случае, когда у вас будет несколько инстансов бота.

В вашем случае я бы создал уникальный индекс по полю chatID, т.е. две таких записи у вас не появится уже 100%

[gelerum]

Евгений Мамонов, chatId с разных устройств будет одинаковым оно равно id аккаунта. Я не могу понять, если взять пример из вашего прошлого коментария, то как будет происходить переключение между горутинами, как будет определяться, где сейчас идет write или read, чтобы начать обрабатывать данные в другой горутине(сообщенте бругого пользователя)?

[Евгений Мамонов]

Это всё делает планировщик Go.
Вам не нужно об этом думать, ваша основная задача убедиться, что не будет коллизии данных.
В вашем коде я вижу только то, о чём писал выше.

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

Если всё же не сложилась картина - напишите мне в телеграм, можно будет созвониться