multi bot webhook only use WebhookHTTPServeMux

[tonyhorton]

Document cases can only use ServeMux of the standard library, and use fastthttp to achieve the same function. Therefore, with the appearance of the following code, it can continue to use fastthttp and support multi_bot_webhook while supporting multi_bot_webhook.

I implemented a method myself, which can then use fastthttp as the webhook server and supports multiplexed handlers(The code is handwritten for extracting key parts. The actual project structure is too complex. The following code is for reference only):

BotWebhookServer := &fasthttp.Server{
    Name:                          "Bot-Webhook-HTTP-Server",
    Concurrency:                   1024 * 32,            // 最大并发请求数
    DisableKeepalive:              false,                // 开启Keepalive (Telegram推荐开启)
    ReadBufferSize:                4096,                 // 读取缓冲区 (4KB)
    WriteBufferSize:               4096,                 // 写入缓冲区 (4KB)
    ReadTimeout:                   10 * time.Second,     // 读取超时 (避免长连接无响应)
    WriteTimeout:                  10 * time.Second,     // 写入超时
    IdleTimeout:                   2 * 60 * time.Minute, // 闲置连接超时（保持一定的连接）
    MaxConnsPerIP:                 0,                    // 0表示无限制 (Telegram官方IP有限，无需限制)
    MaxRequestsPerConn:            0,                    // 每个连接允许无限请求数（Telegram Keepalive友好）
    TCPKeepalive:                  true,                 // TCP层Keepalive
    TCPKeepalivePeriod:            2 * 60 * time.Minute, // TCP Keepalive周期
    MaxRequestBodySize:            2 * 1024 * 1024,      // 2MB (Telegram官方请求Body小于2MB)
    ReduceMemoryUsage:             true,                 // 降低内存占用，适合大量并发场景
    LogAllErrors:                  false,                // 不记录所有错误日志，避免日志过多
    SecureErrorLogMessage:         true,                 // 不在错误中输出敏感信息
    DisableHeaderNamesNormalizing: true,                 // 提升性能 (Telegram Header 固定格式)
    NoDefaultServerHeader:         true,                 // 不返回默认的Server头信息（提升安全性）
    NoDefaultDate:                 false,                // 保持默认时间头信息 (HTTP协议建议)
    NoDefaultContentType:          true,                 // 不设置默认Content-Type，处理器自行决定
    StreamRequestBody:             false,                // Telegram Bot 无需流式Body
}
BotWebhookServerRouter := router.New()
BotsList := map[string]*telego.Bot
BotsList = map[string]*telego.Bot{}
go func() {
    _ = core.BotWebhookServer.ListenAndServe(":2000")     
}()

for _, botID := range BotsList{
	updates, err := BotsList[botID].UpdatesViaWebhook(
		ctx,
		// 手动重写 FastHTTP Server
		func(handler TGBotAPI.WebhookHandler) error {
			return func(handler TGBotAPI.WebhookHandler) error {
				BotWebhookServerRouter.POST(fmt.Sprintf("/bot_%d", botConfig.BotID), func(ctx *fasthttp.RequestCtx) {
					// 验证 Secret Token
					if BotsList[botID].SecretToken() != "" && BotsList[botID].SecretToken() != string(ctx.Request.Header.Peek(TGBotAPI.WebhookSecretTokenHeader)) {
						ctx.SetStatusCode(fasthttp.StatusUnauthorized)
						return
					}
					if err := handler(ctx, ctx.PostBody()); err != nil {
						ctx.SetStatusCode(fasthttp.StatusInternalServerError)
						return
					}
					
					ctx.SetStatusCode(fasthttp.StatusOK)
				})
				
				return nil
			}(handler)
			
		},
		
		TGBotAPI.WithWebhookSet(ctx, &TGBotAPI.SetWebhookParams{
			// 服务器需要发送 webhook 的 URL
			URL:       "webhookURL",
			IPAddress: botConfig.LocalListenHost,
			
			// 允许的update
			AllowedUpdates: nil,
			
			// 最大链接数
			MaxConnections: 10000,
			
			// 是否跳过旧Update
			DropPendingUpdates: Config.RunType == "Developments",
			
			// 安全密钥
			SecretToken: bots["botID"].SecretToken(),
		}),
	)
	if err != nil {
		logger.Warn("Get Updates Via WebHookHost Warn: '%s', Bot BotID: '%d'", err.Error(), botConfig.BotID)
	}
}

BotWebhookServer.Handler = BotWebhookServerRouter.Handler
select {}

[tonyhorton]

core code:

func(handler TGBotAPI.WebhookHandler) error {
	return func(handler TGBotAPI.WebhookHandler) error {
		BotWebhookServerRouter.POST(fmt.Sprintf("/bot_%d", botConfig.BotID), func(ctx *fasthttp.RequestCtx) {
			// 验证 Secret Token
			if BotsList[botID].SecretToken() != "" && BotsList[botID].SecretToken() != string(ctx.Request.Header.Peek(TGBotAPI.WebhookSecretTokenHeader)) {
				ctx.SetStatusCode(fasthttp.StatusUnauthorized)
				return
			}
			if err := handler(ctx, ctx.PostBody()); err != nil {
				ctx.SetStatusCode(fasthttp.StatusInternalServerError)
				return
			}
			
			ctx.SetStatusCode(fasthttp.StatusOK)
		})
		
		return nil
	}(handler)
},

[mymmrac]

Yep, fasthttp can be used for multibot webhooks too, I just didn't add this example because pure fasthttp requires router (https://github.com/fasthttp/router or https://github.com/gofiber/fiber) to be used, I guess it still will be usefull to add that example

[mymmrac]

Here is example using Fiber
https://github.com/mymmrac/telego/blob/main/examples/multi_bot_webhook_fiber/main.go

[tonyhorton]

it's cool!

[tonyhorton]

Can I use the Deepwiki tool to generate documents on the official website?

https://deepwiki.com/mymmrac/telego

https://telego.pixelbox.dev/docs/introduction/overview/

[mymmrac]

Hmm, looks very promising, thanks, I will look into that