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 ( 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
}
}
} }