Баг с FSM с telebot?

Question

[0xsetup]

В прошлом вопросе думал насчёт реализации FSM и решил сделать с помощью redis.

Сделал простую реализацию FSM:

FSM

func (f *FSM) GetState(ctx context.Context, userID int) (string, error) {
	state, err := f.redisPool.Get(ctx, f.stateKey(userID)).Result()
	if errors.Is(err, redis.Nil) {
		return NilState, nil
	} else if err != nil {
		return "", err
	}

	return state, nil
}

func (f *FSM) SetState(ctx context.Context, userID int, state string) error {
	return f.redisPool.Set(ctx, f.stateKey(userID), state, T).Err()
}

func (f *FSM) stateKey(userID int) string {
	return fmt.Sprintf("user:%d:state", userID)
}

В коде бота меняю состояние примерно вот так:

Code

// AddCategoryHandler ...
func (h *Handlers) AddCategoryHandler() tele.HandlerFunc {
	return func(tb tele.Context) error {
		ctx := context.Background()

		if err := h.fsm.SetState(ctx, int(tb.Sender().ID), fsm.InputAddCategoryState); err != nil {
			h.logger.Err(err).Msg("failed to set state")
			return tb.Respond(&tele.CallbackResponse{
				CallbackID: tb.Callback().ID,
				Text:       messages.InternalErrorStr,
				ShowAlert:  false,
				URL:        "",
			})
		} ....

Вроде бы все работает нормально, когда ботом пользуется мало человек, но когда идёт большой наплыв, примерно 100 RPS, то состояния могут ломаться, т.е если 1 человек нажал эту кнопку, и в это время другой тоже, то состояние может перейти другому пользователю. Итак вопрос: как этого избежать?

Comments

[shurshur]

Если "состояние ушло другому пользователю", то это изначально ошибочная реализация.

Состояние должно в ключе иметь id пользователя. От количества пользователей бот может перестать справляться, но взять по id одного пользователя состояние другого пользователя - никак.

[0xsetup]

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

[0xsetup]

shurshur, вот полный код:

code

func (h *Handlers) AddCategoryHandler() tele.HandlerFunc {
	return func(tb tele.Context) error {
		ctx := context.Background()
		senderID := int(tb.Sender().ID)

		if err := h.fsm.SetState(ctx, senderID, fsm.InputAddCategoryState); err != nil {
			h.logger.Err(err).Msg("failed to set state")
			return tb.Respond(&tele.CallbackResponse{
				CallbackID: tb.Callback().ID,
				Text:       messages.InternalErrorStr,
				ShowAlert:  false,
				URL:        "",
			})
		}

		selector.Inline(selector.Row(BackBtn))
		if err := tb.EditCaption("<b> Работа с категориями</b>\n\nℹ️ Введите название категории:", selector); err != nil {
			h.logger.Err(err).Msg("failed to send message")
			return nil
		}

		tb.Bot().Handle(tele.OnText, func(tbText tele.Context) error {
			currentState, err := h.fsm.GetState(ctx, senderID)
			if err != nil {
				h.logger.Err(err).Msg("failed to get state")
				return tb.Respond(&tele.CallbackResponse{
					CallbackID: tb.Callback().ID,
					Text:       messages.InternalErrorStr,
					ShowAlert:  false,
					URL:        "",
				})
			}

			if currentState != fsm.InputAddCategoryState {
				return nil
			}

			categoryName := strings.TrimSpace(tbText.Message().Text)

			if err = tbText.Delete(); err != nil {
				h.logger.Err(err).Msg("failed to delete message")

				return tb.Respond(&tele.CallbackResponse{
					CallbackID: tb.Callback().ID,
					Text:       messages.InternalErrorStr,
					ShowAlert:  false,
					URL:        "",
				})
			}

			if err = h.categoriesService.CategoryWebService.Save(ctx, categoryName); err != nil {
				if err = h.fsm.SetState(ctx, senderID, fsm.StartState); err != nil {
					h.logger.Err(err).Msg("failed to set state")
					return tb.Respond(&tele.CallbackResponse{
						CallbackID: tb.Callback().ID,
						Text:       messages.InternalErrorStr,
						ShowAlert:  false,
						URL:        "",
					})
				}

				h.logger.Err(err).Msg("failed to save new category")

				return tb.EditCaption(messages.InternalErrorStr, selector)
			}

			if err = h.fsm.SetState(ctx, senderID, fsm.StartState); err != nil {
				h.logger.Err(err).Msg("failed to set state")
				return tb.Respond(&tele.CallbackResponse{
					CallbackID: tb.Callback().ID,
					Text:       messages.InternalErrorStr,
					ShowAlert:  false,
					URL:        "",
				})
			}

			selector.Inline(selector.Row(BackAdminPanelBtn))

			return tb.EditCaption(fmt.Sprintf("✅ Новая категория с именем <b>%s</b> успешно добавлена!", categoryName), selector)
		})

		tb.Bot().Handle(&BackBtn, func(tb tele.Context) error {
			if err := h.fsm.SetState(ctx, senderID, fsm.StartState); err != nil {
				h.logger.Err(err).Msg("failed to set state")
				return tb.Respond(&tele.CallbackResponse{
					CallbackID: tb.Callback().ID,
					Text:       messages.InternalErrorStr,
					ShowAlert:  false,
					URL:        "",
				})
			}

			return h.categoryPanel(tb)
		})

		return nil
	}
}

[shurshur]

0xsetup, я go не знаю и поэтому опишу "на пальцах".

Предствим себе, у нас есть "состояние" - это может быть строка, а может быть класс с полями. Самый простой способ хранить состояния - это массив:

user_states[user_id] = state

Но это имеет недостатки: мы вынуждены обходиться одним процессом, а также теряем состояние при падении/перезапуске. Ещё на это требуется память, при большой аудитории это может оказаться важно.

Ещё одна проблема может быть с множественным доступом в многозадачных реализациях, особенно если выполняемое действие не атомарно. Ну, там надо это средствами языка учитывать, например, использовать блокировки на чём-нибудь типа мутексов.

Довольно частая ошибка новичков - они хранят state в одной переменной без учёта конкретного пользователя. Тогда если между запросами одного пользователя влезает другой (третий и так далее), случаются очевидные проблемы.

Вот, и вместо массива состояний можно класть/брать в базу или в key-value.

В данном случае видимо используется redis. Но если состояния путаются, то я полагаю там что-то путается в ключах. Например, может быть stateKey не только userId использует, но и как-то зависит от других внешних сущностей, которые успевают поменяться между запусками?

Повторю, я go не знаю, рассказываю общие вещи.