2018-10-09 03:46:15 +00:00
|
|
|
package puller
|
|
|
|
|
|
|
|
import (
|
|
|
|
"math"
|
|
|
|
|
|
|
|
"github.com/zelenin/go-tdlib/client"
|
|
|
|
)
|
|
|
|
|
|
|
|
func Chats(tdlibClient *client.Client) (chan *client.Chat, chan error) {
|
|
|
|
chatChan := make(chan *client.Chat, 10)
|
|
|
|
errChan := make(chan error, 1)
|
|
|
|
|
|
|
|
var offsetOrder client.JsonInt64 = math.MaxInt64
|
|
|
|
var offsetChatId int64 = 0
|
|
|
|
var limit int32 = 100
|
|
|
|
|
|
|
|
go chats(tdlibClient, chatChan, errChan, offsetOrder, offsetChatId, limit)
|
|
|
|
|
|
|
|
return chatChan, errChan
|
|
|
|
}
|
|
|
|
|
|
|
|
func chats(tdlibClient *client.Client, chatChan chan *client.Chat, errChan chan error, offsetOrder client.JsonInt64, offsetChatId int64, limit int32) {
|
|
|
|
defer func() {
|
|
|
|
close(chatChan)
|
|
|
|
close(errChan)
|
|
|
|
}()
|
|
|
|
|
|
|
|
for {
|
2018-10-16 20:32:57 +00:00
|
|
|
chats, err := tdlibClient.GetChats(&client.GetChatsRequest{
|
|
|
|
OffsetOrder: offsetOrder,
|
|
|
|
OffsetChatId: offsetChatId,
|
|
|
|
Limit: limit,
|
|
|
|
})
|
2018-10-09 03:46:15 +00:00
|
|
|
if err != nil {
|
|
|
|
errChan <- err
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(chats.ChatIds) == 0 {
|
|
|
|
errChan <- EOP
|
|
|
|
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, chatId := range chats.ChatIds {
|
2018-10-19 16:04:48 +00:00
|
|
|
chat, err := tdlibClient.GetChat(&client.GetChatRequest{
|
|
|
|
ChatId: chatId,
|
|
|
|
})
|
2018-10-09 03:46:15 +00:00
|
|
|
if err != nil {
|
|
|
|
errChan <- err
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
offsetOrder = chat.Order
|
|
|
|
offsetChatId = chat.Id
|
|
|
|
|
|
|
|
chatChan <- chat
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|