parent
1a272f93df
commit
75ffa84daa
@ -0,0 +1,147 @@
|
|||||||
|
package gojobs
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/robfig/cron/v3"
|
||||||
|
"go.dtapp.net/gojobs/pb"
|
||||||
|
"go.dtapp.net/gouuid"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"sync"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestJobs(t *testing.T) {
|
||||||
|
wg := sync.WaitGroup{}
|
||||||
|
wg.Add(4)
|
||||||
|
go testServer(&wg)
|
||||||
|
go testCron(&wg)
|
||||||
|
go testWorker1(&wg)
|
||||||
|
go testWorker2(&wg)
|
||||||
|
wg.Wait()
|
||||||
|
}
|
||||||
|
|
||||||
|
func testServer(wg *sync.WaitGroup) {
|
||||||
|
|
||||||
|
server := NewServer(&ServerConfig{
|
||||||
|
PublishTimeout: time.Millisecond * 100,
|
||||||
|
PubBuffer: 10,
|
||||||
|
Address: "0.0.0.0:8888",
|
||||||
|
})
|
||||||
|
|
||||||
|
// 启动定时任务
|
||||||
|
server.StartCron()
|
||||||
|
|
||||||
|
// 启动服务
|
||||||
|
server.StartUp()
|
||||||
|
|
||||||
|
<-make(chan bool)
|
||||||
|
|
||||||
|
wg.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func testCron(wg *sync.WaitGroup) {
|
||||||
|
|
||||||
|
server := NewCron(&CronConfig{
|
||||||
|
Address: "localhost:8888",
|
||||||
|
})
|
||||||
|
defer server.Conn.Close()
|
||||||
|
|
||||||
|
// 创建一个cron实例 精确到秒
|
||||||
|
c := cron.New(cron.WithSeconds())
|
||||||
|
|
||||||
|
// 每隔15秒执行一次
|
||||||
|
_, _ = c.AddFunc("*/15 * * * * *", func() {
|
||||||
|
|
||||||
|
server.Send(&pb.PublishRequest{
|
||||||
|
Id: gouuid.GetUuId(),
|
||||||
|
Value: prefix,
|
||||||
|
Method: "wechat.1.send",
|
||||||
|
Ip: "127.0.0.1",
|
||||||
|
})
|
||||||
|
|
||||||
|
})
|
||||||
|
|
||||||
|
// 每隔30秒执行一次
|
||||||
|
_, _ = c.AddFunc("*/30 * * * * *", func() {
|
||||||
|
|
||||||
|
server.Send(&pb.PublishRequest{
|
||||||
|
Id: gouuid.GetUuId(),
|
||||||
|
Value: prefix,
|
||||||
|
Method: "wechat.2.send",
|
||||||
|
Ip: "14.155.157.19",
|
||||||
|
})
|
||||||
|
|
||||||
|
})
|
||||||
|
|
||||||
|
// 启动任务
|
||||||
|
c.Start()
|
||||||
|
|
||||||
|
// 关闭任务
|
||||||
|
defer c.Stop()
|
||||||
|
select {}
|
||||||
|
|
||||||
|
wg.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func testWorker1(wg *sync.WaitGroup) {
|
||||||
|
|
||||||
|
server := NewWorker(&WorkerConfig{
|
||||||
|
Address: "localhost:8888",
|
||||||
|
ClientIp: "127.0.0.1",
|
||||||
|
})
|
||||||
|
defer server.Conn.Close()
|
||||||
|
|
||||||
|
// 订阅服务
|
||||||
|
stream := server.SubscribeCron()
|
||||||
|
|
||||||
|
// 启动任务,会想过滤器函数,订阅者应该收到的信息为 cron:任务名称
|
||||||
|
//stream := server.StartCron()
|
||||||
|
|
||||||
|
// 阻塞遍历流,输出结果
|
||||||
|
for {
|
||||||
|
reply, err := stream.Recv()
|
||||||
|
if io.EOF == err {
|
||||||
|
log.Println("[跑业务1]已关闭:", err.Error())
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if nil != err {
|
||||||
|
log.Println("[跑业务1]异常:", err.Error())
|
||||||
|
break
|
||||||
|
}
|
||||||
|
log.Printf("[跑业务1]{收到}编号:%s 方法:%s\n", reply.GetId(), reply.GetMethod())
|
||||||
|
}
|
||||||
|
|
||||||
|
wg.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func testWorker2(wg *sync.WaitGroup) {
|
||||||
|
|
||||||
|
server := NewWorker(&WorkerConfig{
|
||||||
|
Address: "localhost:8888",
|
||||||
|
ClientIp: "14.155.157.19",
|
||||||
|
})
|
||||||
|
defer server.Conn.Close()
|
||||||
|
|
||||||
|
// 订阅服务
|
||||||
|
stream := server.SubscribeCron()
|
||||||
|
|
||||||
|
// 启动任务,会想过滤器函数,订阅者应该收到的信息为 cron:任务名称
|
||||||
|
//stream := server.StartCron()
|
||||||
|
|
||||||
|
// 阻塞遍历流,输出结果
|
||||||
|
for {
|
||||||
|
reply, err := stream.Recv()
|
||||||
|
if io.EOF == err {
|
||||||
|
log.Println("[跑业务2]已关闭:", err.Error())
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if nil != err {
|
||||||
|
log.Println("[跑业务2]异常:", err.Error())
|
||||||
|
break
|
||||||
|
}
|
||||||
|
log.Printf("[跑业务2]{收到}编号:%s 方法:%s\n", reply.GetId(), reply.GetMethod())
|
||||||
|
}
|
||||||
|
|
||||||
|
wg.Done()
|
||||||
|
}
|
@ -0,0 +1,70 @@
|
|||||||
|
package gojobs
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
CodeAbnormal = 0 // 异常
|
||||||
|
CodeError = http.StatusInternalServerError // 失败
|
||||||
|
CodeSuccess = http.StatusOK // 成功
|
||||||
|
CodeEnd = http.StatusCreated // 结束
|
||||||
|
)
|
||||||
|
|
||||||
|
// 每隔n秒执行一次
|
||||||
|
const specSeconds = "*/%d * * * * *"
|
||||||
|
|
||||||
|
// GetSpecSeconds 每隔n秒执行一次
|
||||||
|
var GetSpecSeconds = func(n int64) string {
|
||||||
|
if n < 0 && n > 59 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return fmt.Sprintf(specSeconds, n)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetFrequencySeconds 每隔n秒执行一次
|
||||||
|
var GetFrequencySeconds = func(n int64) int64 {
|
||||||
|
if n < 0 && n > 59 {
|
||||||
|
return -1
|
||||||
|
}
|
||||||
|
return n
|
||||||
|
}
|
||||||
|
|
||||||
|
// 每隔n分钟执行一次
|
||||||
|
const specMinutes = "0 */%d * * * *"
|
||||||
|
|
||||||
|
// GetSpecMinutes 每隔n分钟执行一次
|
||||||
|
var GetSpecMinutes = func(n int64) string {
|
||||||
|
if n < 0 && n > 59 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return fmt.Sprintf(specMinutes, n)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetFrequencyMinutes 每隔n分钟执行一次
|
||||||
|
var GetFrequencyMinutes = func(n int64) int64 {
|
||||||
|
if n < 0 && n > 59 {
|
||||||
|
return -1
|
||||||
|
}
|
||||||
|
return n * 60
|
||||||
|
}
|
||||||
|
|
||||||
|
// 每天n点执行一次
|
||||||
|
const specHour = "0 0 */%d * * *"
|
||||||
|
|
||||||
|
// GetSpecHour 每天n点执行一次
|
||||||
|
var GetSpecHour = func(n int64) string {
|
||||||
|
if n < 0 && n > 23 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return fmt.Sprintf(specHour, n)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetFrequencyHour 每天n点执行一次
|
||||||
|
var GetFrequencyHour = func(n int64) int64 {
|
||||||
|
if n < 0 && n > 23 {
|
||||||
|
return -1
|
||||||
|
}
|
||||||
|
return n * 60 * 60
|
||||||
|
}
|
@ -1,147 +1,18 @@
|
|||||||
package gojobs
|
package gojobs
|
||||||
|
|
||||||
import (
|
import "testing"
|
||||||
"github.com/robfig/cron/v3"
|
|
||||||
"go.dtapp.net/gojobs/pb"
|
|
||||||
"go.dtapp.net/gouuid"
|
|
||||||
"io"
|
|
||||||
"log"
|
|
||||||
"sync"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestJobs(t *testing.T) {
|
func TestSpec(t *testing.T) {
|
||||||
wg := sync.WaitGroup{}
|
t.Log(GetSpecSeconds(10))
|
||||||
wg.Add(4)
|
t.Log(GetFrequencySeconds(10))
|
||||||
go testServer(&wg)
|
|
||||||
go testCron(&wg)
|
|
||||||
go testWorker1(&wg)
|
|
||||||
go testWorker2(&wg)
|
|
||||||
wg.Wait()
|
|
||||||
}
|
|
||||||
|
|
||||||
func testServer(wg *sync.WaitGroup) {
|
|
||||||
|
|
||||||
server := NewServer(&ServerConfig{
|
|
||||||
PublishTimeout: time.Millisecond * 100,
|
|
||||||
PubBuffer: 10,
|
|
||||||
Address: "0.0.0.0:8888",
|
|
||||||
})
|
|
||||||
|
|
||||||
// 启动定时任务
|
|
||||||
server.StartCron()
|
|
||||||
|
|
||||||
// 启动服务
|
|
||||||
server.StartUp()
|
|
||||||
|
|
||||||
<-make(chan bool)
|
|
||||||
|
|
||||||
wg.Done()
|
|
||||||
}
|
|
||||||
|
|
||||||
func testCron(wg *sync.WaitGroup) {
|
|
||||||
|
|
||||||
server := NewCron(&CronConfig{
|
|
||||||
Address: "localhost:8888",
|
|
||||||
})
|
|
||||||
defer server.Conn.Close()
|
|
||||||
|
|
||||||
// 创建一个cron实例 精确到秒
|
|
||||||
c := cron.New(cron.WithSeconds())
|
|
||||||
|
|
||||||
// 每隔15秒执行一次
|
|
||||||
_, _ = c.AddFunc("*/15 * * * * *", func() {
|
|
||||||
|
|
||||||
server.Send(&pb.PublishRequest{
|
|
||||||
Id: gouuid.GetUuId(),
|
|
||||||
Value: prefix,
|
|
||||||
Method: "wechat.1.send",
|
|
||||||
Ip: "127.0.0.1",
|
|
||||||
})
|
|
||||||
|
|
||||||
})
|
|
||||||
|
|
||||||
// 每隔30秒执行一次
|
|
||||||
_, _ = c.AddFunc("*/30 * * * * *", func() {
|
|
||||||
|
|
||||||
server.Send(&pb.PublishRequest{
|
|
||||||
Id: gouuid.GetUuId(),
|
|
||||||
Value: prefix,
|
|
||||||
Method: "wechat.2.send",
|
|
||||||
Ip: "14.155.157.19",
|
|
||||||
})
|
|
||||||
|
|
||||||
})
|
|
||||||
|
|
||||||
// 启动任务
|
|
||||||
c.Start()
|
|
||||||
|
|
||||||
// 关闭任务
|
|
||||||
defer c.Stop()
|
|
||||||
select {}
|
|
||||||
|
|
||||||
wg.Done()
|
|
||||||
}
|
|
||||||
|
|
||||||
func testWorker1(wg *sync.WaitGroup) {
|
|
||||||
|
|
||||||
server := NewWorker(&WorkerConfig{
|
|
||||||
Address: "localhost:8888",
|
|
||||||
ClientIp: "127.0.0.1",
|
|
||||||
})
|
|
||||||
defer server.Conn.Close()
|
|
||||||
|
|
||||||
// 订阅服务
|
|
||||||
stream := server.SubscribeCron()
|
|
||||||
|
|
||||||
// 启动任务,会想过滤器函数,订阅者应该收到的信息为 cron:任务名称
|
|
||||||
//stream := server.StartCron()
|
|
||||||
|
|
||||||
// 阻塞遍历流,输出结果
|
|
||||||
for {
|
|
||||||
reply, err := stream.Recv()
|
|
||||||
if io.EOF == err {
|
|
||||||
log.Println("[跑业务1]已关闭:", err.Error())
|
|
||||||
break
|
|
||||||
}
|
|
||||||
if nil != err {
|
|
||||||
log.Println("[跑业务1]异常:", err.Error())
|
|
||||||
break
|
|
||||||
}
|
|
||||||
log.Printf("[跑业务1]{收到}编号:%s 方法:%s\n", reply.GetId(), reply.GetMethod())
|
|
||||||
}
|
|
||||||
|
|
||||||
wg.Done()
|
|
||||||
}
|
|
||||||
|
|
||||||
func testWorker2(wg *sync.WaitGroup) {
|
|
||||||
|
|
||||||
server := NewWorker(&WorkerConfig{
|
|
||||||
Address: "localhost:8888",
|
|
||||||
ClientIp: "14.155.157.19",
|
|
||||||
})
|
|
||||||
defer server.Conn.Close()
|
|
||||||
|
|
||||||
// 订阅服务
|
|
||||||
stream := server.SubscribeCron()
|
|
||||||
|
|
||||||
// 启动任务,会想过滤器函数,订阅者应该收到的信息为 cron:任务名称
|
|
||||||
//stream := server.StartCron()
|
|
||||||
|
|
||||||
// 阻塞遍历流,输出结果
|
t.Log(GetSpecMinutes(1))
|
||||||
for {
|
t.Log(GetFrequencyMinutes(1))
|
||||||
reply, err := stream.Recv()
|
t.Log(GetSpecMinutes(10))
|
||||||
if io.EOF == err {
|
t.Log(GetFrequencyMinutes(10))
|
||||||
log.Println("[跑业务2]已关闭:", err.Error())
|
t.Log(GetSpecMinutes(30))
|
||||||
break
|
t.Log(GetFrequencyMinutes(30))
|
||||||
}
|
|
||||||
if nil != err {
|
|
||||||
log.Println("[跑业务2]异常:", err.Error())
|
|
||||||
break
|
|
||||||
}
|
|
||||||
log.Printf("[跑业务2]{收到}编号:%s 方法:%s\n", reply.GetId(), reply.GetMethod())
|
|
||||||
}
|
|
||||||
|
|
||||||
wg.Done()
|
t.Log(GetSpecHour(10))
|
||||||
|
t.Log(GetFrequencyHour(10))
|
||||||
}
|
}
|
||||||
|
@ -1,3 +1,3 @@
|
|||||||
package gojobs
|
package gojobs
|
||||||
|
|
||||||
const Version = "1.0.28"
|
const Version = "1.0.29"
|
||||||
|
Loading…
Reference in new issue