github.com/amazechain/amc@v0.1.3/modules/state/db_state_writer.go (about)

     1  // Copyright 2023 The AmazeChain Authors
     2  // This file is part of the AmazeChain library.
     3  //
     4  // The AmazeChain library is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU Lesser General Public License as published by
     6  // the Free Software Foundation, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // The AmazeChain library is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    12  // GNU Lesser General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU Lesser General Public License
    15  // along with the AmazeChain library. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package state
    18  
    19  import (
    20  	"bytes"
    21  	"fmt"
    22  	"github.com/amazechain/amc/common/account"
    23  	"github.com/amazechain/amc/common/crypto"
    24  	"github.com/amazechain/amc/common/types"
    25  	"github.com/amazechain/amc/modules"
    26  	"math"
    27  
    28  	"github.com/amazechain/amc/modules/changeset"
    29  
    30  	"github.com/RoaringBitmap/roaring/roaring64"
    31  
    32  	"github.com/amazechain/amc/modules/ethdb/bitmapdb"
    33  	"github.com/ledgerwatch/erigon-lib/kv"
    34  )
    35  
    36  func originalAccountData(original *account.StateAccount, omitHashes bool) []byte {
    37  	var originalData []byte
    38  	if !original.Initialised {
    39  		originalData = []byte{}
    40  	} else if omitHashes {
    41  		testAcc := original.SelfCopy()
    42  		copy(testAcc.CodeHash[:], emptyCodeHash)
    43  		testAcc.Root = crypto.Keccak256Hash(nil)
    44  		originalDataLen := testAcc.EncodingLengthForStorage()
    45  		originalData = make([]byte, originalDataLen)
    46  		testAcc.EncodeForStorage(originalData)
    47  	} else {
    48  		originalDataLen := original.EncodingLengthForStorage()
    49  		originalData = make([]byte, originalDataLen)
    50  		original.EncodeForStorage(originalData)
    51  	}
    52  	return originalData
    53  }
    54  
    55  func writeIndex(blocknum uint64, changes *changeset.ChangeSet, bucket string, changeDb kv.RwTx) error {
    56  	buf := bytes.NewBuffer(nil)
    57  	for _, change := range changes.Changes {
    58  		k := modules.CompositeKeyWithoutIncarnation(change.Key)
    59  
    60  		index, err := bitmapdb.Get64(changeDb, bucket, k, math.MaxUint32, math.MaxUint32)
    61  		if err != nil {
    62  			return fmt.Errorf("find chunk failed: %w", err)
    63  		}
    64  		index.Add(blocknum)
    65  		if err = bitmapdb.WalkChunkWithKeys64(k, index, bitmapdb.ChunkLimit, func(chunkKey []byte, chunk *roaring64.Bitmap) error {
    66  			buf.Reset()
    67  			if _, err = chunk.WriteTo(buf); err != nil {
    68  				return err
    69  			}
    70  			return changeDb.Put(bucket, chunkKey, types.CopyBytes(buf.Bytes()))
    71  		}); err != nil {
    72  			return err
    73  		}
    74  	}
    75  
    76  	return nil
    77  }