Dmitry Verkhoturov
9ad3be2e97
After this commit, dev auth would start working with the `REMARK_URL` hostname instead of the previously hardcoded 127.0.0.1. Breaks development setup where `REMARK_URL` was set to a non-standard value and dev auth was running on 127.0.0.1 and working, as, after that change, it would stop working.
137 lines
3.2 KiB
Go
137 lines
3.2 KiB
Go
package lcw
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/go-redis/redis/v8"
|
|
"github.com/hashicorp/go-multierror"
|
|
)
|
|
|
|
// New parses uri and makes any of supported caches
|
|
// supported URIs:
|
|
// - redis://<ip>:<port>?db=123&max_keys=10
|
|
// - mem://lru?max_keys=10&max_cache_size=1024
|
|
// - mem://expirable?ttl=30s&max_val_size=100
|
|
// - nop://
|
|
func New(uri string) (LoadingCache, error) {
|
|
u, err := url.Parse(uri)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("parse cache uri %s: %w", uri, err)
|
|
}
|
|
|
|
query := u.Query()
|
|
opts, err := optionsFromQuery(query)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("parse uri options %s: %w", uri, err)
|
|
}
|
|
|
|
switch u.Scheme {
|
|
case "redis":
|
|
redisOpts, e := redisOptionsFromURL(u)
|
|
if e != nil {
|
|
return nil, e
|
|
}
|
|
res, e := NewRedisCache(redis.NewClient(redisOpts), opts...)
|
|
if e != nil {
|
|
return nil, fmt.Errorf("make redis for %s: %w", uri, e)
|
|
}
|
|
return res, nil
|
|
case "mem":
|
|
switch u.Hostname() {
|
|
case "lru":
|
|
return NewLruCache(opts...)
|
|
case "expirable":
|
|
return NewExpirableCache(opts...)
|
|
default:
|
|
return nil, fmt.Errorf("unsupported mem cache type %s", u.Hostname())
|
|
}
|
|
case "nop":
|
|
return NewNopCache(), nil
|
|
}
|
|
return nil, fmt.Errorf("unsupported cache type %s", u.Scheme)
|
|
}
|
|
|
|
func optionsFromQuery(q url.Values) (opts []Option, err error) {
|
|
errs := new(multierror.Error)
|
|
|
|
if v := q.Get("max_val_size"); v != "" {
|
|
vv, e := strconv.Atoi(v)
|
|
if e != nil {
|
|
errs = multierror.Append(errs, fmt.Errorf("max_val_size query param %s: %w", v, e))
|
|
} else {
|
|
opts = append(opts, MaxValSize(vv))
|
|
}
|
|
}
|
|
|
|
if v := q.Get("max_key_size"); v != "" {
|
|
vv, e := strconv.Atoi(v)
|
|
if e != nil {
|
|
errs = multierror.Append(errs, fmt.Errorf("max_key_size query param %s: %w", v, e))
|
|
} else {
|
|
opts = append(opts, MaxKeySize(vv))
|
|
}
|
|
}
|
|
|
|
if v := q.Get("max_keys"); v != "" {
|
|
vv, e := strconv.Atoi(v)
|
|
if e != nil {
|
|
errs = multierror.Append(errs, fmt.Errorf("max_keys query param %s: %w", v, e))
|
|
} else {
|
|
opts = append(opts, MaxKeys(vv))
|
|
}
|
|
}
|
|
|
|
if v := q.Get("max_cache_size"); v != "" {
|
|
vv, e := strconv.ParseInt(v, 10, 64)
|
|
if e != nil {
|
|
errs = multierror.Append(errs, fmt.Errorf("max_cache_size query param %s: %w", v, e))
|
|
} else {
|
|
opts = append(opts, MaxCacheSize(vv))
|
|
}
|
|
}
|
|
|
|
if v := q.Get("ttl"); v != "" {
|
|
vv, e := time.ParseDuration(v)
|
|
if e != nil {
|
|
errs = multierror.Append(errs, fmt.Errorf("ttl query param %s: %w", v, e))
|
|
} else {
|
|
opts = append(opts, TTL(vv))
|
|
}
|
|
}
|
|
|
|
return opts, errs.ErrorOrNil()
|
|
}
|
|
|
|
func redisOptionsFromURL(u *url.URL) (*redis.Options, error) {
|
|
query := u.Query()
|
|
|
|
db, err := strconv.Atoi(query.Get("db"))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("db from %s: %w", u, err)
|
|
}
|
|
|
|
res := &redis.Options{
|
|
Addr: u.Hostname() + ":" + u.Port(),
|
|
DB: db,
|
|
Password: query.Get("password"),
|
|
Network: query.Get("network"),
|
|
}
|
|
|
|
if dialTimeout, err := time.ParseDuration(query.Get("dial_timeout")); err == nil {
|
|
res.DialTimeout = dialTimeout
|
|
}
|
|
|
|
if readTimeout, err := time.ParseDuration(query.Get("read_timeout")); err == nil {
|
|
res.ReadTimeout = readTimeout
|
|
}
|
|
|
|
if writeTimeout, err := time.ParseDuration(query.Get("write_timeout")); err == nil {
|
|
res.WriteTimeout = writeTimeout
|
|
}
|
|
|
|
return res, nil
|
|
}
|