Implement login resolver (#15)

* add sign_up_method to users table

* add session store

* implement login resolver
This commit is contained in:
Lakhan Samani
2021-07-15 00:13:19 +05:30
committed by GitHub
parent 336fe10ca4
commit 27264c6e29
20 changed files with 874 additions and 884 deletions

81
server/session/session.go Normal file
View File

@@ -0,0 +1,81 @@
package session
import (
"context"
"log"
"github.com/go-redis/redis/v8"
"github.com/yauthdev/yauth/server/constants"
)
type SessionStore struct {
inMemoryStoreObj *InMemoryStore
redisMemoryStoreObj *RedisStore
}
var SessionStoreObj SessionStore
func SetToken(userId, token string) {
if SessionStoreObj.redisMemoryStoreObj != nil {
SessionStoreObj.redisMemoryStoreObj.AddToken(userId, token)
}
if SessionStoreObj.inMemoryStoreObj != nil {
SessionStoreObj.inMemoryStoreObj.AddToken(userId, token)
}
}
func DeleteToken(userId string) {
if SessionStoreObj.redisMemoryStoreObj != nil {
SessionStoreObj.redisMemoryStoreObj.DeleteToken(userId)
}
if SessionStoreObj.inMemoryStoreObj != nil {
SessionStoreObj.inMemoryStoreObj.DeleteToken(userId)
}
}
func GetToken(userId string) string {
if SessionStoreObj.redisMemoryStoreObj != nil {
return SessionStoreObj.redisMemoryStoreObj.GetToken(userId)
}
if SessionStoreObj.inMemoryStoreObj != nil {
return SessionStoreObj.inMemoryStoreObj.GetToken(userId)
}
return ""
}
func ClearStore() {
if SessionStoreObj.redisMemoryStoreObj != nil {
SessionStoreObj.redisMemoryStoreObj.ClearStore()
}
if SessionStoreObj.inMemoryStoreObj != nil {
SessionStoreObj.inMemoryStoreObj.ClearStore()
}
}
func init() {
if constants.REDIS_URL != "" {
log.Println("Using redis store to save sessions")
opt, err := redis.ParseURL(constants.REDIS_URL)
if err != nil {
log.Fatalln("Error parsing redis url:", err)
}
rdb := redis.NewClient(opt)
ctx := context.Background()
_, err = rdb.Ping(ctx).Result()
if err != nil {
log.Fatalln("Error connecting to redis server", err)
}
SessionStoreObj.redisMemoryStoreObj = &RedisStore{
ctx: ctx,
store: rdb,
}
} else {
log.Println("Using in memory store to save sessions")
SessionStoreObj.inMemoryStoreObj = &InMemoryStore{
store: make(map[string]string),
}
}
}