2

使用 HTTP Router 处理 Telegram Bot 按钮回调

 2 years ago
source link: https://jiajunhuang.com/articles/2022_03_12-tgbot_httprouter.md.html
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.

使用 HTTP Router 处理 Telegram Bot 按钮回调

写 Telegram Bot 的时候,我们可能会选择使用 Inline keyboard, 这样的好处是界面比较好看,交互也好,消息下面可以有很多按钮,点击一下就可以更新消息和按钮,但是开发起来就比较麻烦, 因为每一个点击,对于 Telegram Bot 程序来说,都是要处理一个回调,而且大部分情况下按钮是需要带一定的消息回去的。

比如分页按钮,用户点了数字为 3 的按钮,在回调的时候,就需要知道用户是点击的 3 的按钮,从而展示第三页的内容。 通常我们都是把回调信息放到 callback_data 里,从定义可以看到, callback_data 是一个字符串,最长64个字符,最短1个,如果不填写的话,就不会触发 callback 的操作(点击按钮也就没有响应)。

Telegram Inline Keyboard

如果只有一两个按钮,那么处理起来就会很简单,在处理回调的地方,写一堆的 if...else... 或者是用一个 map[string]Func 来 保存 callback_data 和具体逻辑的关系就可以,但是当按钮多了以后,我们就需要一种更好的方式来切分回调函数的处理逻辑,因此, 我就把 httprouter 引入到 Telegram Bot 的开发中。

httprouter

httprouter 是一个HTTP用的路由框架,它底层的原理是把每一个 handler 挂载到 radix tree 的节点上,当我们有一个 HTTP 请求到来 时,它就根据 URL 去 radix tree 上查找,找到对应的 handler 然后执行:

Priority   Path             Handle
9          \                *<1>
3          ├s               nil
2          |├earch\         *<2>
1          |└upport\        *<3>
2          ├blog\           *<4>
1          |    └:post      nil
1          |         └\     *<5>
2          ├about-us\       *<6>
1          |        └team\  *<7>
1          └contact\        *<8>

它的用法就像是这样:

func Index(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
	fmt.Fprint(w, "Not protected!\n")
}

func main() {
	router := httprouter.New()
	router.GET("/", Index)

	log.Fatal(http.ListenAndServe(":8080", router))
}

具体原理可以参考我的这篇 httprouter源码分析

结合 Telegram Bot

我使用的 Telegram Bot 框架是 telebot,这个框架想要处理回调,就得注册这么一个函数:

b	bot, err = tb.NewBot(tb.Settings{}) // ...

bot.Handle(tb.OnCallback, HandleCallback)

func HandleCallback(c tb.Context) error {
	callback := c.Callback()
	logger.Info("received callback", zap.Any("callback_data", callback.Data))

    // 处理回调

	return nil
}

如果有多个按钮需要处理回调,例如分页按钮,可能展示了 1-10项,而且很有可能是动态的,比如点击到第2页的时候,还要把 10-20项 展示出来,那就不好在 HandleCallback 中分别 if...else... 方式处理了,那样很麻烦,那有没有更好的方案呢?

我们平时写 HTTP 服务的时候,通常 HTTP 请求,我们可以在四个地方传参数:

  • URL本身,例如 RESTful 中的 /groups/1/users/1/
  • Query String,例如 ?user=1&page=2&offset=3
  • Header,例如 Authorization: Bearer blablabla
  • Payload,也就是请求的Body,例如传一个JSON或者XML

但是 Telegram Bot 的回调没有这么强大,这也是我一直觉得 Telegram Bot 的表达能力不足的原因之一,由于 callback_data 是一个最长 64字节的字符串,我们只能考虑放 URLQuery String,而且由于 Query String 一般都是 ?a=b&c=d 的形式,会浪费不少 空间,因此我就只考虑 URL 了,那么很自然,可以考虑使用 httprouter 来做分发,并且把参数放到 URL 里面。

因此,最后的使用就会变的像这样,首先在 main 函数里注册路由:

router.Handle("/users/list/page/:page", CallbackGetUsersByPage)

然后定义这个函数的逻辑:

func CallbackGetUsersByPage(c tb.Context, params router.Params) {
    // ...
}

接下来就是具体的代码逻辑了,想要取参数,可以直接在 params 里获取。

当然,具体源码我就没有贴出来了,思路已经展示完毕,如果有需要的话,可以自己整一个。

由于 Telegram Bot 回调按钮的表达力偏弱,当有很多回调按钮时,我们就需要有一个合理的框架能够帮我们做到逻辑拆分,因此我 结合了 httprouter 来完成这个目标。此后就可以通过注册路由,然后把不同路由的逻辑处理放到不同的函数里,并且可以实现传参 等功能。


微信公众号
关注公众号,获得及时更新

读《系统之美:决策者的系统思考》

Linux高分屏支持

GCC默认的头文件搜索路径

读《远见-如何规划职业生涯3大阶段》

后端工程师学前端(五): SASS

后端工程师学前端(四): CSS进阶(盒子模型)

读《投资中最简单的事》

后端工程师学前端(三): CSS进阶(特指度、单位和字体族)

后端工程师学前端(二): CSS基础知识(规则与选择器)

Swift语法笔记

读《管理的实践》

后端工程师学前端(一): HTML

frp 源码阅读与分析(二):TCP内网穿透的实现

五天不用微信 - 爽得很

frp 源码阅读与分析(一):流程和概念




About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK