totole/services
1
0
Fork 0
mirror of https://github.com/kevin-DL/services.git synced 2026-01-19 05:55:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tenancy to seen and otp services

Browse Source
This commit is contained in:
Asim Aslam
2021-05-02 18:11:22 +01:00
parent eb82bebd03
commit 5368aec8e4
4 changed files with 86 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
otp/handler/otp.go
View File

@@ -23,7 +23,7 @@ func (e *Otp) Generate(ctx context.Context, req *pb.GenerateRequest, rsp *pb.Gen
// check if a key exists for the user
var secret string
if err := cache.Get(req.Id, &secret); err != nil {
if err := cache.Context(ctx).Get(req.Id, &secret); err != nil {
// generate a key
key, err := totp.Generate(totp.GenerateOpts{
Issuer: "Micro",
@@ -38,7 +38,7 @@ func (e *Otp) Generate(ctx context.Context, req *pb.GenerateRequest, rsp *pb.Gen
secret = key.Secret()
if err := cache.Put(req.Id, secret, time.Now().Add(time.Minute*5)); err != nil {
if err := cache.Context(ctx).Put(req.Id, secret, time.Now().Add(time.Minute*5)); err != nil {
logger.Error("Failed to store secret: %v", err)
return errors.InternalServerError("otp.generate", "failed to generate code")
}
@@ -72,7 +72,7 @@ func (e *Otp) Validate(ctx context.Context, req *pb.ValidateRequest, rsp *pb.Val
var secret string
if err := cache.Get(req.Id, &secret); err != nil {
if err := cache.Context(ctx).Get(req.Id, &secret); err != nil {
logger.Error("Failed to get secret from store: %v", err)
return errors.InternalServerError("otp.generate", "failed to validate code")
}