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         // GetAccounts()
  31         rawAccounts := m.store.GetAccounts("")
  32
  33         for _, rawAccount := range rawAccounts {
  34                 account := new(Account)
  35                 if err := json.Unmarshal(rawAccount, account); err != nil {
  36                         return nil, err
  37                 }
  38                 image.Slice = append(image.Slice, &ImageSlice{
  39                         Account:       account,
  40                         ContractIndex: m.GetContractIndex(account.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         m.store.InitBatch()
  52         defer m.store.CommitBatch()
  53
  54         for _, slice := range image.Slice {
  55                 if existed := m.store.GetAccountByAccountID(slice.Account.ID); existed != nil {
  56                         log.WithFields(log.Fields{
  57                                 "module": logModule,
  58                                 "alias":  slice.Account.Alias,
  59                                 "id":     slice.Account.ID,
  60                         }).Warning("skip restore account due to already existed")
  61                         continue
  62                 }
  63                 if existed := m.store.GetAccountByAccountAlias(slice.Account.Alias); existed != nil {
  64                         return ErrDuplicateAlias
  65                 }
  66
  67                 rawAccount, err := json.Marshal(slice.Account)
  68                 if err != nil {
  69                         return ErrMarshalAccount
  70                 }
  71
  72                 m.store.SetAccount(slice.Account.ID, slice.Account.Alias, rawAccount)
  73         }
  74
  75         return nil
  76 }