From bc14b51a1bc305c163715fffaab7255f62649f9c Mon Sep 17 00:00:00 2001
From: Daniil Porokhnin <dkharmsd@gmail.com>
Date: Thu, 14 May 2026 13:43:51 +0300
Subject: [PATCH] perf: find `TokenTable` index once

---
 frac/sealed/token/table_loader.go | 44 ++++++++++++++++++-------------
 1 file changed, 26 insertions(+), 18 deletions(-)

diff --git a/frac/sealed/token/table_loader.go b/frac/sealed/token/table_loader.go
index cd04830b..5cca1a21 100644
--- a/frac/sealed/token/table_loader.go
+++ b/frac/sealed/token/table_loader.go
@@ -3,6 +3,7 @@ package token
 import (
 	"encoding/binary"
 	"slices"
+	"sync"
 	"unsafe"
 
 	"go.uber.org/zap"
@@ -22,7 +23,9 @@ type TableLoader struct {
 	reader *storage.IndexReader
 	cache  *cache.Cache[Table]
 
-	i   uint32
+	once       sync.Once
+	tableIndex uint32
+
 	buf []byte
 }
 
@@ -47,6 +50,8 @@ func (l *TableLoader) Load() Table {
 			err    error
 		)
 
+		l.advanceToTable()
+
 		if l.isLegacy {
 			blocks, err = l.loadBlocksLegacy()
 		} else {
@@ -98,29 +103,24 @@ func TableFromBlocks(blocks []TableBlock) Table {
 }
 
 func (l *TableLoader) readHeader() storage.IndexBlockHeader {
-	h, e := l.reader.GetBlockHeader(l.i)
+	h, e := l.reader.GetBlockHeader(l.tableIndex)
 	if e != nil {
 		logger.Panic("error reading block header", zap.Error(e))
 	}
-	l.i++
+	l.tableIndex++
 	return h
 }
 
 func (l *TableLoader) readBlock() ([]byte, error) {
-	block, _, err := l.reader.ReadIndexBlock(l.i, l.buf)
+	block, _, err := l.reader.ReadIndexBlock(l.tableIndex, l.buf)
 	l.buf = block
-	l.i++
+	l.tableIndex++
 	return block, err
 }
 
 func (l *TableLoader) loadBlocksLegacy() ([]TableBlock, error) {
-	l.i = 1 // Skip info block immediately.
-
-	for h := l.readHeader(); h.Len() > 0; h = l.readHeader() {
-		// Skip token blocks, go for token table.
-	}
-
 	blocks := make([]TableBlock, 0)
+
 	for blockData, err := l.readBlock(); len(blockData) > 0; blockData, err = l.readBlock() {
 		if err != nil {
 			return nil, err
@@ -134,19 +134,13 @@ func (l *TableLoader) loadBlocksLegacy() ([]TableBlock, error) {
 }
 
 func (l *TableLoader) loadBlocks() ([]TableBlock, error) {
-	l.i = 0
-
 	blocksCount, err := l.reader.BlocksCount()
 	if err != nil {
 		return nil, err
 	}
 
-	for h := l.readHeader(); h.Len() > 0; h = l.readHeader() {
-		// Skip token blocks, go for token table.
-	}
-
 	var blocks []TableBlock
-	for l.i < uint32(blocksCount) {
+	for l.tableIndex < uint32(blocksCount) {
 		data, err := l.readBlock()
 		if err != nil {
 			return nil, err
@@ -161,6 +155,20 @@ func (l *TableLoader) loadBlocks() ([]TableBlock, error) {
 	return blocks, nil
 }
 
+func (l *TableLoader) advanceToTable() {
+	l.once.Do(func() {
+		// This is correct for both legacy and non-legacy sealed fractions:
+		// 	- in legacy fractions we have following layout: [info][token][separator][token-table][...];
+		//	- in non-legacy fraction we have following layout: [token][separator][token-table];
+		// As you can see, it is safe to start from 0-th block in both cases.
+		l.tableIndex = 0
+
+		for h := l.readHeader(); h.Len() > 0; h = l.readHeader() {
+			// Skip token blocks, go for token table.
+		}
+	})
+}
+
 // TableBlock represents how token.Table is stored on disk
 type TableBlock struct {
 	FieldsTables []FieldTable