account/image.go

   1 // Package account stores and tracks accounts within a Bytom Core.
   2 package account
   3
   4 import (
   5         "encoding/json"
   6
   7         log "github.com/sirupsen/logrus"
   8 )
   9
  10 // ImageSlice record info of single account
  11 type ImageSlice struct {
  12         Account       *Account `json:"account"`
  13         ContractIndex uint64   `json:"contract_index"`
  14 }
  15
  16 // Image is the struct for hold export account data
  17 type Image struct {
  18         Slice []*ImageSlice `json:"slices"`
  19 }
  20
  21 // Backup export all the account info into image
  22 func (m *Manager) Backup() (*Image, error) {
  23         m.accountMu.Lock()
  24         defer m.accountMu.Unlock()
  25
  26         image := &Image{
  27                 Slice: []*ImageSlice{},
  28         }
  29
  30         accountIter := m.db.IteratorPrefix(accountPrefix)
  31         defer accountIter.Release()
  32         for accountIter.Next() {
  33                 a := &Account{}
  34                 if err := json.Unmarshal(accountIter.Value(), a); err != nil {
  35                         return nil, err
  36                 }
  37
  38                 image.Slice = append(image.Slice, &ImageSlice{
  39                         Account:       a,
  40                         ContractIndex: m.GetContractIndex(a.ID),
  41                 })
  42         }
  43         return image, nil
  44 }
  45
  46 // Restore import the accountImages into account manage
  47 func (m *Manager) Restore(image *Image) error {
  48         m.accountMu.Lock()
  49         defer m.accountMu.Unlock()
  50
  51         storeBatch := m.db.NewBatch()
  52         for _, slice := range image.Slice {
  53                 if existed := m.db.Get(Key(slice.Account.ID)); existed != nil {
  54                         log.WithFields(log.Fields{
  55                                 "alias": slice.Account.Alias,
  56                                 "id":    slice.Account.ID,
  57                         }).Warning("skip restore account due to already existed")
  58                         continue
  59                 }
  60                 if existed := m.db.Get(aliasKey(slice.Account.Alias)); existed != nil {
  61                         return ErrDuplicateAlias
  62                 }
  63
  64                 rawAccount, err := json.Marshal(slice.Account)
  65                 if err != nil {
  66                         return ErrMarshalAccount
  67                 }
  68
  69                 storeBatch.Set(Key(slice.Account.ID), rawAccount)
  70                 storeBatch.Set(aliasKey(slice.Account.Alias), []byte(slice.Account.ID))
  71         }
  72
  73         storeBatch.Write()
  74         return nil
  75 }