如何在Go的telebot库中创建多处理器实现分步授权对话？

要实现分步式的授权流程，核心是跟踪用户的对话状态，根据用户当前所处的步骤来处理输入。以下是具体实现步骤：

1. 定义状态常量与用户数据结构 #

首先需要定义用户在授权流程中的不同状态，以及临时存储用户输入的结构：

import "sync"

// 定义授权流程的状态
const (
    StateIdle = iota // 无状态（未进入授权流程）
    StateWaitingUsername // 等待用户输入用户名
    StateWaitingPassword // 等待用户输入密码
)

// 临时存储用户的授权数据
type UserAuthData struct {
    Username string
}

// 用线程安全的map存储用户状态和临时数据（适配telebot的并发处理）
var userStates = sync.Map{}       // key: int64(用户ID), value: int(状态)
var userAuthTempData = sync.Map{} // key: int64(用户ID), value: *UserAuthData

2. 处理授权按钮的触发 #

在你已有的授权按钮处理器中，启动授权流程，切换用户状态并发送第一步提示：

a.bot.Handle(&btnAuth, func(c tele.Context) error {
    userID := c.Sender().ID

    // 将用户状态切换为「等待用户名」
    userStates.Store(userID, StateWaitingUsername)
    // 初始化该用户的临时授权数据
    userAuthTempData.Store(userID, &UserAuthData{})

    // 发送第一步提示
    return c.Send("Enter the username.")
})

3. 处理用户的文本输入 #

添加一个全局文本消息处理器，根据用户当前状态来处理输入：

a.bot.Handle(tele.OnText, func(c tele.Context) error {
    userID := c.Sender().ID
    inputText := c.Text()

    // 获取用户当前状态，若未进入授权流程则忽略
    currentState, exists := userStates.Load(userID)
    if !exists || currentState.(int) == StateIdle {
        return nil
    }

    // 根据状态分支处理输入
    switch currentState.(int) {
    case StateWaitingUsername:
        // 保存用户名到临时数据
        authData, _ := userAuthTempData.Load(userID)
        authData.(*UserAuthData).Username = inputText

        // 切换状态为「等待密码」并发送提示
        userStates.Store(userID, StateWaitingPassword)
        return c.Send("Enter the password.")

    case StateWaitingPassword:
        // 获取之前保存的用户名
        authData, _ := userAuthTempData.Load(userID)
        username := authData.(*UserAuthData).Username
        password := inputText

        // --- 这里替换为你的实际授权逻辑 ---
        // 示例：假设授权验证通过
        authSuccess := true
        // authSuccess := yourAuthService.Verify(username, password)

        if authSuccess {
            // 发送授权完成提示
            err := c.Send("Authorization completed")
            // 清理用户状态和临时数据
            userStates.Delete(userID)
            userAuthTempData.Delete(userID)
            return err
        } else {
            // 授权失败，重置状态让用户重新开始
            userStates.Store(userID, StateWaitingUsername)
            authData.(*UserAuthData).Username = ""
            return c.Send("Authorization failed, please re-enter the username.")
        }
    }

    return nil
})

补充说明 #

• 并发安全：使用sync.Map是因为telebot会用多个goroutine处理消息，普通map会有并发读写问题。
• 超时处理（可选）：如果需要处理用户长时间不操作的情况，可以添加定时任务，定期清理超过一定时间的用户状态和临时数据。
• 异常处理：可以在每个步骤中添加错误捕获，比如发送消息失败的情况，避免处理器崩溃。

内容的提问来源于stack exchange，提问作者Ilya