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

3
internal/db_channel.go
View File

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