From e0875bfe6196fdf4abdc7c099bbb59e7282f1b48 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 22 五月 2019 13:38:04 +0800
Subject: [PATCH] update deliver

---
 runShm.go |   21 ++++++++++++++++++---
 1 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/shm.go b/runShm.go
similarity index 78%
rename from shm.go
rename to runShm.go
index abb0ba5..1f05617 100644
--- a/shm.go
+++ b/runShm.go
@@ -38,8 +38,15 @@
 }
 
 func shmSender(url string, args ...interface{}) {
-	s := deliver.NewServer(deliver.Shm, url, args...)
-
+	s, err := deliver.NewServerWithError(deliver.Shm, url, args...)
+	for {
+		if err == nil {
+			break
+		}
+		fmt.Println("create shm error : ", err)
+		time.Sleep(1 * time.Second)
+		s, err = deliver.NewServerWithError(deliver.Shm, url, args)
+	}
 	go shmSenderImpl(s)
 
 	c := make(chan os.Signal, 1)
@@ -97,7 +104,15 @@
 	var cs []deliver.Deliver
 	for i := 0; i < count; i++ {
 		wg.Add(1)
-		c := deliver.NewClient(deliver.Shm, url)
+		c, err := deliver.NewClientWithError(deliver.Shm, url)
+		for {
+			if err == nil {
+				break
+			}
+			time.Sleep(1 * time.Second)
+			c, err = deliver.NewClientWithError(deliver.Shm, url)
+			fmt.Println(i, " client create failed : ", err)
+		}
 		cs = append(cs, c)
 		go shmRecvImpl(&wg, c, url, i)
 	}

--
Gitblit v1.8.0