github.com/amazechain/amc@v0.1.3/modules/rawdb/accessors_reward.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 rawdb
    18  
    19  import (
    20  	"fmt"
    21  	"github.com/amazechain/amc/common/types"
    22  	"github.com/holiman/uint256"
    23  
    24  	"github.com/amazechain/amc/modules"
    25  	"github.com/ledgerwatch/erigon-lib/kv"
    26  )
    27  
    28  // PutAccountReward
    29  func PutAccountReward(db kv.Putter, account types.Address, val *uint256.Int) error {
    30  	key := fmt.Sprintf("account:%s", account.String())
    31  	return db.Put(modules.Reward, []byte(key), val.Bytes())
    32  }
    33  
    34  // GetAccountReward
    35  func GetAccountReward(db kv.Getter, account types.Address) (*uint256.Int, error) {
    36  	key := fmt.Sprintf("account:%s", account.String())
    37  	val, err := db.GetOne(modules.Reward, []byte(key))
    38  	if err != nil {
    39  		return uint256.NewInt(0), err
    40  	}
    41  	return uint256.NewInt(0).SetBytes(val), nil
    42  }