Updated code for upload and download

tools/object-storage/obj.go

@@ -28,7 +28,8 @@ func main() {
 	Objstore := app.Command("block-sync", `Using Thanos to store the data`)
 	Objstore.Flag("path", "Path for The TSDB data in prometheus").Required().StringVar(&s.TsdbPath)
 	Objstore.Flag("objstore.config-file", "Path for The Config file").Required().StringVar(&s.ObjectConfig)
-	Objstore.Flag("key", "Path for the Key where to store block data").StringVar(&s.ObjectKey)
-	Objstore.Command("upload", "Uploading data").Action(s.Upload)
+	Objstore.Flag("key", "Path for the Key where to store block data").Required().StringVar(&s.ObjectKey)
+	Objstore.Command("upload", "Uploading data").Action(s.upload)
+	Objstore.Command("download", "Downloading data").Action(s.download)
 	kingpin.MustParse(app.Parse(os.Args[1:]))
 }

tools/object-storage/upload_download.go

@@ -0,0 +1,77 @@
+package main
+
+import (
+	"context"
+	"fmt"
+
+	// "io"
+	"os"
+
+	"github.com/go-kit/log"
+	"github.com/thanos-io/objstore"
+	"github.com/thanos-io/objstore/client"
+	"gopkg.in/alecthomas/kingpin.v2"
+)
+
+type Store struct {
+	TsdbPath     string
+	ObjectKey    string
+	ObjectConfig string
+}
+
+func newstore() *Store {
+	return &Store{
+		TsdbPath:     "",
+		ObjectKey:    "",
+		ObjectConfig: "",
+	}
+}
+
+func (c *Store) upload(*kingpin.ParseContext) error {
+	config := c.ObjectConfig
+	configBytes, err := os.ReadFile(config)
+	if err != nil {
+		panic(fmt.Errorf("failed to read config file: %v", err))
+	}
+	commitDir := c.ObjectKey
+
+	bucket, err := client.NewBucket(log.NewNopLogger(), configBytes, "example")
+	if err != nil {
+		panic(err)
+	}
+	exists, err := bucket.Exists(context.Background(), "example")
+	if err != nil {
+		panic(err)
+	}
+	fmt.Println(exists)
+
+	err = objstore.UploadDir(context.Background(), log.NewNopLogger(), bucket, c.TsdbPath, commitDir)
+	if err != nil {
+		panic(err)
+	}
+	return nil
+}
+
+func (c *Store) download(*kingpin.ParseContext) error {
+	config := c.ObjectConfig
+	configBytes, err := os.ReadFile(config)
+	if err != nil {
+		panic(fmt.Errorf("failed to read config file: %v", err))
+	}
+	commitDir := c.ObjectKey
+	bucket, err := client.NewBucket(log.NewNopLogger(), configBytes, "example")
+	if err != nil {
+		panic(err)
+	}
+	exists, err := bucket.Exists(context.Background(), "example")
+	if err != nil {
+		return fmt.Errorf("failed to create bucket: %v", err)
+	}
+	fmt.Println(exists)
+
+	err = objstore.DownloadDir(context.Background(), log.NewNopLogger(), bucket, "dir/", commitDir, c.TsdbPath)
+	if err != nil {
+		return fmt.Errorf("failed to download directory: %v", err)
+	}
+	return nil
+}