package kvs

import (
   "strconv"
   "time"

   "github.com/go-redis/redis"
   "github.com/pkg/errors"
)

// 값 취득 때 없던 경우는 redis.Nil 이 error 타입으로 반환된다.
// 에러 핸들링을 위해 internal한 package 에 정의한다
const Nil = redis.Nil

// New create instance of redis.Client
func New(dsn string) (*redis.Client, error) {
   client := redis.NewClient(&redis.Options{
       Addr:     dsn,
       Password: "",
       DB:       0,
   })
   if err := client.Ping().Err(); err != nil {
       return nil, errors.Wrapf(err, "failed to ping redis server")
   }
   return client, nil
}

// 이번은 토큰 베이스에서 값을 얻도록 하는 예를 구현한다
const (
   tokenKey      = "TOKEN_"
   tokenDuration = time.Second * 3600
)

// SetToken set token and value
func SetToken(cli *redis.Client, token string, userID int) error {
   if err := cli.Set(tokenKey+token, userID, tokenDuration).Err(); err != nil {
       return err
   }
   return nil
}

// SetToken set token and value
func GetIDByToken(cli *redis.Client, token string) (int, error) {
   v, err := cli.Get(tokenKey+token).Result()
   if err != nil {
       return 0, errors.Wrapf(err, "failed to get id from redis by token")
   }
   id, err := strconv.Atoi(v)
   if err != nil {
       return 0, errors.Wrapf(err, "failed to convert string to int")
   }
   return id, nil
}

package testhelper

import (
   "testing"

   "github.com/alicebob/miniredis"
   "github.com/go-redis/redis"
)

func NewMockRedis(t *testing.T) *redis.Client {
   t.Helper()

   // redis 서버를 만든다
   s, err := miniredis.Run()
   if err != nil {
       t.Fatalf("unexpected error while createing test redis server '%#v'", err)
   }
   // *redis.Client를 준비
   client := redis.NewClient(&redis.Options{
       Addr:     s.Addr(),
       Password: "",
       DB:       0,
   })
   return client
}

package kvs_test

import (
   "testing"
   "time"

   "github.com/higasgt/go-snippets/redis-cli/persistence/kvs"
   "github.com/higasgt/go-snippets/redis-cli/testhelper"
)

func TestSetToken(t *testing.T) {
   client := testhelper.NewMockRedis(t)

   if err := kvs.SetToken(client, "test", 1); err != nil {
       t.Fatalf("unexpected error while SetToken '%#v'", err)
   }
   actual, err := client.Get("TOKEN_test").Result()
   if err != nil {
       t.Fatalf("unexpected error while get value '%#v'", err)
   }

   if expected := "1"; expected != actual {
       t.Errorf("expected value '%s', actual value '%s'", expected, actual)
   }
}

func TestGetIDByToken(t *testing.T) {
   client := testhelper.NewMockRedis(t)
   if err := client.Set("TOKEN_test", 1, time.Second*1000).Err(); err != nil {
       t.Fatalf("unexpected error while set test data '%#v'", err)
   }

   actual, err := kvs.GetIDByToken(client, "test")
   if err != nil {
       t.Fatalf("unexpected error while GetIDByToken '%#v'", err)
   }
   if expected := 1; expected != actual {
       t.Errorf("expected value '%#v', actual value '%#v'", expected, actual)
   }
}