highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-11-10 22:27:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

kvdb/postgres: fix context cancellation

Browse Source
This commit is contained in:
Joost Jager
2021-12-22 12:33:50 +01:00
parent 62a2f3c809
commit 0cae55c162
5 changed files with 69 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
kvdb/postgres/readwrite_cursor.go
View File

@@ -31,11 +31,13 @@ func (c *readWriteCursor) First() ([]byte, []byte) {
key []byte
value []byte
)
err := c.bucket.tx.QueryRow(
"SELECT key, value FROM "+c.bucket.table+" WHERE "+
parentSelector(c.bucket.id)+
row, cancel := c.bucket.tx.QueryRow(
"SELECT key, value FROM " + c.bucket.table + " WHERE " +
parentSelector(c.bucket.id) +
" ORDER BY key LIMIT 1",
).Scan(&key, &value)
)
defer cancel()
err := row.Scan(&key, &value)
switch {
case err == sql.ErrNoRows:
@@ -59,11 +61,13 @@ func (c *readWriteCursor) Last() ([]byte, []byte) {
key []byte
value []byte
)
err := c.bucket.tx.QueryRow(
"SELECT key, value FROM "+c.bucket.table+" WHERE "+
parentSelector(c.bucket.id)+
row, cancel := c.bucket.tx.QueryRow(
"SELECT key, value FROM " + c.bucket.table + " WHERE " +
parentSelector(c.bucket.id) +
" ORDER BY key DESC LIMIT 1",
).Scan(&key, &value)
)
defer cancel()
err := row.Scan(&key, &value)
switch {
case err == sql.ErrNoRows:
@@ -87,12 +91,14 @@ func (c *readWriteCursor) Next() ([]byte, []byte) {
key []byte
value []byte
)
err := c.bucket.tx.QueryRow(
row, cancel := c.bucket.tx.QueryRow(
"SELECT key, value FROM "+c.bucket.table+" WHERE "+
parentSelector(c.bucket.id)+
" AND key>$1 ORDER BY key LIMIT 1",
c.currKey,
).Scan(&key, &value)
)
defer cancel()
err := row.Scan(&key, &value)
switch {
case err == sql.ErrNoRows:
@@ -116,12 +122,14 @@ func (c *readWriteCursor) Prev() ([]byte, []byte) {
key []byte
value []byte
)
err := c.bucket.tx.QueryRow(
row, cancel := c.bucket.tx.QueryRow(
"SELECT key, value FROM "+c.bucket.table+" WHERE "+
parentSelector(c.bucket.id)+
" AND key<$1 ORDER BY key DESC LIMIT 1",
c.currKey,
).Scan(&key, &value)
)
defer cancel()
err := row.Scan(&key, &value)
switch {
case err == sql.ErrNoRows:
@@ -152,12 +160,14 @@ func (c *readWriteCursor) Seek(seek []byte) ([]byte, []byte) {
key []byte
value []byte
)
err := c.bucket.tx.QueryRow(
row, cancel := c.bucket.tx.QueryRow(
"SELECT key, value FROM "+c.bucket.table+" WHERE "+
parentSelector(c.bucket.id)+
" AND key>=$1 ORDER BY key LIMIT 1",
seek,
).Scan(&key, &value)
)
defer cancel()
err := row.Scan(&key, &value)
switch {
case err == sql.ErrNoRows:
@@ -180,12 +190,14 @@ func (c *readWriteCursor) Seek(seek []byte) ([]byte, []byte) {
func (c *readWriteCursor) Delete() error {
// Get first record at or after cursor.
var key []byte
err := c.bucket.tx.QueryRow(
row, cancel := c.bucket.tx.QueryRow(
"SELECT key FROM "+c.bucket.table+" WHERE "+
parentSelector(c.bucket.id)+
" AND key>=$1 ORDER BY key LIMIT 1",
c.currKey,
).Scan(&key)
)
defer cancel()
err := row.Scan(&key)
switch {
case err == sql.ErrNoRows: