- repair bug

client.go

@@ -0,0 +1,38 @@
+package gojobs
+
+import (
+	"google.golang.org/grpc"
+)
+
+// ClientConfig 客户端配置
+type ClientConfig struct {
+	Address string // 服务端口 127.0.0.1:8888
+}
+
+// Client 定时任务
+type Client struct {
+	ClientConfig                  // 配置
+	Conn         *grpc.ClientConn // 链接信息
+}
+
+// NewClient 创建客户端
+func NewClient(config *ClientConfig) *Client {
+
+	if config.Address == "" {
+		panic("[客户端]请填写服务端口")
+	}
+
+	c := &Client{}
+
+	c.Address = config.Address
+
+	var err error
+
+	// 建立连接 获取client
+	c.Conn, err = grpc.Dial(c.Address, grpc.WithInsecure())
+	if err != nil {
+		panic("[客户端]{连接失败} " + err.Error())
+	}
+
+	return c
+}

cron.go

@@ -23,7 +23,7 @@ type Cron struct {
 func NewCron(config *CronConfig) *Cron {
 
 	if config.Address == "" {
-		panic("请填写服务端口")
+		panic("[定时任务]请填写服务端口")
 	}
 
 	c := &Cron{}
@@ -35,7 +35,7 @@ func NewCron(config *CronConfig) *Cron {
 	// 建立连接 获取client
 	c.Conn, err = grpc.Dial(c.Address, grpc.WithInsecure())
 	if err != nil {
-		panic("连接失败: " + err.Error())
+		panic("[定时任务]{连接失败} " + err.Error())
 	}
 
 	// 新建一个客户端
@@ -45,11 +45,12 @@ func NewCron(config *CronConfig) *Cron {
 }
 
 // Send 发送
-func (c *Cron) Send(in *pb.PublishRequest) (*pb.PublishRequest, error) {
+func (c *Cron) Send(in *pb.PublishRequest) (*pb.PublishResponse, error) {
+	log.Printf("[定时任务]{广播开始} 编号：%s 类型：%s ip：%s\n", in.GetId(), in.GetValue(), in.GetIp())
 	stream, err := c.Pub.Publish(context.Background(), in)
 	if err != nil {
-		log.Printf("[定时任务]发送失败：%v\n", err)
+		log.Printf("[定时任务]{广播失败} 编号：%s %v\n", in.GetId(), err)
 	}
-	log.Println("[定时任务]发送成功", stream)
+	log.Printf("[定时任务]{广播成功} 编号：%s 类型：%s ip：%s\n", in.GetId(), in.GetValue(), in.GetIp())
 	return stream, err
 }

server.go

@@ -28,7 +28,7 @@ type Server struct {
 func NewServer(config *ServerConfig) *Server {
 
 	if config.Address == "" {
-		panic("请填写服务端口")
+		panic("[服务中转]请填写服务端口")
 	}
 
 	s := &Server{}
@@ -54,14 +54,14 @@ func (s *Server) StartUp() error {
 	// 监听本地端口
 	lis, err := net.Listen("tcp", s.Address)
 	if err != nil {
-		return errors.New("创建监听失败:" + err.Error())
+		return errors.New("[服务中转]{创建监听失败} " + err.Error())
 	}
-	log.Printf("正在监听的地址：%v", lis.Addr())
+	log.Println("[服务中转]{监听] ", lis.Addr())
 
 	// 启动grpc
 	err = s.Conn.Serve(lis)
 	if err != nil {
-		return errors.New("创建服务失败:" + err.Error())
+		return errors.New("[服务中转]{创建服务失败} " + err.Error())
 	}
 
 	return nil

worker.go

@@ -21,7 +21,7 @@ type Worker struct {
 func NewWorker(config *WorkerConfig) *Worker {
 
 	if config.Address == "" {
-		panic("请填写服务端口")
+		panic("[工作线]请填写服务端口")
 	}
 
 	w := &Worker{}
@@ -33,7 +33,7 @@ func NewWorker(config *WorkerConfig) *Worker {
 	// 建立连接 获取client
 	w.Conn, err = grpc.Dial(w.Address, grpc.WithInsecure())
 	if err != nil {
-		panic("连接失败: " + err.Error())
+		panic("[工作线]{连接失败} " + err.Error())
 	}
 
 	// 新建一个客户端