From 21e92e5dea4d8efc65403eeaee91b32856b86cb6 Mon Sep 17 00:00:00 2001 From: Luke Shumaker Date: Fri, 3 Feb 2023 13:04:17 -0700 Subject: btrfsitem: Add `Free` and `CloneItem` methods to Items --- lib/btrfs/btrfsitem/item_chunk.go | 28 ++++++++++++++++++++++------ 1 file changed, 22 insertions(+), 6 deletions(-) (limited to 'lib/btrfs/btrfsitem/item_chunk.go') diff --git a/lib/btrfs/btrfsitem/item_chunk.go b/lib/btrfs/btrfsitem/item_chunk.go index e0776ad..2280a0b 100644 --- a/lib/btrfs/btrfsitem/item_chunk.go +++ b/lib/btrfs/btrfsitem/item_chunk.go @@ -15,7 +15,7 @@ import ( // // key.objectid = BTRFS_FIRST_CHUNK_TREE_OBJECTID // key.offset = logical_addr -type Chunk struct { // CHUNK_ITEM=228 +type Chunk struct { // complex CHUNK_ITEM=228 Head ChunkHeader Stripes []ChunkStripe } @@ -60,20 +60,36 @@ func (chunk Chunk) Mappings(key btrfsprim.Key) []btrfsvol.Mapping { return ret } +var chunkStripePool containers.SlicePool[ChunkStripe] + +func (chunk *Chunk) Free() { + for i := range chunk.Stripes { + chunk.Stripes[i] = ChunkStripe{} + } + chunkStripePool.Put(chunk.Stripes) + *chunk = Chunk{} + chunkPool.Put(chunk) +} + +func (chunk Chunk) Clone() Chunk { + ret := chunk + ret.Stripes = chunkStripePool.Get(len(chunk.Stripes)) + copy(ret.Stripes, chunk.Stripes) + return ret +} + func (chunk *Chunk) UnmarshalBinary(dat []byte) (int, error) { n, err := binstruct.Unmarshal(dat, &chunk.Head) if err != nil { return n, err } - chunk.Stripes = make([]ChunkStripe, 0, chunk.Head.NumStripes) - for i := 0; i < int(chunk.Head.NumStripes); i++ { - var stripe ChunkStripe - _n, err := binstruct.Unmarshal(dat[n:], &stripe) + chunk.Stripes = chunkStripePool.Get(int(chunk.Head.NumStripes)) + for i := range chunk.Stripes { + _n, err := binstruct.Unmarshal(dat[n:], &chunk.Stripes[i]) n += _n if err != nil { return n, err } - chunk.Stripes = append(chunk.Stripes, stripe) } return n, nil } -- cgit v1.2.3-2-g168b