fix: max id issues

This commit is contained in:
Rogee
2024-09-04 23:40:08 +08:00
parent b00217b321
commit 95647179ab
5 changed files with 89 additions and 15 deletions

View File

@@ -120,6 +120,24 @@ func (c *DBChannel) SaveMessage(ctx context.Context, msg *ChannelMessage) error
}
tbl := table.ChannelMessages
createNew := func() error {
_, 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
}
if msg.GroupID > 0 {
if err := createNew(); err != nil {
return err
}
return nil
}
cond := tbl.GroupID.EQ(Int(message.GroupID)).AND(
tbl.ChannelID.EQ(Int(message.ChannelID)),
@@ -134,16 +152,7 @@ func (c *DBChannel) SaveMessage(ctx context.Context, msg *ChannelMessage) error
if err != nil {
// 如果没有找到记录,那么插入新记录
if errors.Is(err, qrm.ErrNoRows) {
_, 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
return createNew()
}
return errors.Wrap(err, "select message")
}