OSDN Git Service

optimise
authoroysheng <oysheng@bytom.io>
Sun, 22 Apr 2018 08:24:16 +0000 (16:24 +0800)
committeroysheng <oysheng@bytom.io>
Sun, 22 Apr 2018 08:24:16 +0000 (16:24 +0800)
account/accounts.go
asset/asset.go

index 4763ee9..8e90848 100644 (file)
@@ -409,7 +409,7 @@ func (m *Manager) DeleteAccount(aliasOrID string) (err error) {
 // ListAccounts will return the accounts in the db
 func (m *Manager) ListAccounts(id string) ([]*Account, error) {
        accounts := []*Account{}
-       accountIter := m.db.IteratorPrefix(Key(id))
+       accountIter := m.db.IteratorPrefix(Key(strings.TrimSpace(id)))
        defer accountIter.Release()
 
        for accountIter.Next() {
index b173c30..d75ed20 100644 (file)
@@ -303,12 +303,16 @@ func (reg *Registry) GetAsset(id string) (*Asset, error) {
 func (reg *Registry) ListAssets(id string) ([]*Asset, error) {
        assets := []*Asset{DefaultNativeAsset}
 
-       assetID := &bc.AssetID{}
-       if err := assetID.UnmarshalText([]byte(id)); err != nil {
-               return nil, err
+       assetKey := assetPrefix
+       if strings.Compare(strings.TrimSpace(id), "") == 0 {
+               assetID := &bc.AssetID{}
+               if err := assetID.UnmarshalText([]byte(strings.TrimSpace(id))); err != nil {
+                       return nil, err
+               }
+               assetKey = Key(assetID)
        }
 
-       assetIter := reg.db.IteratorPrefix(Key(assetID))
+       assetIter := reg.db.IteratorPrefix(assetKey)
        defer assetIter.Release()
 
        for assetIter.Next() {