83 lines
2.3 KiB
Go
83 lines
2.3 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"github.com/TelegramExchange/pkg/transaction"
|
|
"github.com/TelegramExchange/tgex-backend/tg_parser/internal/domain"
|
|
)
|
|
|
|
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}
|
|
}
|
|
|
|
// Log warning if both identity fields are empty (will violate CHECK constraint)
|
|
if !usernameDTO.Valid && !inviteLinkDTO.Valid {
|
|
log.Warn().
|
|
Str("channel_id", channel.ID.String()).
|
|
Int64("telegram_id", channel.TelegramID).
|
|
Str("title", channel.Title).
|
|
Msg("UpdateChannel: both username and invite_link are empty, this may violate CHECK constraint")
|
|
}
|
|
|
|
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
|
|
}
|