Merge pull request #127 from swift-fs/v2.0-fix

refactor: 代码优化相关

server/internal/controller/admin/common/site.go

@@ -7,18 +7,19 @@ package common
 
 import (
 	"context"
-	"github.com/gogf/gf/v2/errors/gerror"
-	"github.com/gogf/gf/v2/frame/g"
-	"github.com/gogf/gf/v2/net/ghttp"
-	"github.com/gogf/gf/v2/text/gstr"
-	"github.com/gogf/gf/v2/util/gconv"
-	"github.com/gogf/gf/v2/util/gmode"
 	"hotgo/api/admin/common"
 	"hotgo/internal/consts"
 	"hotgo/internal/library/captcha"
 	"hotgo/internal/library/token"
 	"hotgo/internal/service"
 	"hotgo/utility/validate"
+
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/net/ghttp"
+	"github.com/gogf/gf/v2/text/gstr"
+	"github.com/gogf/gf/v2/util/gconv"
+	"github.com/gogf/gf/v2/util/gmode"
 )
 
 var Site = cSite{}
@@ -102,7 +103,7 @@ func (c *cSite) AccountLogin(ctx context.Context, req *common.AccountLoginReq) (
 		return
 	}
 
-	if !req.IsLock && login.CaptchaSwitch == 1 {
+	if !req.IsLock && login.CaptchaSwitch == consts.StatusEnabled {
 		// 校验 验证码
 		if !captcha.Verify(req.Cid, req.Code) {
 			err = gerror.New("验证码错误")

server/internal/library/captcha/captcha.go

@@ -7,12 +7,16 @@ package captcha
 
 import (
 	"context"
+	"image/color"
+
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/text/gstr"
 	"github.com/mojocn/base64Captcha"
-	"image/color"
 )
 
+// store 验证码存储方式
+var store = base64Captcha.DefaultMemStore
+
 // Generate 生成验证码
 func Generate(ctx context.Context) (id string, base64 string) {
 	// 字符
@@ -47,7 +51,7 @@ func Generate(ctx context.Context) (id string, base64 string) {
 		Fonts: []string{"chromohv.ttf"},
 	}
 
-	c := base64Captcha.NewCaptcha(driver.ConvertFonts(), base64Captcha.DefaultMemStore)
+	c := base64Captcha.NewCaptcha(driver.ConvertFonts(), store)
 	id, base64, _, err := c.Generate()
 	if err != nil {
 		g.Log().Errorf(ctx, "captcha.Generate err:%+v", err)
@@ -60,6 +64,5 @@ func Verify(id, answer string) bool {
 	if id == "" || answer == "" {
 		return false
 	}
-	c := base64Captcha.NewCaptcha(new(base64Captcha.DriverString), base64Captcha.DefaultMemStore)
-	return c.Verify(id, gstr.ToLower(answer), true)
+	return store.Verify(id, gstr.ToLower(answer), true)
 }

server/internal/logic/admin/member.go

@@ -629,17 +629,7 @@ func (s *sAdminMember) List(ctx context.Context, in *adminin.MemberListInp) (lis
 		mod = mod.WhereBetween(cols.CreatedAt, gtime.New(in.CreatedAt[0]), gtime.New(in.CreatedAt[1]))
 	}
 
-	totalCount, err = mod.Count()
-	if err != nil {
-		err = gerror.Wrap(err, "获取用户数据行失败！")
-		return
-	}
-
-	if totalCount == 0 {
-		return
-	}
-
-	if err = mod.Hook(hook.MemberInfo).Page(in.Page, in.PerPage).OrderDesc(cols.Id).Scan(&list); err != nil {
+	if err = mod.Hook(hook.MemberInfo).Page(in.Page, in.PerPage).OrderDesc(cols.Id).ScanAndCount(&list, &totalCount, true); err != nil {
 		err = gerror.Wrap(err, "获取用户列表失败！")
 		return
 	}

web/src/views/org/dept/dept.vue

@@ -21,6 +21,7 @@
         ref="actionRef"
         openChecked
         :columns="columns"
+        :pagination="false"
         :request="loadDataTable"
         :row-key="(row) => row.id"
         :actionColumn="actionColumn"