wader/fq
1
1
Fork 0
mirror of https://github.com/wader/fq.git synced 2024-11-26 21:55:57 +03:00
Code Issues Projects Releases Wiki Activity

postgres: add page arg in pg_btree, change args names in pg_heap

Browse Source
This commit is contained in:
Pavel Safonov 2022-12-22 11:22:02 +03:00
parent 0ea20e6827
commit 97bbc22a1d
5 changed files with 26 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
format/format.go
View File

@ -333,6 +333,10 @@ type PostgresIn struct {
type PostgresHeapIn struct { type PostgresHeapIn struct {
Flavour string `doc:"PostgreSQL flavour: postgres14, pgproee14.., postgres10"` Flavour string `doc:"PostgreSQL flavour: postgres14, pgproee14.., postgres10"`
PageNumber int `doc:"First page number in file, default is 0"` Page int `doc:"First page number in file, default is 0"`
SegmentNumber int `doc:"Segment file number (16790.1 is 1), default is 0"` Segment int `doc:"Segment file number (16790.1 is 1), default is 0"`
}
type PostgresBTreeIn struct {
Page int `doc:"First page number in file, default is 0"`
} }

8
format/postgres/common/pg_btree/postgres/pg_btree.go
View File

@ -1,6 +1,7 @@
package postgres package postgres
import ( import (
"github.com/wader/fq/format"
"github.com/wader/fq/format/postgres/common" "github.com/wader/fq/format/postgres/common"
"github.com/wader/fq/format/postgres/common/pg_heap/postgres" "github.com/wader/fq/format/postgres/common/pg_heap/postgres"
"github.com/wader/fq/pkg/decode" "github.com/wader/fq/pkg/decode"
@ -66,8 +67,9 @@ const (
// IndexTupleData *IndexTuple; // IndexTupleData *IndexTuple;
/* total size (bytes): 8 */ /* total size (bytes): 8 */
func DecodePgBTree(d *decode.D) any { func DecodePgBTree(d *decode.D, args format.PostgresBTreeIn) any {
btree := &BTree{ btree := &BTree{
Args: args,
PageSize: common.PageSize, PageSize: common.PageSize,
} }
decodeBTreePages(btree, d) decodeBTreePages(btree, d)
@ -75,13 +77,13 @@ func DecodePgBTree(d *decode.D) any {
} }
type BTree struct { type BTree struct {
Args format.PostgresBTreeIn
PageSize uint64 PageSize uint64
page *postgres.HeapPage page *postgres.HeapPage
} }
func decodeBTreePages(btree *BTree, d *decode.D) { func decodeBTreePages(btree *BTree, d *decode.D) {
for i := 0; ; i++ { for i := btree.Args.Page; ; i++ {
page := &postgres.HeapPage{} page := &postgres.HeapPage{}
if btree.page != nil { if btree.page != nil {
// use prev page // use prev page

2
format/postgres/common/pg_heap/postgres/pg_heap.go
View File

@ -144,7 +144,7 @@ func Decode(heap *Heap, d *decode.D) any {
} }
func decodeHeapPages(heap *Heap, d *decode.D) { func decodeHeapPages(heap *Heap, d *decode.D) {
blockNumber := uint32(heap.Args.PageNumber + heap.Args.SegmentNumber*common.RelSegSize) blockNumber := uint32(heap.Args.Page + heap.Args.Segment*common.RelSegSize)
count := int64(0) count := int64(0)
for { for {
if end, _ := d.TryEnd(); end { if end, _ := d.TryEnd(); end {

10
format/postgres/pg_btree.go
View File

@ -18,6 +18,9 @@ func init() {
Name: format.PG_BTREE, Name: format.PG_BTREE,
Description: "PostgreSQL btree index file", Description: "PostgreSQL btree index file",
DecodeFn: decodePgBTree, DecodeFn: decodePgBTree,
DecodeInArg: format.PostgresBTreeIn{
Page: 0,
},
RootArray: true, RootArray: true,
RootName: "pages", RootName: "pages",
}) })
@ -25,6 +28,9 @@ func init() {
} }
func decodePgBTree(d *decode.D, in any) any { func decodePgBTree(d *decode.D, in any) any {
d.Endian = decode.LittleEndian pgIn, ok := in.(format.PostgresBTreeIn)
return postgres.DecodePgBTree(d) if !ok {
d.Fatalf("DecodeInArg must be PostgresBTreeIn!\n")
}
return postgres.DecodePgBTree(d, pgIn)
} }

4
format/postgres/pg_heap.go
View File

@ -24,8 +24,8 @@ func init() {
DecodeFn: decodePgheap, DecodeFn: decodePgheap,
DecodeInArg: format.PostgresHeapIn{ DecodeInArg: format.PostgresHeapIn{
Flavour: PG_FLAVOUR_POSTGRES14, Flavour: PG_FLAVOUR_POSTGRES14,
PageNumber: 0, Page: 0,
SegmentNumber: 0, Segment: 0,
}, },
RootArray: true, RootArray: true,
RootName: "pages", RootName: "pages",