Reformatting

This commit is contained in:
OsamaSayegh 2021-03-16 03:20:43 +03:00
parent a9aa54d567
commit ac66838f6b
No known key found for this signature in database
GPG Key ID: 060E5AC82223685F
3 changed files with 18 additions and 18 deletions

View File

@ -19,9 +19,9 @@ import (
"text/template"
"time"
"github.com/go-redis/redis/v8"
"github.com/golang/groupcache/lru"
"github.com/pborman/uuid"
"github.com/go-redis/redis/v8"
"github.com/discourse/discourse-auth-proxy/internal/httpproxy"
)
@ -323,11 +323,11 @@ func computeHMAC(message string, secret string) string {
func setupStorage(config *Config) {
if config.RedisAddress != "" {
client := redis.NewClient(&redis.Options{
Addr: config.RedisAddress,
Addr: config.RedisAddress,
Password: config.RedisPassword,
})
storageInstance = &RedisStore{
Redis: client,
Redis: client,
Namespace: "_discourse-auth-proxy_",
}
} else {

View File

@ -6,14 +6,14 @@ import (
"net/http"
"net/http/httptest"
"net/url"
"strconv"
"testing"
"sync"
"os"
"strconv"
"sync"
"testing"
"github.com/stretchr/testify/assert"
"github.com/golang/groupcache/lru"
"github.com/go-redis/redis/v8"
"github.com/golang/groupcache/lru"
"github.com/stretchr/testify/assert"
)
type SSOOptions struct {
@ -45,7 +45,7 @@ func NewMemoryStore() *MemoryStore {
var redisStore = NewRedisStore()
var memoryStore = NewMemoryStore()
var stores = [2]CacheStore{ memoryStore, redisStore }
var stores = [2]CacheStore{memoryStore, redisStore}
func mustParseURL(s string) *url.URL {
u, err := url.Parse(s)
@ -337,11 +337,11 @@ func TestGetSetNXCookieSecretIfRedis(t *testing.T) {
func TestMain(m *testing.M) {
for _, s := range stores {
err := s.Clear()
if err != nil {
fmt.Printf("%s", err)
os.Exit(1)
}
err := s.Clear()
if err != nil {
fmt.Printf("%s", err)
os.Exit(1)
}
}
code := m.Run()
os.Exit(code)

View File

@ -6,8 +6,8 @@ import (
"sync"
"time"
"github.com/golang/groupcache/lru"
"github.com/go-redis/redis/v8"
"github.com/golang/groupcache/lru"
"github.com/pborman/uuid"
)
@ -55,12 +55,12 @@ func (store *MemoryStore) Clear() error {
}
type RedisStore struct {
Redis *redis.Client
Redis *redis.Client
Namespace string
}
func (store *RedisStore) AddNonce(nonce string, val string) error {
err := store.Redis.SetEX(bgCtx, store.Prefix(nonce), val, 600 * time.Second).Err()
err := store.Redis.SetEX(bgCtx, store.Prefix(nonce), val, 600*time.Second).Err()
if err != nil {
return err
}
@ -82,7 +82,7 @@ func (store *RedisStore) GetAndDeleteNonce(nonce string) (val string, err error)
// used in tests only
func (store *RedisStore) Clear() error {
keys, err := store.Redis.Keys(bgCtx, store.Namespace + "*").Result()
keys, err := store.Redis.Keys(bgCtx, store.Namespace+"*").Result()
if err != nil {
return err
}