rogee/tg_exporter
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: support export media

Browse Source
This commit is contained in:
Rogee
2024-09-05 14:28:11 +08:00
parent 25234937d4
commit 7d1c9764b5
3 changed files with 52 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
database/telegram_resource/public/model/channels.go
View File

@@ -20,4 +20,5 @@ type Channels struct {
UpdatedAt *time.Time
Offset int64
MinID int64
ExportMedia bool
}

7
database/telegram_resource/public/table/channels.go
View File

@@ -25,6 +25,7 @@ type channelsTable struct {
UpdatedAt postgres.ColumnTimestampz
Offset postgres.ColumnInteger
MinID postgres.ColumnInteger
ExportMedia postgres.ColumnBool
AllColumns postgres.ColumnList
MutableColumns postgres.ColumnList
@@ -73,8 +74,9 @@ func newChannelsTableImpl(schemaName, tableName, alias string) channelsTable {
UpdatedAtColumn = postgres.TimestampzColumn("updated_at")
OffsetColumn = postgres.IntegerColumn("offset")
MinIDColumn = postgres.IntegerColumn("min_id")
allColumns = postgres.ColumnList{IDColumn, UUIDColumn, UsernameColumn, TitleColumn, CreatedAtColumn, UpdatedAtColumn, OffsetColumn, MinIDColumn}
mutableColumns = postgres.ColumnList{UUIDColumn, UsernameColumn, TitleColumn, CreatedAtColumn, UpdatedAtColumn, OffsetColumn, MinIDColumn}
ExportMediaColumn = postgres.BoolColumn("export_media")
allColumns = postgres.ColumnList{IDColumn, UUIDColumn, UsernameColumn, TitleColumn, CreatedAtColumn, UpdatedAtColumn, OffsetColumn, MinIDColumn, ExportMediaColumn}
mutableColumns = postgres.ColumnList{UUIDColumn, UsernameColumn, TitleColumn, CreatedAtColumn, UpdatedAtColumn, OffsetColumn, MinIDColumn, ExportMediaColumn}
)
return channelsTable{
@@ -89,6 +91,7 @@ func newChannelsTableImpl(schemaName, tableName, alias string) channelsTable {
UpdatedAt: UpdatedAtColumn,
Offset: OffsetColumn,
MinID: MinIDColumn,
ExportMedia: ExportMediaColumn,
AllColumns: allColumns,
MutableColumns: mutableColumns,

3
internal/db_channel.go
View File

@@ -24,6 +24,7 @@ type DBChannel struct {
Username string
Title string
MinID int
ExportMedia bool
}
func NewDBChannel(uuid int64, username, title string) *DBChannel {
@@ -73,6 +74,7 @@ func (c *DBChannel) Get(ctx context.Context) error {
Title: c.Title,
MinID: 0,
Offset: 0,
ExportMedia: false,
CreatedAt: lo.ToPtr(time.Now()),
UpdatedAt: lo.ToPtr(time.Now()),
}
@@ -85,6 +87,7 @@ func (c *DBChannel) Get(ctx context.Context) error {
}
}
c.MinID = int(m.MinID)
c.ExportMedia = m.ExportMedia
return nil
}