update GetUTXOs

[bytom/vapor.git] / wallet / store.go

diff --git a/wallet/store.go b/wallet/store.go
index 4b7b02d..2b64302 100644 (file)
--- a/wallet/store.go
+++ b/wallet/store.go
@@ -1,13 +1,41 @@
 package wallet
 
 import (
+       "github.com/vapor/blockchain/query"
        "github.com/vapor/common"
        "github.com/vapor/protocol/bc"
 )
 
-// DB interface contains wallet storage functions.
-type DB interface {
-       GetAssetDefinitionByAssetID(*bc.AssetID) []byte
-       GetRawProgramByAccountHash(common.Hash) []byte
-       GetAccountValueByAccountID(string) []byte
+// WalletStorer interface contains wallet storage functions.
+type WalletStorer interface {
+       InitBatch()
+       CommitBatch()
+       GetAssetDefinition(*bc.AssetID) []byte
+       SetAssetDefinition(*bc.AssetID, []byte)
+       GetRawProgram(common.Hash) []byte
+       GetAccountByAccountID(string) []byte
+       DeleteTransactions(uint64)
+       SetTransaction(uint64, uint32, string, []byte)
+       DeleteUnconfirmedTransaction(string)
+       SetGlobalTransactionIndex(string, *bc.Hash, uint64)
+       GetStandardUTXO(bc.Hash) []byte
+       GetTransaction(string) ([]byte, error)
+       GetGlobalTransaction(string) []byte
+       GetTransactions() ([]*query.AnnotatedTx, error)
+       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)
+       DeleteWalletTransactions()
+       DeleteWalletUTXOs()
+       GetAccountUTXOs(key string) [][]byte
+       SetRecoveryStatus([]byte, []byte)
+       DeleteRecoveryStatus([]byte)
+       GetRecoveryStatus([]byte) []byte
 }
+