15561477203

Committed 10 Jun 2025 01:54PM UTC coverage: 58.351% (-10.1%) from 68.487%

Build # 15561477203

Build Type

Pull #9356

github

Committed by

web-flow

Commit Message

Merge 6440b25db into c6d6d4c0b

Pull Request Pull Request #9356: lnrpc: add incoming/outgoing channel ids filter to forwarding history request

Run Details

33 of 36 new or added lines in 2 files covered. (91.67%)

28366 existing lines in 455 files now uncovered.

97715 of 167461 relevant lines covered (58.35%)

1.81 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/watchtower/wtdb/migration5/client_db.go

package migration5

import (
        "encoding/binary"
        "errors"

        "github.com/lightningnetwork/lnd/kvdb"
)

var (
        // cTowerBkt is a top-level bucket storing:
        //    tower-id -> encoded Tower.
        cTowerBkt = []byte("client-tower-bucket")

        // cTowerIDToSessionIDIndexBkt is a top-level bucket storing:
        //         tower-id -> session-id -> 1
        cTowerIDToSessionIDIndexBkt = []byte(
                "client-tower-to-session-index-bucket",
        )

        // ErrUninitializedDB signals that top-level buckets for the database
        // have not been initialized.
        ErrUninitializedDB = errors.New("db not initialized")

        // byteOrder is the default endianness used when serializing integers.
        byteOrder = binary.BigEndian
)

// MigrateCompleteTowerToSessionIndex ensures that the tower-to-session index
// contains entries for all towers in the db. This is necessary because
// migration1 only created entries in the index for towers that the client had
// at least one session with. This migration thus makes sure that there is
// always a tower-to-sessions index entry for a tower even if there are no
// sessions with that tower.
func MigrateCompleteTowerToSessionIndex(tx kvdb.RwTx) error {
        log.Infof("Migrating the tower client db to ensure that there is an " +
                "entry in the towerID-to-sessionID index for every tower in " +
                "the db")

        // First, we collect all the towers that we should add an entry for in
        // the index.
        towerIDs, err := listTowerIDs(tx)
        if err != nil {
                return err
        }

        // Create a new top-level bucket for the index if it does not yet exist.
        indexBkt, err := tx.CreateTopLevelBucket(cTowerIDToSessionIDIndexBkt)
        if err != nil {
                return err
        }

        // Finally, ensure that there is an entry in the tower-to-session index
        // for each of our towers.
        for _, id := range towerIDs {
                // Create a sub-bucket using the tower ID.
                _, err := indexBkt.CreateBucketIfNotExists(id.Bytes())
                if err != nil {
                        return err
                }
        }

        return nil
}

// listTowerIDs iterates through the cTowerBkt and collects a list of all the
// TowerIDs.
func listTowerIDs(tx kvdb.RTx) ([]*TowerID, error) {
        var ids []*TowerID
        towerBucket := tx.ReadBucket(cTowerBkt)
        if towerBucket == nil {
                return nil, ErrUninitializedDB
        }

        err := towerBucket.ForEach(func(towerIDBytes, _ []byte) error {
                id, err := TowerIDFromBytes(towerIDBytes)
                if err != nil {
                        return err
                }

                ids = append(ids, &id)

                return nil
        })
        if err != nil {
                return nil, err
        }

        return ids, nil
}

1	package migration5
2
3	import (
4	"encoding/binary"
5	"errors"
6
7	"github.com/lightningnetwork/lnd/kvdb"
8	)
9
10	var (
11	// cTowerBkt is a top-level bucket storing:
12	// tower-id -> encoded Tower.
13	cTowerBkt = []byte("client-tower-bucket")
14
15	// cTowerIDToSessionIDIndexBkt is a top-level bucket storing:
16	// tower-id -> session-id -> 1
17	cTowerIDToSessionIDIndexBkt = []byte(
18	"client-tower-to-session-index-bucket",
19	)
20
21	// ErrUninitializedDB signals that top-level buckets for the database
22	// have not been initialized.
23	ErrUninitializedDB = errors.New("db not initialized")
24
25	// byteOrder is the default endianness used when serializing integers.
26	byteOrder = binary.BigEndian
27	)
28
29	// MigrateCompleteTowerToSessionIndex ensures that the tower-to-session index
30	// contains entries for all towers in the db. This is necessary because
31	// migration1 only created entries in the index for towers that the client had
32	// at least one session with. This migration thus makes sure that there is
33	// always a tower-to-sessions index entry for a tower even if there are no
34	// sessions with that tower.
UNCOV 35	func MigrateCompleteTowerToSessionIndex(tx kvdb.RwTx) error {	×
UNCOV 36	log.Infof("Migrating the tower client db to ensure that there is an " +	×
UNCOV 37	"entry in the towerID-to-sessionID index for every tower in " +	×
UNCOV 38	"the db")	×
UNCOV 39		×
UNCOV 40	// First, we collect all the towers that we should add an entry for in	×
UNCOV 41	// the index.	×
UNCOV 42	towerIDs, err := listTowerIDs(tx)	×
UNCOV 43	if err != nil {	×
UNCOV 44	return err	×
UNCOV 45	}	×
46
47	// Create a new top-level bucket for the index if it does not yet exist.
UNCOV 48	indexBkt, err := tx.CreateTopLevelBucket(cTowerIDToSessionIDIndexBkt)	×
UNCOV 49	if err != nil {	×
50	return err	×
51	}	×
52
53	// Finally, ensure that there is an entry in the tower-to-session index
54	// for each of our towers.
UNCOV 55	for _, id := range towerIDs {	×
UNCOV 56	// Create a sub-bucket using the tower ID.	×
UNCOV 57	_, err := indexBkt.CreateBucketIfNotExists(id.Bytes())	×
UNCOV 58	if err != nil {	×
59	return err	×
60	}	×
61	}
62
UNCOV 63	return nil	×
64	}
65
66	// listTowerIDs iterates through the cTowerBkt and collects a list of all the
67	// TowerIDs.
UNCOV 68	func listTowerIDs(tx kvdb.RTx) ([]*TowerID, error) {	×
UNCOV 69	var ids []*TowerID	×
UNCOV 70	towerBucket := tx.ReadBucket(cTowerBkt)	×
UNCOV 71	if towerBucket == nil {	×
72	return nil, ErrUninitializedDB	×
73	}	×
74
UNCOV 75	err := towerBucket.ForEach(func(towerIDBytes, _ []byte) error {	×
UNCOV 76	id, err := TowerIDFromBytes(towerIDBytes)	×
UNCOV 77	if err != nil {	×
UNCOV 78	return err	×
UNCOV 79	}	×
80
UNCOV 81	ids = append(ids, &id)	×
UNCOV 82		×
UNCOV 83	return nil	×
84	})
UNCOV 85	if err != nil {	×
UNCOV 86	return nil, err	×
UNCOV 87	}	×
88
UNCOV 89	return ids, nil	×
90	}

lightningnetwork / lnd / 15561477203

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous