/
repository.go
76 lines (67 loc) · 1.81 KB
/
repository.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package redis
import (
"fmt"
"strconv"
"github.com/go-redis/redis"
"github.com/pkg/errors"
"github.com/praveen4g0/url-shortner/shortner"
)
type redisRepository struct {
client *redis.Client
}
func newRedisClient(redisURL string) (*redis.Client, error) {
opts, err := redis.ParseURL(redisURL)
if err != nil {
return nil, err
}
client := redis.NewClient(opts)
_, err = client.Ping().Result()
if err != nil {
return nil, err
}
return client, nil
}
func NewRedisRepository(redisURL string) (shortner.RedirectRepository, error) {
repo := &redisRepository{}
client, err := newRedisClient(redisURL)
if err != nil {
return nil, errors.Wrap(err, "repository.NewRedisRepository")
}
repo.client = client
return repo, nil
}
func (r *redisRepository) generateKey(code string) string {
return fmt.Sprintf("redirect:%s", code)
}
func (r *redisRepository) Find(code string) (*shortner.Redirect, error) {
redirect := &shortner.Redirect{}
key := r.generateKey(code)
data, err := r.client.HGetAll(key).Result()
if err != nil {
return nil, errors.Wrap(err, "repository.Redirect.Find")
}
if len(data) == 0 {
return nil, errors.Wrap(shortner.ErrRedirectNotFound, "repository.Redirect.Find")
}
createdAt, err := strconv.ParseInt(data["created_at"], 10, 64)
if err != nil {
return nil, errors.Wrap(err, "repository.Redirect.Find")
}
redirect.Code = data["code"]
redirect.URL = data["url"]
redirect.CreatedAt = createdAt
return redirect, nil
}
func (r *redisRepository) Store(redirect *shortner.Redirect) error {
key := r.generateKey(redirect.Code)
data := map[string]interface{}{
"code": redirect.Code,
"url": redirect.URL,
"created_at": redirect.CreatedAt,
}
_, err := r.client.HMSet(key, data).Result()
if err != nil {
return errors.Wrap(err, "repository.Redirect.Store")
}
return nil
}