up (#579)

[bytom/vapor.git] / toolbar / osssync / sync / sync.go

diff --git a/toolbar/osssync/sync/sync.go b/toolbar/osssync/sync/sync.go
index 88ac2f3..1c1e75c 100644 (file)
--- a/toolbar/osssync/sync/sync.go
+++ b/toolbar/osssync/sync/sync.go
@@ -1,22 +1,42 @@
 package sync
 
 import (
-       "github.com/bytom/vapor/protocol"
-       "github.com/bytom/vapor/protocol/bc/types"
+       "github.com/aliyun/aliyun-oss-go-sdk/oss"
+
+       "github.com/bytom/vapor/toolbar/osssync/config"
+       "github.com/bytom/vapor/toolbar/osssync/util"
 )
 
-// GetLatestDownloadBlockHeight returns the current height of the node wait for download.
-func GetLatestDownloadBlockHeight(c *protocol.Chain) uint64 {
-       return c.BestBlockHeight()
+// Sync the struct of the Sync
+type Sync struct {
+       OssClient *oss.Client
+       OssBucket *oss.Bucket
+       FileUtil  *util.FileUtil
 }
 
-// Sync
-func Sync(c *protocol.Chain, blocks []*types.Block) error {
-       for i := 0; i < len(blocks); i++ {
-               _, err := c.ProcessBlock(blocks[i])
-               if err != nil {
-                       return err
-               }
+// NewSync return one new instance of Sync
+func NewSync() (*Sync, error) {
+       cfg := &config.Config{}
+       err := config.LoadConfig(&cfg)
+       if err != nil {
+               return nil, err
+       }
+
+       ossClient, err := oss.New(cfg.Oss.Endpoint, cfg.Oss.AccessKeyID, cfg.Oss.AccessKeySecret)
+       if err != nil {
+               return nil, err
        }
-       return nil
+
+       ossBucket, err := ossClient.Bucket("bytom-seed")
+       if err != nil {
+               return nil, err
+       }
+
+       fileUtil := util.NewFileUtil("./blocks")
+
+       return &Sync{
+               OssClient: ossClient,
+               OssBucket: ossBucket,
+               FileUtil:  fileUtil,
+       }, nil
 }