|
|
|
@ -2,8 +2,8 @@ package gojobs
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"context"
|
|
|
|
|
pb2 "go.dtapp.net/library/gojobs/pb"
|
|
|
|
|
"go.dtapp.net/library/gouuid"
|
|
|
|
|
"go.dtapp.net/library/utils/gojobs/pb"
|
|
|
|
|
"go.dtapp.net/library/utils/gouuid"
|
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
@ -16,7 +16,7 @@ type WorkerConfig struct {
|
|
|
|
|
// Worker 工作
|
|
|
|
|
type Worker struct {
|
|
|
|
|
WorkerConfig // 配置
|
|
|
|
|
Pub pb2.PubSubClient // 订阅
|
|
|
|
|
Pub pb.PubSubClient // 订阅
|
|
|
|
|
Conn *grpc.ClientConn // 链接信息
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -44,14 +44,14 @@ func NewWorker(config *WorkerConfig) *Worker {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 新建一个客户端
|
|
|
|
|
w.Pub = pb2.NewPubSubClient(w.Conn)
|
|
|
|
|
w.Pub = pb.NewPubSubClient(w.Conn)
|
|
|
|
|
|
|
|
|
|
return w
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// SubscribeCron 订阅服务
|
|
|
|
|
func (w *Worker) SubscribeCron() pb2.PubSub_SubscribeClient {
|
|
|
|
|
stream, err := w.Pub.Subscribe(context.Background(), &pb2.SubscribeRequest{
|
|
|
|
|
func (w *Worker) SubscribeCron() pb.PubSub_SubscribeClient {
|
|
|
|
|
stream, err := w.Pub.Subscribe(context.Background(), &pb.SubscribeRequest{
|
|
|
|
|
Id: gouuid.GetUuId(),
|
|
|
|
|
Value: prefix,
|
|
|
|
|
Ip: w.ClientIp,
|
|
|
|
@ -63,8 +63,8 @@ func (w *Worker) SubscribeCron() pb2.PubSub_SubscribeClient {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// StartCron 启动任务
|
|
|
|
|
func (w *Worker) StartCron() pb2.PubSub_SubscribeClient {
|
|
|
|
|
stream, err := w.Pub.Subscribe(context.Background(), &pb2.SubscribeRequest{
|
|
|
|
|
func (w *Worker) StartCron() pb.PubSub_SubscribeClient {
|
|
|
|
|
stream, err := w.Pub.Subscribe(context.Background(), &pb.SubscribeRequest{
|
|
|
|
|
Id: gouuid.GetUuId(),
|
|
|
|
|
Value: prefixSprintf(w.ClientIp),
|
|
|
|
|
Ip: w.ClientIp,
|