From 6f92119b9028dc74827839dee496e638da8316bf Mon Sep 17 00:00:00 2001 From: yokowu <18836617@qq.com> Date: Tue, 2 Sep 2025 10:06:28 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=8F=96=E6=B6=88=E7=94=A8=E6=88=B7?= =?UTF-8?q?=E9=99=90=E5=88=B6=E9=85=8D=E7=BD=AE=E5=8F=82=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/config/config.go | 2 -- backend/internal/user/repo/user.go | 21 --------------------- 2 files changed, 23 deletions(-) diff --git a/backend/config/config.go b/backend/config/config.go index 2c7c298..384ee83 100644 --- a/backend/config/config.go +++ b/backend/config/config.go @@ -30,7 +30,6 @@ type Config struct { Admin struct { User string `mapstructure:"user"` Password string `mapstructure:"password"` - Limit int `mapstructure:"limit"` } `mapstructure:"admin"` Session struct { @@ -128,7 +127,6 @@ func Init() (*Config, error) { v.SetDefault("server.port", "") v.SetDefault("admin.user", "admin") v.SetDefault("admin.password", "") - v.SetDefault("admin.limit", 100) v.SetDefault("session.expire_day", 30) v.SetDefault("database.master", "") v.SetDefault("database.slave", "") diff --git a/backend/internal/user/repo/user.go b/backend/internal/user/repo/user.go index adb38d4..a9a8750 100644 --- a/backend/internal/user/repo/user.go +++ b/backend/internal/user/repo/user.go @@ -139,9 +139,6 @@ func (r *UserRepo) innerValidateInviteCode(ctx context.Context, tx *db.Tx, code func (r *UserRepo) CreateUser(ctx context.Context, us *db.User) (*db.User, error) { var res *db.User err := entx.WithTx(ctx, r.db, func(tx *db.Tx) error { - if err := r.checkLimit(ctx, tx); err != nil { - return err - } n, err := tx.User.Query().Where(user.Email(us.Email)).Count(ctx) if err != nil { return err @@ -371,10 +368,6 @@ func (r *UserRepo) DeleteAdmin(ctx context.Context, id string) error { func (r *UserRepo) OAuthRegister(ctx context.Context, platform consts.UserPlatform, inviteCode string, req *domain.OAuthUserInfo) (*db.User, error) { var u *db.User err := entx.WithTx(ctx, r.db, func(tx *db.Tx) error { - if err := r.checkLimit(ctx, tx); err != nil { - return err - } - if _, err := r.innerValidateInviteCode(ctx, tx, inviteCode); err != nil { return errcode.ErrInviteCodeInvalid.Wrap(err) } @@ -447,17 +440,6 @@ func (r *UserRepo) updateAvatar(ctx context.Context, tx *db.Tx, ui *db.UserIdent return tx.User.UpdateOneID(ui.UserID).SetAvatarURL(avatar).Exec(ctx) } -func (r *UserRepo) checkLimit(ctx context.Context, tx *db.Tx) error { - count, err := tx.User.Query().Count(ctx) - if err != nil { - return err - } - if count >= r.cfg.Admin.Limit { - return errcode.ErrUserLimit.Wrap(err) - } - return nil -} - func (r *UserRepo) SignUpOrIn(ctx context.Context, platform consts.UserPlatform, req *domain.OAuthUserInfo) (*db.User, error) { var u *db.User err := entx.WithTx(ctx, r.db, func(tx *db.Tx) error { @@ -480,9 +462,6 @@ func (r *UserRepo) SignUpOrIn(ctx context.Context, platform consts.UserPlatform, if !db.IsNotFound(err) { return err } - if err = r.checkLimit(ctx, tx); err != nil { - return err - } user, err := tx.User.Create(). SetUsername(req.Name). SetEmail(req.Email).