start building tg worker

This commit is contained in:
Pavel 2024-07-11 14:09:17 +03:00
parent 334968d6fd
commit 1665d40f27

@ -2,28 +2,35 @@ package workers
import (
"context"
"encoding/json"
"fmt"
"github.com/go-redis/redis/v8"
"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"
)
type Deps struct {
BotID int64
Redis *redis.Client
Dal *dal.DAL
BotID int64
Redis *redis.Client
Dal *dal.DAL
TgClient *telegram.TelegramClient
}
type TgListenerWorker struct {
botID int64
redis *redis.Client
dal *dal.DAL
botID int64
redis *redis.Client
dal *dal.DAL
tgClient *telegram.TelegramClient
}
func NewTgListenerWC(deps Deps) *TgListenerWorker {
return &TgListenerWorker{
botID: deps.BotID,
redis: deps.Redis,
dal: deps.Dal,
botID: deps.BotID,
redis: deps.Redis,
dal: deps.Dal,
tgClient: deps.TgClient,
}
}
@ -42,5 +49,44 @@ func (wc *TgListenerWorker) Start(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
}
}
}