"github.com/vapor/common"
dbm "github.com/vapor/database/leveldb"
"github.com/vapor/protocol/bc"
- "github.com/vapor/protocol/bc/types"
)
-// DB interface contains wallet storage functions.
-type DB interface {
- GetAssetDefinitionByAssetID(*bc.AssetID) []byte
+// Store interface contains wallet storage functions.
+type Store interface {
+ GetAssetDefinition(*bc.AssetID) []byte
SetAssetDefinition(*bc.AssetID, []byte)
- GetRawProgramByAccountHash(common.Hash) []byte
- GetAccountValueByAccountID(string) []byte
- DeleteTransactionByHeight(uint64)
+ GetRawProgramByHash(common.Hash) []byte
+ GetAccount(string) []byte
+ DeleteTransaction(uint64)
SetRawTransaction(uint64, uint32, []byte)
- SaveExternalAssetDefinition(*types.Block)
SetHeightAndPostion(string, uint64, uint32)
- DeleteUnconfirmedTxByTxID(string)
+ DeleteUnconfirmedTx(string)
SetGlobalTxIndex(string, *bc.Hash, uint64)
- GetStandardUTXOByID(bc.Hash) []byte
- GetTxIndexByTxID(string) []byte
- GetTxByTxIndex([]byte) []byte
- GetGlobalTxByTxID(string) []byte
+ GetStandardUTXO(bc.Hash) []byte
+ GetTransactionIndex(string) []byte
+ GetTransaction([]byte) []byte
+ GetGlobalTransaction(string) []byte
GetTransactions() ([]*query.AnnotatedTx, error)
- GetAllUnconfirmedTxs() ([]*query.AnnotatedTx, error)
- GetUnconfirmedTxByTxID(string) []byte
- SetUnconfirmedTx(string, []byte)
- DeleteStardardUTXOByOutputID(bc.Hash)
- DeleteContractUTXOByOutputID(bc.Hash)
+ GetUnconfirmedTransactions() ([]*query.AnnotatedTx, error)
+ GetUnconfirmedTransaction(string) []byte
+ SetUnconfirmedTransaction(string, []byte)
+ DeleteStardardUTXO(bc.Hash)
+ DeleteContractUTXO(bc.Hash)
SetStandardUTXO(bc.Hash, []byte)
SetContractUTXO(bc.Hash, []byte)
GetWalletInfo() []byte
SetWalletInfo([]byte)
- DeleteAllWalletTxs()
- DeleteAllWalletUTXOs()
- GetAccountUtxos(string, string, bool, bool, []*account.UTXO) []*account.UTXO
+ DeleteWalletTransactions()
+ DeleteWalletUTXOs()
+ GetAccountUTXOs(key string) []*account.UTXO
+ SetRecoveryStatus([]byte, []byte)
+ DeleteRecoveryStatus([]byte)
+ GetRecoveryStatus([]byte) []byte
}
// LevelDBStore store wallet using leveldb
}
// NewLevelDBStore create new LevelDBStore struct
-func NewLevelDBStore(db dbm.DB) *LevelDBStore {
+func NewStore(db dbm.DB) *LevelDBStore {
return &LevelDBStore{
DB: db,
}
}
-// GetAssetDefinitionByAssetID get asset definition by assetiD
-func (store *LevelDBStore) GetAssetDefinitionByAssetID(assetID *bc.AssetID) []byte {
+// GetAssetDefinition get asset definition by assetiD
+func (store *LevelDBStore) GetAssetDefinition(assetID *bc.AssetID) []byte {
return store.DB.Get(asset.ExtAssetKey(assetID))
}
func (store *LevelDBStore) SetAssetDefinition(assetID *bc.AssetID, definition []byte) {
batch := store.DB.NewBatch()
batch.Set(asset.ExtAssetKey(assetID), definition)
+ batch.Write()
}
-// GetRawProgramByAccountHash get raw program by account hash
-func (store *LevelDBStore) GetRawProgramByAccountHash(hash common.Hash) []byte {
+// GetRawProgramByHash get raw program by hash
+func (store *LevelDBStore) GetRawProgramByHash(hash common.Hash) []byte {
return store.DB.Get(account.ContractKey(hash))
}
-// GetAccountValueByAccountID get account value by account ID
-func (store *LevelDBStore) GetAccountValueByAccountID(accountID string) []byte {
+// GetAccount get account value by account ID
+func (store *LevelDBStore) GetAccount(accountID string) []byte {
return store.DB.Get(account.Key(accountID))
}
-// DeleteTransactionByHeight delete transactions when orphan block rollback
-func (store *LevelDBStore) DeleteTransactionByHeight(height uint64) {
+// 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))
}
batch.Delete(txIter.Key())
}
+ batch.Write()
}
// SetRawTransaction set raw transaction by block height and tx position
func (store *LevelDBStore) SetRawTransaction(height uint64, position uint32, rawTx []byte) {
batch := store.DB.NewBatch()
batch.Set(calcAnnotatedKey(formatKey(height, position)), rawTx)
+ batch.Write()
}
// SetHeightAndPostion set block height and tx position according to tx ID
func (store *LevelDBStore) SetHeightAndPostion(txID string, height uint64, position uint32) {
batch := store.DB.NewBatch()
batch.Set(calcTxIndexKey(txID), []byte(formatKey(height, position)))
+ batch.Write()
}
-// DeleteUnconfirmedTxByTxID delete unconfirmed tx by txID
-func (store *LevelDBStore) DeleteUnconfirmedTxByTxID(txID string) {
+// DeleteUnconfirmedTx delete unconfirmed tx by txID
+func (store *LevelDBStore) DeleteUnconfirmedTx(txID string) {
batch := store.DB.NewBatch()
batch.Delete(calcUnconfirmedTxKey(txID))
+ batch.Write()
}
// SetGlobalTxIndex set global tx index by blockhash and position
func (store *LevelDBStore) SetGlobalTxIndex(globalTxID string, blockHash *bc.Hash, position uint64) {
batch := store.DB.NewBatch()
batch.Set(calcGlobalTxIndexKey(globalTxID), calcGlobalTxIndex(blockHash, position))
+ batch.Write()
}
-// GetStandardUTXOByID get standard utxo by id
-func (store *LevelDBStore) GetStandardUTXOByID(outid bc.Hash) []byte {
+// GetStandardUTXO get standard utxo by id
+func (store *LevelDBStore) GetStandardUTXO(outid bc.Hash) []byte {
return store.DB.Get(account.StandardUTXOKey(outid))
}
-// GetTxIndexByTxID get tx index by txID
-func (store *LevelDBStore) GetTxIndexByTxID(txID string) []byte {
+// GetTransactionIndex get tx index by txID
+func (store *LevelDBStore) GetTransactionIndex(txID string) []byte {
return store.DB.Get(calcTxIndexKey(txID))
}
-// GetTxByTxIndex get tx by tx index
-func (store *LevelDBStore) GetTxByTxIndex(txIndex []byte) []byte {
+// GetTransaction get tx by tx index
+func (store *LevelDBStore) GetTransaction(txIndex []byte) []byte {
return store.DB.Get(calcAnnotatedKey(string(txIndex)))
}
-// GetGlobalTxByTxID get global tx by txID
-func (store *LevelDBStore) GetGlobalTxByTxID(txID string) []byte {
+// GetGlobalTransaction get global tx by txID
+func (store *LevelDBStore) GetGlobalTransaction(txID string) []byte {
return store.DB.Get(calcGlobalTxIndexKey(txID))
}
return annotatedTxs, nil
}
-// GetAllUnconfirmedTxs get all unconfirmed txs
-func (store *LevelDBStore) GetAllUnconfirmedTxs() ([]*query.AnnotatedTx, error) {
+// GetUnconfirmedTransactions get all unconfirmed txs
+func (store *LevelDBStore) GetUnconfirmedTransactions() ([]*query.AnnotatedTx, error) {
annotatedTxs := []*query.AnnotatedTx{}
txIter := store.DB.IteratorPrefix([]byte(UnconfirmedTxPrefix))
defer txIter.Release()
return annotatedTxs, nil
}
-// GetUnconfirmedTxByTxID get unconfirmed tx by txID
-func (store *LevelDBStore) GetUnconfirmedTxByTxID(txID string) []byte {
+// GetUnconfirmedTransaction get unconfirmed tx by txID
+func (store *LevelDBStore) GetUnconfirmedTransaction(txID string) []byte {
return store.DB.Get(calcUnconfirmedTxKey(txID))
}
-// SetUnconfirmedTx set unconfirmed tx by txID
-func (store *LevelDBStore) SetUnconfirmedTx(txID string, rawTx []byte) {
+// SetUnconfirmedTransaction set unconfirmed tx by txID
+func (store *LevelDBStore) SetUnconfirmedTransaction(txID string, rawTx []byte) {
store.DB.Set(calcUnconfirmedTxKey(txID), rawTx)
}
-// DeleteStardardUTXOByOutputID delete stardard utxo by outputID
-func (store *LevelDBStore) DeleteStardardUTXOByOutputID(outputID bc.Hash) {
+// DeleteStardardUTXO delete stardard utxo by outputID
+func (store *LevelDBStore) DeleteStardardUTXO(outputID bc.Hash) {
batch := store.DB.NewBatch()
batch.Delete(account.StandardUTXOKey(outputID))
+ batch.Write()
}
-// DeleteContractUTXOByOutputID delete contract utxo by outputID
-func (store *LevelDBStore) DeleteContractUTXOByOutputID(outputID bc.Hash) {
+// DeleteContractUTXO delete contract utxo by outputID
+func (store *LevelDBStore) DeleteContractUTXO(outputID bc.Hash) {
batch := store.DB.NewBatch()
batch.Delete(account.ContractUTXOKey(outputID))
+ batch.Write()
}
// SetStandardUTXO set standard utxo
func (store *LevelDBStore) SetStandardUTXO(outputID bc.Hash, data []byte) {
batch := store.DB.NewBatch()
batch.Set(account.StandardUTXOKey(outputID), data)
+ batch.Write()
}
// SetContractUTXO set standard utxo
func (store *LevelDBStore) SetContractUTXO(outputID bc.Hash, data []byte) {
batch := store.DB.NewBatch()
batch.Set(account.ContractUTXOKey(outputID), data)
+ batch.Write()
}
// GetWalletInfo get wallet information
batch.Write()
}
-// DeleteAllWalletTxs delete all txs in wallet
-func (store *LevelDBStore) DeleteAllWalletTxs() {
+// DeleteWalletTransactions delete all txs in wallet
+func (store *LevelDBStore) DeleteWalletTransactions() {
storeBatch := store.DB.NewBatch()
txIter := store.DB.IteratorPrefix([]byte(TxPrefix))
storeBatch.Write()
}
-// DeleteAllWalletUTXOs delete all txs in wallet
-func (store *LevelDBStore) DeleteAllWalletUTXOs() {
+// DeleteWalletUTXOs delete all txs in wallet
+func (store *LevelDBStore) DeleteWalletUTXOs() {
storeBatch := store.DB.NewBatch()
ruIter := store.DB.IteratorPrefix([]byte(account.UTXOPreFix))
defer ruIter.Release()
storeBatch.Write()
}
-// GetAccountUtxos get all account unspent outputs
-func (store *LevelDBStore) GetAccountUtxos(accountID, id string, isSmartContract, vote bool, accountUtxos []*account.UTXO) []*account.UTXO {
- prefix := account.UTXOPreFix
- if isSmartContract {
- prefix = account.SUTXOPrefix
- }
- accountUtxoIter := store.DB.IteratorPrefix([]byte(prefix + id))
+// 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")
+ log.WithFields(log.Fields{"module": logModule, "err": err}).Warn("GetAccountUTXOs fail on unmarshal utxo")
continue
}
+ accountUtxos = append(accountUtxos, accountUtxo)
- if vote && accountUtxo.Vote == nil {
- continue
- }
-
- if accountID == accountUtxo.AccountID || accountID == "" {
- accountUtxos = append(accountUtxos, accountUtxo)
- }
}
- newAccountUtxos := []*account.UTXO{}
- newAccountUtxos = accountUtxos
- return newAccountUtxos
+ 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)
}