X-Git-Url: http://git.osdn.net/view?p=bytom%2Fvapor.git;a=blobdiff_plain;f=account%2Fimage.go;fp=account%2Fimage.go;h=3562c5e92f7e0bdbbae37630e1f0b2d0f834a442;hp=0000000000000000000000000000000000000000;hb=db158dcf09436b003defd333f1a665e7e051d820;hpb=d09b7a78d44dc259725902b8141cdba0d716b121

diff --git a/account/image.go b/account/image.go
new file mode 100644
index 00000000..3562c5e9
--- /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
+}