fix: max id issues
This commit is contained in:
@@ -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")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user