限制非商业版本从用户端登录

This commit is contained in:
GoEdgeLab
2021-11-24 20:06:43 +08:00
parent 77fa42e48b
commit 1922bc1aa2

View File

@@ -3,6 +3,7 @@ package services
import ( import (
"context" "context"
"encoding/json" "encoding/json"
teaconst "github.com/TeaOSLab/EdgeAPI/internal/const"
"github.com/TeaOSLab/EdgeAPI/internal/db/models" "github.com/TeaOSLab/EdgeAPI/internal/db/models"
rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils" rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
"github.com/TeaOSLab/EdgeAPI/internal/utils" "github.com/TeaOSLab/EdgeAPI/internal/utils"
@@ -221,11 +222,19 @@ func (this *UserService) CheckUserUsername(ctx context.Context, req *pb.CheckUse
// LoginUser 登录 // LoginUser 登录
func (this *UserService) LoginUser(ctx context.Context, req *pb.LoginUserRequest) (*pb.LoginUserResponse, error) { func (this *UserService) LoginUser(ctx context.Context, req *pb.LoginUserRequest) (*pb.LoginUserResponse, error) {
_, _, _, err := rpcutils.ValidateRequest(ctx) _, err := this.ValidateUserNode(ctx)
if err != nil { if err != nil {
return nil, err return nil, err
} }
if !teaconst.IsPlus {
return &pb.LoginUserResponse{
UserId: 0,
IsOk: false,
Message: "你正在使用的系统版本为非商业版本或已过期,请管理员续费后才能登录",
}, nil
}
if len(req.Username) == 0 || len(req.Password) == 0 { if len(req.Username) == 0 || len(req.Password) == 0 {
return &pb.LoginUserResponse{ return &pb.LoginUserResponse{
UserId: 0, UserId: 0,