fixup! feat(db): Add Epoch and Snapshot

cartesi · Aug 9, 2024 · 54e758a · 54e758a
1 parent f277774
commit 54e758a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/internal/repository/base.go b/internal/repository/base.go
@@ -268,7 +268,7 @@ func (pg *Database) InsertReport(
 func (pg *Database) InsertSnapshot(
 	ctx context.Context,
 	snapshot *Snapshot,
-) error {
+) (id uint64, _ error) {
 	query := `
 	INSERT INTO snapshot
 		(input_id,
@@ -277,20 +277,21 @@ func (pg *Database) InsertSnapshot(
 	VALUES
 		(@inputId,
 		@appAddress,
-		@uri)`
+		@uri)
+	RETURNING id`
 
 	args := pgx.NamedArgs{
 		"inputId":    snapshot.InputId,
 		"appAddress": snapshot.AppAddress,
 		"uri":        snapshot.URI,
 	}
 
-	_, err := pg.db.Exec(ctx, query, args)
+	err := pg.db.QueryRow(ctx, query, args).Scan(&id)
 	if err != nil {
-		return fmt.Errorf("%w: %w", ErrInsertRow, err)
+		return 0, fmt.Errorf("%w: %w", ErrInsertRow, err)
 	}
 
-	return nil
+	return id, nil
 }
 
 func (pg *Database) GetNodeConfig(

diff --git a/internal/repository/base_test.go b/internal/repository/base_test.go
@@ -222,8 +222,9 @@ func (s *RepositorySuite) SetupDatabase() {
 		URI:        "/some/path",
 	}
 
-	err = s.database.InsertSnapshot(s.ctx, &snapshot)
+	id, err := s.database.InsertSnapshot(s.ctx, &snapshot)
 	s.Require().Nil(err)
+	s.Require().Equal(uint64(1), id)
 }
 
 func (s *RepositorySuite) TestApplicationExists() {