80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"image/color"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"go-common/app/interface/main/captcha/conf"
|
||
|
|
"go-common/app/interface/main/captcha/dao"
|
||
|
|
"go-common/library/cache"
|
||
|
|
"go-common/library/ecode"
|
||
|
|
|
||
|
|
"github.com/golang/freetype/truetype"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Captcha captcha.
|
||
|
|
type Captcha struct {
|
||
|
|
frontColors []color.Color
|
||
|
|
bkgColors []color.Color
|
||
|
|
disturbLevel int
|
||
|
|
fonts []*truetype.Font
|
||
|
|
}
|
||
|
|
|
||
|
|
// Service captcha service.
|
||
|
|
type Service struct {
|
||
|
|
conf *conf.Config
|
||
|
|
dao *dao.Dao
|
||
|
|
captcha *Captcha
|
||
|
|
cacheCh *cache.Cache
|
||
|
|
// captcha mem.
|
||
|
|
init bool
|
||
|
|
lock sync.RWMutex
|
||
|
|
mCode map[string][]string
|
||
|
|
mImage map[string]map[string][]byte
|
||
|
|
}
|
||
|
|
|
||
|
|
// New new a service.
|
||
|
|
func New(c *conf.Config) (s *Service) {
|
||
|
|
s = &Service{
|
||
|
|
conf: c,
|
||
|
|
dao: dao.New(c),
|
||
|
|
captcha: newCaptcha(c.Captcha),
|
||
|
|
cacheCh: cache.New(1, 1024),
|
||
|
|
mCode: make(map[string][]string, len(c.Business)),
|
||
|
|
mImage: make(map[string]map[string][]byte, len(c.Business)),
|
||
|
|
}
|
||
|
|
go s.generaterProc()
|
||
|
|
return s
|
||
|
|
}
|
||
|
|
|
||
|
|
// Close close all dao.
|
||
|
|
func (s *Service) Close() {
|
||
|
|
s.dao.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Ping ping dao.
|
||
|
|
func (s *Service) Ping(c context.Context) error {
|
||
|
|
if !s.init {
|
||
|
|
return ecode.CaptchaNotCreate
|
||
|
|
}
|
||
|
|
return s.dao.Ping(c)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) generaterProc() {
|
||
|
|
waiter := &sync.WaitGroup{}
|
||
|
|
for _, b := range s.conf.Business {
|
||
|
|
waiter.Add(1)
|
||
|
|
go s.initGenerater(waiter, b.BusinessID, b.LenStart, b.LenEnd, b.Width, b.Length)
|
||
|
|
}
|
||
|
|
waiter.Wait()
|
||
|
|
s.init = true
|
||
|
|
for {
|
||
|
|
for _, b := range s.conf.Business {
|
||
|
|
go s.generater(b.BusinessID, b.LenStart, b.LenEnd, b.Width, b.Length)
|
||
|
|
}
|
||
|
|
time.Sleep(time.Duration(s.conf.Captcha.Interval))
|
||
|
|
}
|
||
|
|
}
|