From 7563ce479c466d95d8580d0f78ba69df69cf5ed2 Mon Sep 17 00:00:00 2001 From: c0re100 Date: Sat, 25 Nov 2023 21:08:02 +0800 Subject: [PATCH] Allow disable sendMessage patch --- client/client.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/client/client.go b/client/client.go index 114818c..e3ac867 100644 --- a/client/client.go +++ b/client/client.go @@ -21,6 +21,7 @@ type Client struct { successMsgStore *sync.Map updatesTimeout time.Duration catchTimeout time.Duration + DisablePatch bool } type Option func(*Client) @@ -43,6 +44,12 @@ func WithProxy(req *AddProxyRequest) Option { } } +func WithoutSendMessagePatch() Option { + return func(client *Client) { + client.DisablePatch = true + } +} + func SetLogLevel(level int32) { _, _ = SetLogVerbosityLevel(&SetLogVerbosityLevelRequest{ NewVerbosityLevel: level, @@ -184,7 +191,7 @@ func (client *Client) Send(req Request) (*Response, error) { select { case response := <-catcher: - if response.Type != "error" && req.Type == "sendMessage" { + if !client.DisablePatch && response.Type != "error" && req.Type == "sendMessage" { m, err := UnmarshalMessage(response.Data) if err != nil { return nil, err