125 lines
3.3 KiB
Go
125 lines
3.3 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/TelegramExchange/pkg/transaction"
|
|
"github.com/TelegramExchange/tgex-backend/tg_parser/internal/domain"
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
)
|
|
|
|
func (d *Database) UpdateChannelIfNotAccessible(ctx context.Context, channel domain.Channel) error {
|
|
query := `UPDATE channel
|
|
SET telegram_id = $2,
|
|
username = $3,
|
|
title = $4,
|
|
access_hash = $5,
|
|
pts = 0,
|
|
is_accessible = TRUE,
|
|
invite_link = $6,
|
|
updated_at = CURRENT_TIMESTAMP
|
|
WHERE telegram_id = $1
|
|
AND is_accessible = FALSE;`
|
|
|
|
txOrPool := transaction.TryExtractTX(ctx)
|
|
|
|
username := strings.TrimSpace(channel.Username)
|
|
var usernameDTO pgtype.Text
|
|
if username != "" {
|
|
usernameDTO = pgtype.Text{String: username, Valid: true}
|
|
} else {
|
|
usernameDTO = pgtype.Text{Valid: false}
|
|
}
|
|
|
|
inviteLink := strings.TrimSpace(channel.InviteLink)
|
|
var inviteLinkDTO pgtype.Text
|
|
if inviteLink != "" {
|
|
inviteLinkDTO = pgtype.Text{String: inviteLink, Valid: true}
|
|
} else {
|
|
inviteLinkDTO = pgtype.Text{Valid: false}
|
|
}
|
|
|
|
result, err := txOrPool.Exec(ctx, query,
|
|
channel.TelegramID,
|
|
pgtype.Int8{Int64: channel.TelegramID, Valid: true},
|
|
usernameDTO,
|
|
pgtype.Text{String: channel.Title, Valid: true},
|
|
pgtype.Int8{Int64: channel.AccessHash, Valid: true},
|
|
inviteLinkDTO,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("txOrPool.Exec: %w", err)
|
|
}
|
|
|
|
// Log if no rows were updated (channel either doesn't exist or is already accessible)
|
|
rowsAffected := result.RowsAffected()
|
|
if rowsAffected == 0 {
|
|
// Channel doesn't exist or is already accessible - this is fine
|
|
return nil
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Database) UpdateChannel(ctx context.Context, channel domain.Channel) error {
|
|
query := `UPDATE channel
|
|
SET telegram_id = $2,
|
|
username = $3,
|
|
title = $4,
|
|
access_hash = $5,
|
|
pts = $6,
|
|
is_accessible = $7,
|
|
invite_link = $8,
|
|
updated_at = CURRENT_TIMESTAMP
|
|
WHERE id = $1;`
|
|
|
|
txOrPool := transaction.TryExtractTX(ctx)
|
|
|
|
username := strings.TrimSpace(channel.Username)
|
|
var usernameDTO pgtype.Text
|
|
if username != "" {
|
|
usernameDTO = pgtype.Text{String: username, Valid: true}
|
|
} else {
|
|
usernameDTO = pgtype.Text{Valid: false}
|
|
}
|
|
|
|
inviteLink := strings.TrimSpace(channel.InviteLink)
|
|
var inviteLinkDTO pgtype.Text
|
|
if inviteLink != "" {
|
|
inviteLinkDTO = pgtype.Text{String: inviteLink, Valid: true}
|
|
} else {
|
|
inviteLinkDTO = pgtype.Text{Valid: false}
|
|
}
|
|
|
|
dto := updateChannelDTO{
|
|
ID: pgtype.UUID{Bytes: channel.ID, Valid: true},
|
|
TelegramID: pgtype.Int8{Int64: channel.TelegramID, Valid: true},
|
|
Username: usernameDTO,
|
|
Title: pgtype.Text{String: channel.Title, Valid: true},
|
|
AccessHash: pgtype.Int8{Int64: channel.AccessHash, Valid: true},
|
|
Pts: pgtype.Int4{Int32: int32(channel.Pts), Valid: true},
|
|
IsAccessible: pgtype.Bool{Bool: channel.IsAccessible, Valid: true},
|
|
InviteLink: inviteLinkDTO,
|
|
}
|
|
|
|
_, err := txOrPool.Exec(ctx, query, dto.ID, dto.TelegramID, dto.Username, dto.Title, dto.AccessHash, dto.Pts, dto.IsAccessible, dto.InviteLink)
|
|
if err != nil {
|
|
return fmt.Errorf("txOrPool.Exec: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type updateChannelDTO struct {
|
|
ID pgtype.UUID
|
|
TelegramID pgtype.Int8
|
|
Username pgtype.Text
|
|
Title pgtype.Text
|
|
AccessHash pgtype.Int8
|
|
Pts pgtype.Int4
|
|
IsAccessible pgtype.Bool
|
|
InviteLink pgtype.Text
|
|
}
|