start building tg worker
This commit is contained in:
parent
334968d6fd
commit
1665d40f27
@ -2,28 +2,35 @@ package workers
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
"github.com/go-redis/redis/v8"
|
"github.com/go-redis/redis/v8"
|
||||||
"penahub.gitlab.yandexcloud.net/backend/quiz/common.git/dal"
|
"penahub.gitlab.yandexcloud.net/backend/quiz/common.git/dal"
|
||||||
|
"penahub.gitlab.yandexcloud.net/backend/quiz/common.git/model"
|
||||||
|
"penahub.gitlab.yandexcloud.net/backend/quiz/core/clients/telegram"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Deps struct {
|
type Deps struct {
|
||||||
BotID int64
|
BotID int64
|
||||||
Redis *redis.Client
|
Redis *redis.Client
|
||||||
Dal *dal.DAL
|
Dal *dal.DAL
|
||||||
|
TgClient *telegram.TelegramClient
|
||||||
}
|
}
|
||||||
|
|
||||||
type TgListenerWorker struct {
|
type TgListenerWorker struct {
|
||||||
botID int64
|
botID int64
|
||||||
redis *redis.Client
|
redis *redis.Client
|
||||||
dal *dal.DAL
|
dal *dal.DAL
|
||||||
|
tgClient *telegram.TelegramClient
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewTgListenerWC(deps Deps) *TgListenerWorker {
|
func NewTgListenerWC(deps Deps) *TgListenerWorker {
|
||||||
return &TgListenerWorker{
|
return &TgListenerWorker{
|
||||||
botID: deps.BotID,
|
botID: deps.BotID,
|
||||||
redis: deps.Redis,
|
redis: deps.Redis,
|
||||||
dal: deps.Dal,
|
dal: deps.Dal,
|
||||||
|
tgClient: deps.TgClient,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -42,5 +49,44 @@ func (wc *TgListenerWorker) Start(ctx context.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (wc *TgListenerWorker) processTasks(ctx context.Context) {
|
func (wc *TgListenerWorker) processTasks(ctx context.Context) {
|
||||||
|
var cursor uint64
|
||||||
|
for {
|
||||||
|
var keys []string
|
||||||
|
var err error
|
||||||
|
keys, cursor, err = wc.redis.Scan(ctx, cursor, "telegram_task:*", 0).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Failed scan for telegram tasks:", err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, key := range keys {
|
||||||
|
func() {
|
||||||
|
taskBytes, err := wc.redis.GetDel(ctx, key).Result()
|
||||||
|
if err == redis.Nil {
|
||||||
|
return
|
||||||
|
} else if err != nil {
|
||||||
|
fmt.Println("Failed getdel telegram task:", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
|
||||||
|
}()
|
||||||
|
|
||||||
|
var task model.TgRedisTask
|
||||||
|
if json.Unmarshal([]byte(taskBytes), &task) != nil {
|
||||||
|
fmt.Println("Failed unmarshal telegram task:", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
inviteLink, chatID, err := wc.tgClient.CreateChannel(task.Name, wc.botID)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Failed create tg channel:", err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
if cursor == 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user