package wallet
import (
- "encoding/json"
-
- log "github.com/sirupsen/logrus"
- "github.com/vapor/account"
- "github.com/vapor/asset"
"github.com/vapor/blockchain/query"
"github.com/vapor/common"
- dbm "github.com/vapor/database/leveldb"
"github.com/vapor/protocol/bc"
)
-// Store interface contains wallet storage functions.
-type Store interface {
+// WalletStorer interface contains wallet storage functions.
+type WalletStorer interface {
+ InitBatch()
+ CommitBatch()
GetAssetDefinition(*bc.AssetID) []byte
SetAssetDefinition(*bc.AssetID, []byte)
- GetRawProgramByHash(common.Hash) []byte
- GetAccount(string) []byte
- DeleteTransaction(uint64)
+ GetRawProgram(common.Hash) []byte
+ GetAccountByAccountID(string) []byte
+ DeleteTransactions(uint64)
SetTransaction(uint64, uint32, string, []byte)
DeleteUnconfirmedTransaction(string)
SetGlobalTransactionIndex(string, *bc.Hash, uint64)
SetWalletInfo([]byte)
DeleteWalletTransactions()
DeleteWalletUTXOs()
- GetAccountUTXOs(key string) []*account.UTXO
+ GetAccountUTXOs(key string) [][]byte
SetRecoveryStatus([]byte, []byte)
DeleteRecoveryStatus([]byte)
GetRecoveryStatus([]byte) []byte
}
-// LevelDBStore store wallet using leveldb
-type LevelDBStore struct {
- DB dbm.DB
-}
-
-// NewStore create new LevelDBStore struct
-func NewStore(db dbm.DB) *LevelDBStore {
- return &LevelDBStore{
- DB: db,
- }
-}
-
-// GetAssetDefinition get asset definition by assetiD
-func (store *LevelDBStore) GetAssetDefinition(assetID *bc.AssetID) []byte {
- return store.DB.Get(asset.ExtAssetKey(assetID))
-}
-
-// SetAssetDefinition set assetID and definition
-func (store *LevelDBStore) SetAssetDefinition(assetID *bc.AssetID, definition []byte) {
- store.DB.Set(asset.ExtAssetKey(assetID), definition)
-}
-
-// GetRawProgramByHash get raw program by hash
-func (store *LevelDBStore) GetRawProgramByHash(hash common.Hash) []byte {
- return store.DB.Get(account.ContractKey(hash))
-}
-
-// GetAccount get account value by account ID
-func (store *LevelDBStore) GetAccount(accountID string) []byte {
- return store.DB.Get(account.Key(accountID))
-}
-
-// DeleteTransaction delete transactions when orphan block rollback
-func (store *LevelDBStore) DeleteTransaction(height uint64) {
- tmpTx := query.AnnotatedTx{}
- batch := store.DB.NewBatch()
- txIter := store.DB.IteratorPrefix(calcDeleteKey(height))
- defer txIter.Release()
-
- for txIter.Next() {
- if err := json.Unmarshal(txIter.Value(), &tmpTx); err == nil {
- batch.Delete(calcTxIndexKey(tmpTx.ID.String()))
- }
- batch.Delete(txIter.Key())
- }
- batch.Write()
-}
-
-// SetTransaction set raw transaction by block height and tx position
-func (store *LevelDBStore) SetTransaction(height uint64, position uint32, txID string, rawTx []byte) {
- batch := store.DB.NewBatch()
- batch.Set(calcAnnotatedKey(formatKey(height, position)), rawTx)
- batch.Set(calcTxIndexKey(txID), []byte(formatKey(height, position)))
- batch.Write()
-}
-
-// DeleteUnconfirmedTransaction delete unconfirmed tx by txID
-func (store *LevelDBStore) DeleteUnconfirmedTransaction(txID string) {
- store.DB.Delete(calcUnconfirmedTxKey(txID))
-}
-
-// SetGlobalTransactionIndex set global tx index by blockhash and position
-func (store *LevelDBStore) SetGlobalTransactionIndex(globalTxID string, blockHash *bc.Hash, position uint64) {
- store.DB.Set(calcGlobalTxIndexKey(globalTxID), calcGlobalTxIndex(blockHash, position))
-}
-
-// GetStandardUTXO get standard utxo by id
-func (store *LevelDBStore) GetStandardUTXO(outid bc.Hash) []byte {
- return store.DB.Get(account.StandardUTXOKey(outid))
-}
-
-// GetTransaction get tx by tx index
-func (store *LevelDBStore) GetTransaction(txID string) ([]byte, error) {
- formatKey := store.DB.Get(calcTxIndexKey(txID))
- if formatKey == nil {
- return nil, errAccntTxIDNotFound
- }
- txInfo := store.DB.Get(calcAnnotatedKey(string(formatKey)))
- return txInfo, nil
-}
-
-// GetGlobalTransaction get global tx by txID
-func (store *LevelDBStore) GetGlobalTransaction(txID string) []byte {
- return store.DB.Get(calcGlobalTxIndexKey(txID))
-}
-
-// GetTransactions get all walletDB transactions
-func (store *LevelDBStore) GetTransactions() ([]*query.AnnotatedTx, error) {
- annotatedTxs := []*query.AnnotatedTx{}
-
- txIter := store.DB.IteratorPrefix([]byte(TxPrefix))
- defer txIter.Release()
- for txIter.Next() {
- annotatedTx := &query.AnnotatedTx{}
- if err := json.Unmarshal(txIter.Value(), &annotatedTx); err != nil {
- return nil, err
- }
- annotatedTxs = append(annotatedTxs, annotatedTx)
- }
-
- return annotatedTxs, nil
-}
-
-// GetUnconfirmedTransactions get all unconfirmed txs
-func (store *LevelDBStore) GetUnconfirmedTransactions() ([]*query.AnnotatedTx, error) {
- annotatedTxs := []*query.AnnotatedTx{}
- txIter := store.DB.IteratorPrefix([]byte(UnconfirmedTxPrefix))
- defer txIter.Release()
-
- for txIter.Next() {
- annotatedTx := &query.AnnotatedTx{}
- if err := json.Unmarshal(txIter.Value(), &annotatedTx); err != nil {
- return nil, err
- }
- annotatedTxs = append(annotatedTxs, annotatedTx)
- }
- return annotatedTxs, nil
-}
-
-// GetUnconfirmedTransaction get unconfirmed tx by txID
-func (store *LevelDBStore) GetUnconfirmedTransaction(txID string) []byte {
- return store.DB.Get(calcUnconfirmedTxKey(txID))
-}
-
-// SetUnconfirmedTransaction set unconfirmed tx by txID
-func (store *LevelDBStore) SetUnconfirmedTransaction(txID string, rawTx []byte) {
- store.DB.Set(calcUnconfirmedTxKey(txID), rawTx)
-}
-
-// DeleteStardardUTXO delete stardard utxo by outputID
-func (store *LevelDBStore) DeleteStardardUTXO(outputID bc.Hash) {
- store.DB.Delete(account.StandardUTXOKey(outputID))
-}
-
-// DeleteContractUTXO delete contract utxo by outputID
-func (store *LevelDBStore) DeleteContractUTXO(outputID bc.Hash) {
- store.DB.Delete(account.ContractUTXOKey(outputID))
-}
-
-// SetStandardUTXO set standard utxo
-func (store *LevelDBStore) SetStandardUTXO(outputID bc.Hash, data []byte) {
- store.DB.Set(account.StandardUTXOKey(outputID), data)
-}
-
-// SetContractUTXO set standard utxo
-func (store *LevelDBStore) SetContractUTXO(outputID bc.Hash, data []byte) {
- store.DB.Set(account.ContractUTXOKey(outputID), data)
-}
-
-// GetWalletInfo get wallet information
-func (store *LevelDBStore) GetWalletInfo() []byte {
- return store.DB.Get(walletKey)
-}
-
-// SetWalletInfo get wallet information
-func (store *LevelDBStore) SetWalletInfo(rawWallet []byte) {
- store.DB.Set(walletKey, rawWallet)
-}
-
-// DeleteWalletTransactions delete all txs in wallet
-func (store *LevelDBStore) DeleteWalletTransactions() {
- batch := store.DB.NewBatch()
-
- txIter := store.DB.IteratorPrefix([]byte(TxPrefix))
- defer txIter.Release()
-
- for txIter.Next() {
- batch.Delete(txIter.Key())
- }
-
- txIndexIter := store.DB.IteratorPrefix([]byte(TxIndexPrefix))
- defer txIndexIter.Release()
-
- for txIndexIter.Next() {
- batch.Delete(txIndexIter.Key())
- }
-
- batch.Write()
-}
-
-// DeleteWalletUTXOs delete all txs in wallet
-func (store *LevelDBStore) DeleteWalletUTXOs() {
- batch := store.DB.NewBatch()
- ruIter := store.DB.IteratorPrefix([]byte(account.UTXOPreFix))
- defer ruIter.Release()
- for ruIter.Next() {
- batch.Delete(ruIter.Key())
- }
-
- suIter := store.DB.IteratorPrefix([]byte(account.SUTXOPrefix))
- defer suIter.Release()
- for suIter.Next() {
- batch.Delete(suIter.Key())
- }
- batch.Write()
-}
-
-// GetAccountUTXOs get all account unspent outputs
-func (store *LevelDBStore) GetAccountUTXOs(key string) []*account.UTXO {
- accountUtxos := []*account.UTXO{}
- accountUtxoIter := store.DB.IteratorPrefix([]byte(key))
- defer accountUtxoIter.Release()
-
- for accountUtxoIter.Next() {
- accountUtxo := &account.UTXO{}
- if err := json.Unmarshal(accountUtxoIter.Value(), accountUtxo); err != nil {
- log.WithFields(log.Fields{"module": logModule, "err": err}).Warn("GetAccountUTXOs fail on unmarshal utxo")
- continue
- }
- accountUtxos = append(accountUtxos, accountUtxo)
-
- }
- return accountUtxos
-}
-
-// SetRecoveryStatus set recovery status
-func (store *LevelDBStore) SetRecoveryStatus(recoveryKey, rawStatus []byte) {
- store.DB.Set(recoveryKey, rawStatus)
-}
-
-// DeleteRecoveryStatus delete recovery status
-func (store *LevelDBStore) DeleteRecoveryStatus(recoveryKey []byte) {
- store.DB.Delete(recoveryKey)
-}
-
-// GetRecoveryStatus delete recovery status
-func (store *LevelDBStore) GetRecoveryStatus(recoveryKey []byte) []byte {
- return store.DB.Get(recoveryKey)
-}