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

store messages

Browse Source
This commit is contained in:
Rogee
2024-09-02 18:47:12 +08:00
parent 98cd1a0365
commit 32b53c0b18
3 changed files with 26 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
internal/db_channel.go
View File

@@ -2,7 +2,6 @@ package internal
import (
"context"
"database/sql"
"fmt"
"os"
"path/filepath"
@@ -12,6 +11,8 @@ import (
"exporter/database/telegram_resource/public/table"
. "github.com/go-jet/jet/v2/postgres"
"github.com/go-jet/jet/v2/qrm"
"github.com/lib/pq"
"github.com/pkg/errors"
"github.com/samber/lo"
)
@@ -58,13 +59,13 @@ func (c *DBChannel) Asset(assetID int64, ext string) string {
func (c *DBChannel) Get(ctx context.Context) error {
tbl := table.Channels
var m *model.Channels
var m model.Channels
err := tbl.SELECT(tbl.AllColumns).QueryContext(ctx, db, &m)
err := tbl.SELECT(tbl.AllColumns).WHERE(tbl.UUID.EQ(Int64(c.UUID))).QueryContext(ctx, db, &m)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
if errors.Is(err, qrm.ErrNoRows) {
// create new channel with default value
m = &model.Channels{
m = model.Channels{
UUID: c.UUID,
Username: c.Username,
Title: c.Title,
@@ -74,7 +75,7 @@ func (c *DBChannel) Get(ctx context.Context) error {
UpdatedAt: lo.ToPtr(time.Now()),
}
if _, err := tbl.INSERT(tbl.AllColumns).MODEL(m).ExecContext(ctx, db); err != nil {
if _, err := tbl.INSERT(tbl.AllColumns.Except(tbl.ID)).MODEL(m).ExecContext(ctx, db); err != nil {
return errors.Wrap(err, "insert channel")
}
} else {
@@ -95,10 +96,13 @@ func (c *DBChannel) Update(ctx context.Context, offsetID int) error {
}
tbl := table.Channels
_, err := tbl.UPDATE().SET(
tbl.Offset.SET(Int(int64(c.Offset))),
tbl.MinID.SET(Int(int64(c.MinID))),
).ExecContext(ctx, db)
_, err := tbl.UPDATE().
SET(
tbl.Offset.SET(Int64(int64(c.Offset))),
tbl.MinID.SET(Int64(int64(c.MinID))),
).
WHERE(tbl.UUID.EQ(Int64(c.UUID))).
ExecContext(ctx, db)
if err != nil {
return errors.Wrap(err, "update channel")
}
@@ -118,8 +122,13 @@ func (c *DBChannel) SaveMessage(ctx context.Context, msg *ChannelMessage) error
tbl := table.ChannelMessages
_, err := tbl.INSERT(tbl.AllColumns).MODEL(message).ExecContext(ctx, db)
_, err := tbl.INSERT(tbl.AllColumns.Except(tbl.ID)).MODEL(message).ExecContext(ctx, db)
if err != nil {
if e, ok := err.(*pq.Error); ok {
if e.Code == "23505" {
return nil
}
}
return errors.Wrap(err, "insert message")
}
return nil