This commit is contained in:
Rogee
2024-09-02 19:45:16 +08:00
parent 548b9109eb
commit 399dbe4264
6 changed files with 23 additions and 3 deletions

View File

@@ -8,6 +8,7 @@ import (
type ChannelMessage struct {
ID int
GroupID int64
Message string
Medias []ChannelMessageMedia
PublishAt time.Time
@@ -22,6 +23,11 @@ func NewChannelMessage(id, ts int) *ChannelMessage {
return &ChannelMessage{ID: id, PublishAt: time.Unix(int64(ts), 0)}
}
func (c *ChannelMessage) WithGroupID(groupID int64) *ChannelMessage {
c.GroupID = groupID
return c
}
func (c *ChannelMessage) WithMessage(message string) *ChannelMessage {
c.Message = message
return c

View File

@@ -59,6 +59,9 @@ func (t *TClient) Channel(ctx context.Context, channel *tg.Channel, cfg *DBChann
defer cfg.SaveMessage(ctx, channelMessage)
channelMessage.WithMessage(msg.GetMessage())
if grpID, ok := msg.GetGroupedID(); ok {
channelMessage.WithGroupID(grpID)
}
if mediaClass, ok := msg.GetMedia(); ok {
switch mediaClass.(type) {

View File

@@ -113,6 +113,7 @@ func (c *DBChannel) Update(ctx context.Context, offsetID int) error {
func (c *DBChannel) SaveMessage(ctx context.Context, msg *ChannelMessage) error {
message := &model.ChannelMessages{
ChannelID: c.UUID,
GroupID: msg.GroupID,
UUID: int64(msg.ID),
Content: lo.ToPtr(msg.Message),
Media: msg.GetMedia(),