update

internal/erpserver/service/v1/common/captcha.go

@@ -6,23 +6,25 @@ import (
 	"github.com/mojocn/base64Captcha"
 )
 
-type captchaService struct{}
-
-func NewCaptchaService() v1.CaptchaService {
-	return &captchaService{}
+type captchaService struct {
+	store base64Captcha.Store
 }
 
-var captchaStore = base64Captcha.DefaultMemStore
+func NewCaptchaService() v1.CaptchaService {
+	return &captchaService{
+		store: base64Captcha.DefaultMemStore,
+	}
+}
 
 func (b *captchaService) Generate(height int, width int, length int, maxSkew float64, dotCount int) (id, b64s, answer string, err error) {
 	driver := base64Captcha.NewDriverDigit(height, width, length, maxSkew, dotCount)
 	// driver := base64Captcha.NewDriverString(config.File.Captcha.ImgHeight,
 	// 	config.File.Captcha.ImgWidth,
 	// 	6, 1, keyLong, source, nil, nil, nil)
-	cp := base64Captcha.NewCaptcha(driver, captchaStore)
+	cp := base64Captcha.NewCaptcha(driver, b.store)
 	return cp.Generate()
 }
 
 func (b *captchaService) Verify(id, answer string, clear bool) bool {
-	return captchaStore.Verify(id, answer, clear)
+	return b.store.Verify(id, answer, clear)
 }