sample/internal/logic/login/code.go

53 lines
1.3 KiB
Go

package login
import (
"context"
"git.apinb.com/bsm-sdk/core/errcode"
"git.apinb.com/bsm-sdk/core/print"
"git.apinb.com/bsm-sdk/sample/internal/impl"
"git.apinb.com/bsm-sdk/sample/internal/logic/common"
"git.apinb.com/bsm-sdk/sample/internal/models"
"git.apinb.com/bsm-sdk/sample/internal/vars"
pb "git.apinb.com/bsm-sdk/sample/pb"
)
// 通过验证码登录
func Code(ctx context.Context, in *pb.LoginByCodeRequest) (reply *pb.LoginReply, err error) {
if in.Phone == "" || in.Code == "" {
return nil, errcode.ErrInvalidArgument
}
pa, err := models.GetsampleAccountByField("phone", in.Phone)
if err != nil {
return nil, err
}
if pa.Status == vars.Status_Disable {
return nil, errcode.ErrPermissionDenied
}
paExtend := map[string]string{
"rights": pa.Rights,
}
token, err := common.GenerateTokenAes(uint(pa.ID), pa.Identity, "", pa.Rights, paExtend)
if err != nil {
return nil, err
}
//save token to cache.
err = impl.RedisService.Client.Set(impl.RedisService.Ctx, vars.TokenPrefix+pa.Identity, token, 0).Err()
if err != nil {
print.Error("Set redis cache by key %v error:%v", vars.TokenPrefix+pa.Identity, err)
return nil, errcode.ErrInternal
}
return &pb.LoginReply{
Id: int64(pa.ID),
Identity: pa.Identity,
Token: token,
Extend: paExtend,
}, nil
}