Hulk did something

[bytom/vapor.git] / account / image.go

diff --git a/account/image.go b/account/image.go
new file mode 100644 (file)
index 0000000..3562c5e
--- /dev/null
+++ b/account/image.go
@@ -0,0 +1,76 @@
+// Package account stores and tracks accounts within a Bytom Core.
+package account
+
+import (
+       "encoding/json"
+
+       log "github.com/sirupsen/logrus"
+)
+
+// ImageSlice record info of single account
+type ImageSlice struct {
+       Account       *Account `json:"account"`
+       ContractIndex uint64   `json:"contract_index"`
+}
+
+// Image is the struct for hold export account data
+type Image struct {
+       Slice []*ImageSlice `json:"slices"`
+}
+
+// Backup export all the account info into image
+func (m *Manager) Backup() (*Image, error) {
+       m.accountMu.Lock()
+       defer m.accountMu.Unlock()
+
+       image := &Image{
+               Slice: []*ImageSlice{},
+       }
+
+       accountIter := m.db.IteratorPrefix(accountPrefix)
+       defer accountIter.Release()
+       for accountIter.Next() {
+               a := &Account{}
+               if err := json.Unmarshal(accountIter.Value(), a); err != nil {
+                       return nil, err
+               }
+
+               image.Slice = append(image.Slice, &ImageSlice{
+                       Account:       a,
+                       ContractIndex: m.GetContractIndex(a.ID),
+               })
+       }
+       return image, nil
+}
+
+// Restore import the accountImages into account manage
+func (m *Manager) Restore(image *Image) error {
+       m.accountMu.Lock()
+       defer m.accountMu.Unlock()
+
+       storeBatch := m.db.NewBatch()
+       for _, slice := range image.Slice {
+               if existed := m.db.Get(Key(slice.Account.ID)); existed != nil {
+                       log.WithFields(log.Fields{
+                               "module": logModule,
+                               "alias":  slice.Account.Alias,
+                               "id":     slice.Account.ID,
+                       }).Warning("skip restore account due to already existed")
+                       continue
+               }
+               if existed := m.db.Get(aliasKey(slice.Account.Alias)); existed != nil {
+                       return ErrDuplicateAlias
+               }
+
+               rawAccount, err := json.Marshal(slice.Account)
+               if err != nil {
+                       return ErrMarshalAccount
+               }
+
+               storeBatch.Set(Key(slice.Account.ID), rawAccount)
+               storeBatch.Set(aliasKey(slice.Account.Alias), []byte(slice.Account.ID))
+       }
+
+       storeBatch.Write()
+       return nil
+}