fix: 移除 modular.go 中不必要的代码,修复 timer.Ticker 释放后 handler 指针污染的问题

This commit is contained in:
kercylan98 2024-03-08 16:49:58 +08:00
parent f7c37016ce
commit 17cdad2c6e
3 changed files with 9 additions and 11 deletions

View File

@ -3,7 +3,6 @@ package modular
import (
"fmt"
"github.com/kercylan98/minotaur/utils/log"
"reflect"
"sync"
)
@ -27,14 +26,12 @@ func (m *modular) RegisterServices(s ...Service) {
func Run() {
m := application
var names = make(map[string]bool)
var tvm = make(map[reflect.Type]reflect.Value)
for i := 0; i < len(m.registerServices); i++ {
s := newService(m.registerServices[i])
if names[s.name] {
panic(fmt.Errorf("service %s is already registered", s.name))
}
names[s.name] = true
tvm[s.vof.Type()] = s.vof
m.services = append(m.services, s)
}

View File

@ -2,10 +2,10 @@ package timer
type Option func(ticker *Ticker)
// WithCaller 通过其他的 handle 执行 Caller
// WithCaller 通过其他的 handler 执行 Caller
func WithCaller(handle func(name string, caller func())) Option {
return func(ticker *Ticker) {
ticker.handle = handle
ticker.handler = handle
}
}

View File

@ -16,8 +16,8 @@ type Ticker struct {
timers map[string]*Scheduler
lock sync.RWMutex
handle func(name string, caller func())
mark string
handler func(name string, caller func())
mark string
}
// Mark 获取定时器的标记
@ -33,6 +33,7 @@ func (slf *Ticker) Release() {
slf.lock.Lock()
slf.mark = ""
slf.handler = nil
for name, scheduler := range slf.timers {
scheduler.close()
delete(slf.timers, name)
@ -94,8 +95,8 @@ func (slf *Ticker) CronByInstantly(name, expression string, handleFunc interface
f := reflect.ValueOf(handleFunc)
slf.lock.RLock()
defer slf.lock.RUnlock()
if slf.handle != nil {
slf.handle(name, func() {
if slf.handler != nil {
slf.handler(name, func() {
f.Call(values)
})
} else {
@ -147,9 +148,9 @@ func (slf *Ticker) loop(name string, after, interval time.Duration, expr *cronex
slf.lock.Lock()
slf.timers[name] = scheduler
if slf.handle != nil {
if slf.handler != nil {
scheduler.timer = slf.wheel.ScheduleFunc(scheduler, func() {
slf.handle(scheduler.Name(), scheduler.Caller)
slf.handler(scheduler.Name(), scheduler.Caller)
})
} else {
scheduler.timer = slf.wheel.ScheduleFunc(scheduler, scheduler.Caller)