Skip to content

Commit eab45d8

Browse files
Improved naming from review
Signed-off-by: Peter Broadhurst <[email protected]>
1 parent 1d98b48 commit eab45d8

File tree

6 files changed

+26
-26
lines changed

6 files changed

+26
-26
lines changed

internal/database/sqlcommon/message_sql.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ func (s *SQLCommon) UpsertMessage(ctx context.Context, message *fftypes.Message,
184184
}
185185

186186
// In SQL update+bump is a delete+insert within a TX
187-
func (s *SQLCommon) UpdateAndBumpMessage(ctx context.Context, message *fftypes.Message) (err error) {
187+
func (s *SQLCommon) ReplaceMessage(ctx context.Context, message *fftypes.Message) (err error) {
188188
ctx, tx, autoCommit, err := s.beginOrUseTx(ctx)
189189
if err != nil {
190190
return err

internal/database/sqlcommon/message_sql_test.go

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ func TestUpsertE2EWithDB(t *testing.T) {
207207

208208
// Bump and Update - this is for a ready transition
209209
msgUpdated.State = fftypes.MessageStateReady
210-
err = s.UpdateAndBumpMessage(context.Background(), msgUpdated)
210+
err = s.ReplaceMessage(context.Background(), msgUpdated)
211211
assert.NoError(t, err)
212212
msgRead, err = s.GetMessageByID(ctx, msgUpdated.Header.ID)
213213
msgJson, _ = json.Marshal(&msgUpdated)
@@ -285,34 +285,34 @@ func TestUpsertMessageFailCommit(t *testing.T) {
285285
assert.NoError(t, mock.ExpectationsWereMet())
286286
}
287287

288-
func TestUpdateAndBumpMessageFailBegin(t *testing.T) {
288+
func TestReplaceMessageFailBegin(t *testing.T) {
289289
s, mock := newMockProvider().init()
290290
mock.ExpectBegin().WillReturnError(fmt.Errorf("pop"))
291291
msgID := fftypes.NewUUID()
292-
err := s.UpdateAndBumpMessage(context.Background(), &fftypes.Message{Header: fftypes.MessageHeader{ID: msgID}})
292+
err := s.ReplaceMessage(context.Background(), &fftypes.Message{Header: fftypes.MessageHeader{ID: msgID}})
293293
assert.Regexp(t, "FF10114", err)
294294
assert.NoError(t, mock.ExpectationsWereMet())
295295
}
296296

297-
func TestUpdateAndBumpMessageFailDelete(t *testing.T) {
297+
func TestReplaceMessageFailDelete(t *testing.T) {
298298
s, mock := newMockProvider().init()
299299
mock.ExpectBegin()
300300
mock.ExpectExec("DELETE .*").WillReturnError(fmt.Errorf("pop"))
301301
mock.ExpectRollback()
302302
msgID := fftypes.NewUUID()
303-
err := s.UpdateAndBumpMessage(context.Background(), &fftypes.Message{Header: fftypes.MessageHeader{ID: msgID}})
303+
err := s.ReplaceMessage(context.Background(), &fftypes.Message{Header: fftypes.MessageHeader{ID: msgID}})
304304
assert.Regexp(t, "FF10118", err)
305305
assert.NoError(t, mock.ExpectationsWereMet())
306306
}
307307

308-
func TestUpdateAndBumpMessageFailInsert(t *testing.T) {
308+
func TestReplaceMessageFailInsert(t *testing.T) {
309309
s, mock := newMockProvider().init()
310310
mock.ExpectBegin()
311311
mock.ExpectExec("DELETE .*").WillReturnResult(sqlmock.NewResult(1, 1))
312312
mock.ExpectExec("INSERT .*").WillReturnError(fmt.Errorf("pop"))
313313
mock.ExpectRollback()
314314
msgID := fftypes.NewUUID()
315-
err := s.UpdateAndBumpMessage(context.Background(), &fftypes.Message{Header: fftypes.MessageHeader{ID: msgID}})
315+
err := s.ReplaceMessage(context.Background(), &fftypes.Message{Header: fftypes.MessageHeader{ID: msgID}})
316316
assert.Regexp(t, "FF10116", err)
317317
assert.NoError(t, mock.ExpectationsWereMet())
318318
}

internal/events/tokens_transferred.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ func (em *eventManager) TokensTransferred(ti tokens.Plugin, poolProtocolID strin
128128
if msg.State == fftypes.MessageStateStaged {
129129
// Message can now be sent
130130
msg.State = fftypes.MessageStateReady
131-
if err := em.database.UpdateAndBumpMessage(ctx, msg); err != nil {
131+
if err := em.database.ReplaceMessage(ctx, msg); err != nil {
132132
return err
133133
}
134134
} else {

internal/events/tokens_transferred_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -371,7 +371,7 @@ func TestTokensTransferredWithMessageSend(t *testing.T) {
371371
mdi.On("UpsertTokenTransfer", em.ctx, transfer).Return(nil).Times(2)
372372
mdi.On("UpdateTokenBalances", em.ctx, transfer).Return(nil).Times(2)
373373
mdi.On("GetMessageByID", em.ctx, mock.Anything).Return(message, nil).Times(2)
374-
mdi.On("UpdateAndBumpMessage", em.ctx, mock.MatchedBy(func(msg *fftypes.Message) bool {
374+
mdi.On("ReplaceMessage", em.ctx, mock.MatchedBy(func(msg *fftypes.Message) bool {
375375
return msg.State == fftypes.MessageStateReady
376376
})).Return(fmt.Errorf("pop"))
377377
mdi.On("InsertEvent", em.ctx, mock.MatchedBy(func(ev *fftypes.Event) bool {

mocks/databasemocks/plugin.go

Lines changed: 14 additions & 14 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pkg/database/plugin.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,9 +80,9 @@ type iMessageCollection interface {
8080
// UpdateMessage - Update message
8181
UpdateMessage(ctx context.Context, id *fftypes.UUID, update Update) (err error)
8282

83-
// UpdateAndBumpMessage updates the message, and assigns it a new sequence number at the front of the list.
83+
// ReplaceMessage updates the message, and assigns it a new sequence number at the front of the list.
8484
// A new event is raised for the message, with the new sequence number - as if it was brand new.
85-
UpdateAndBumpMessage(ctx context.Context, message *fftypes.Message) (err error)
85+
ReplaceMessage(ctx context.Context, message *fftypes.Message) (err error)
8686

8787
// UpdateMessages - Update messages
8888
UpdateMessages(ctx context.Context, filter Filter, update Update) (err error)

0 commit comments

Comments
 (0)