From a45d5a83c5a283444a407d648432e9dd08f89478 Mon Sep 17 00:00:00 2001
From: chenshijun <csj_sky@126.com>
Date: 星期三, 11 九月 2019 20:34:16 +0800
Subject: [PATCH] 增加配置文件处理参数

---
 util/util.go |   19 +++++++++++++------
 1 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/util/util.go b/util/util.go
index 264880a..b8fbcea 100644
--- a/util/util.go
+++ b/util/util.go
@@ -1,20 +1,23 @@
 package util
 
 import (
+	"basic.com/valib/deliver.git"
 	"context"
 	"errors"
-	"fmt"
 
-	"basic.com/valib/deliver.git"
-
-	// "github.com/pierrec/lz4"
-	"taskpubsub/logger"
+	"basic.com/valib/logger.git"
 )
+
+const SHMMaxTryCount = 3
 
 type SocketContext struct {
 	Sock    deliver.Deliver
 	Context context.Context
 	Cancel  context.CancelFunc
+
+	Mode   int
+	URL    string
+	UseSHM bool
 }
 
 //  1. oldstring element is not in new  : abandon(delete)
@@ -89,11 +92,15 @@
 	socket.Context = ctx
 	socket.Cancel = cancel
 
+	socket.Mode = mode
+	socket.URL = url
+	socket.UseSHM = shm
+
 	if shm {
 		var err error
 		socket.Sock, err = deliver.NewServerWithError(deliver.Mode(mode), url)
 		if err != nil {
-			fmt.Println("TASK CREATE SERVER SHM ERROR: ", err)
+			logger.Info("TASK CREATE SERVER SHM ERROR: ", err)
 		}
 	} else {
 		socket.Sock = deliver.NewServer(deliver.Mode(mode), url)

--
Gitblit v1.8.0