113 lines
2.1 KiB
Go
113 lines
2.1 KiB
Go
package cronex
|
|
|
|
import (
|
|
"fmt"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/antlabs/timer"
|
|
)
|
|
|
|
func Test_Cronex(t *testing.T) {
|
|
table := []string{
|
|
"*/1 * * * * *", //每秒执行一次
|
|
}
|
|
|
|
cron := New()
|
|
go cron.Run()
|
|
|
|
count := 2
|
|
durationChan := make(chan time.Duration, count)
|
|
now := time.Now()
|
|
var tm TimerNoder
|
|
var err error
|
|
|
|
for _, tc := range table {
|
|
tm, err = cron.AddFunc(tc, func() {
|
|
|
|
durationChan <- time.Since(now)
|
|
})
|
|
if err != nil {
|
|
t.Logf("err(%v)", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// 3s之后关闭
|
|
go func() {
|
|
time.Sleep(time.Second * 3)
|
|
tm.Stop()
|
|
cron.Stop()
|
|
close(durationChan)
|
|
}()
|
|
|
|
count = 0
|
|
first := time.Duration(0)
|
|
for tv := range durationChan {
|
|
if first == 0 {
|
|
first = tv
|
|
continue
|
|
}
|
|
|
|
left := first + time.Duration(count)*time.Second
|
|
right := first + time.Duration(1.2*float64(time.Duration(count)*time.Second))
|
|
|
|
if tv < left || tv > right {
|
|
t.Logf("count(%d), tv(%v), tv < left(%v) || tv > right(%v)", count, tv, left, right)
|
|
return
|
|
}
|
|
count++
|
|
}
|
|
|
|
if count != 1 {
|
|
t.Logf("count(%d), count != 1, callback 没有调用", count)
|
|
return
|
|
}
|
|
}
|
|
|
|
// 测试下Next函数的时间可正确
|
|
func Test_Cronex_ParseNext(t *testing.T) {
|
|
|
|
var schedule timer.Next
|
|
schedule, err := standardParser.Parse("* * * * * *")
|
|
if err != nil {
|
|
t.Logf("err(%v)", err)
|
|
return
|
|
}
|
|
|
|
first := time.Duration(0)
|
|
for count := 1; count < 4; count++ {
|
|
now := schedule.Next(time.Now())
|
|
left := first + time.Duration(0.8*float64(time.Second))
|
|
right := first + time.Duration(1.2*float64(time.Second))
|
|
|
|
tv := time.Until(now)
|
|
if tv < left || tv > right {
|
|
t.Logf("tv(%v), tv < left(%v) || tv > right(%v)", tv, left, right)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// 多次运行的例子
|
|
func Test_Multiple(t *testing.T) {
|
|
cron := New()
|
|
count := int32(0)
|
|
|
|
go cron.Run()
|
|
max := int32(10)
|
|
for i := int32(0); i < max; i++ {
|
|
cron.AddFunc("* * * * * *", func() {
|
|
fmt.Printf("Every Second")
|
|
atomic.AddInt32(&count, 1)
|
|
})
|
|
}
|
|
|
|
time.Sleep(time.Duration(1.1 * float64(time.Second)))
|
|
cron.Stop()
|
|
if count != max {
|
|
t.Errorf("expected %d, got %d", max, count)
|
|
}
|
|
}
|