Redis -> YDB

cmd/vk/main.go

@@ -1,19 +1,21 @@
 package main
 
 import (
-	"encoding/json"
-	"io/ioutil"
+	"context"
+	"fmt"
 	"log"
 	"math/rand"
 	"os"
 	"strconv"
 	"time"
 
+	"gitlab.com/flygrounder/go-mtg-vk/internal/caching"
 	"gitlab.com/flygrounder/go-mtg-vk/internal/cardsinfo"
 	"gitlab.com/flygrounder/go-mtg-vk/internal/scenario"
 
 	"github.com/gin-gonic/gin"
-	"gitlab.com/flygrounder/go-mtg-vk/internal/caching"
+	environ "github.com/ydb-platform/ydb-go-sdk-auth-environ"
+	"github.com/ydb-platform/ydb-go-sdk/v3"
 	"gitlab.com/flygrounder/go-mtg-vk/internal/vk"
 )
 
@@ -22,11 +24,34 @@ func main() {
 
 	r := gin.Default()
 
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	dsn, exists := os.LookupEnv("YDB_CONNECTION_STRING")
+	if !exists {
+		panic("YDB_CONNECTION_STRING environment variable not defined")
+	}
+
+	db, err := ydb.Open(ctx,
+		dsn,
+		environ.WithEnvironCredentials(ctx),
+	)
+	if err != nil {
+		panic(fmt.Errorf("connect error: %w", err))
+	}
+	defer func() { _ = db.Close(ctx) }()
+
+	cache := &caching.CacheClient{
+		Storage:    db.Table(),
+		Expiration: 12 * time.Hour,
+		Prefix:     db.Name(),
+	}
+	err = cache.Init(context.Background())
+	if err != nil {
+		panic(fmt.Errorf("init error: %w", err))
+	}
+
 	groupId, _ := strconv.ParseInt(os.Getenv("VK_GROUP_ID"), 10, 64)
-	dict, _ := os.Open("./assets/additional_cards.json")
-	dictBytes, _ := ioutil.ReadAll(dict)
-	var dictMap map[string]string
-	_ = json.Unmarshal(dictBytes, &dictMap)
 	logger := log.New(os.Stdout, "", 0)
 	handler := vk.Handler{
 		Scenario: &scenario.Scenario{
@@ -34,11 +59,9 @@ func main() {
 				Token:  os.Getenv("VK_TOKEN"),
 				Logger: logger,
 			},
-			Logger: logger,
-			Cache:  caching.NewClient("redis:6379", "", time.Hour*24, 0),
-			InfoFetcher: &cardsinfo.Fetcher{
-				Dict: dictMap,
-			},
+			Logger:      logger,
+			InfoFetcher: &cardsinfo.Fetcher{},
+			Cache:       cache,
 		},
 		SecretKey:          os.Getenv("VK_SECRET_KEY"),
 		GroupId:            groupId,