new request system
This commit is contained in:
47
api.go
47
api.go
@@ -16,6 +16,53 @@ type ApiResponse struct {
|
|||||||
Result any `json:"result,omitempty"`
|
Result any `json:"result,omitempty"`
|
||||||
ErrorCode int `json:"error_code,omitempty"`
|
ErrorCode int `json:"error_code,omitempty"`
|
||||||
}
|
}
|
||||||
|
type ApiResponseG[R any] struct {
|
||||||
|
Ok bool `json:"ok"`
|
||||||
|
Description string `json:"description,omitempty"`
|
||||||
|
Result R `json:"result,omitempty"`
|
||||||
|
ErrorCode int `json:"error_code,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type TelegramRequest[R, P any] struct {
|
||||||
|
method string
|
||||||
|
params P
|
||||||
|
}
|
||||||
|
type EmptyParams struct{}
|
||||||
|
|
||||||
|
func NewRequest[R, P any](method string, params P) TelegramRequest[R, P] {
|
||||||
|
return TelegramRequest[R, P]{
|
||||||
|
method: method,
|
||||||
|
params: params,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (r TelegramRequest[R, P]) Do(bot *Bot) (*R, error) {
|
||||||
|
var buf bytes.Buffer
|
||||||
|
err := json.NewEncoder(&buf).Encode(r.params)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
req, err := http.Post(fmt.Sprintf("https://api.telegram.org/bot%s/%s", bot.token, r.method), "application/json", &buf)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer req.Body.Close()
|
||||||
|
data, err := io.ReadAll(req.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
response := new(ApiResponseG[R])
|
||||||
|
err = json.Unmarshal(data, &response)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !response.Ok {
|
||||||
|
return nil, fmt.Errorf("[%d] %s", response.ErrorCode, response.Description)
|
||||||
|
}
|
||||||
|
return &response.Result, nil
|
||||||
|
}
|
||||||
|
|
||||||
// request is a low-level call to api.
|
// request is a low-level call to api.
|
||||||
func (b *Bot) request(methodName string, params any) (map[string]any, error) {
|
func (b *Bot) request(methodName string, params any) (map[string]any, error) {
|
||||||
|
|||||||
@@ -43,6 +43,8 @@ func (b *Bot) Updates() ([]*Update, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bot) GetMe() (*User, error) {
|
func (b *Bot) GetMe() (*User, error) {
|
||||||
|
//req := NewRequest[User, EmptyParams]("getMe", EmptyParams{})
|
||||||
|
//user, err := req.Do(b)
|
||||||
data, err := b.request("getMe", NoParams)
|
data, err := b.request("getMe", NoParams)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
@@ -69,6 +71,9 @@ type SendMessageP struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bot) SendMessage(params *SendMessageP) (*Message, error) {
|
func (b *Bot) SendMessage(params *SendMessageP) (*Message, error) {
|
||||||
|
req := NewRequest[Message, SendMessageP]("sendMessage", *params)
|
||||||
|
return req.Do(b)
|
||||||
|
|
||||||
data, err := b.request("sendMessage", params)
|
data, err := b.request("sendMessage", params)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|||||||
@@ -61,10 +61,15 @@ func (b *Bot) ExecRunners() {
|
|||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
} else if !runner.Async && runner.Onetime {
|
} else if !runner.Async && runner.Onetime {
|
||||||
|
t := time.Now()
|
||||||
err := runner.Fn(b)
|
err := runner.Fn(b)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
b.logger.Warnf("Runner %s failed: %s\n", runner.Name, err)
|
b.logger.Warnf("Runner %s failed: %s\n", runner.Name, err)
|
||||||
}
|
}
|
||||||
|
elapsed := time.Since(t)
|
||||||
|
if elapsed > time.Second*2 {
|
||||||
|
b.logger.Warnf("Runner %s too slow. Elapsed time %s>=2s", runner.Name, elapsed)
|
||||||
|
}
|
||||||
} else if !runner.Onetime {
|
} else if !runner.Onetime {
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
for {
|
||||||
|
|||||||
Reference in New Issue
Block a user