From f62f2cd8fee44ecf7bc54f2635172e48dc348321 Mon Sep 17 00:00:00 2001
From: 554325746@qq.com <554325746@qq.com>
Date: 星期五, 23 八月 2019 10:34:14 +0800
Subject: [PATCH] add shm

---
 main.go |   20 ++++++++++++++++----
 1 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 15a31ac..53572a8 100644
--- a/main.go
+++ b/main.go
@@ -5,6 +5,7 @@
     "flag"
 	_ "net/http/pprof"
     "time"
+	"fmt"
 
 	"taskpubsub/camera"
 	"taskpubsub/sdk"
@@ -16,6 +17,8 @@
 
 var initchan = make(chan bool)
 
+var useShm bool
+
 func init(){
     var logFile = "./taskpubsub.log"
     var logSaveDays    =    15
@@ -23,13 +26,22 @@
     // 鏃ュ織鍒濆鍖�
     logger.Config(logFile, logger.InfoLevel)
     logger.SetSaveDays(logSaveDays)
-    logger.Info("loginit success !")
+	logger.Info("loginit success !")
+	
+	flag.BoolVar(&useShm, "shm", false, "use shm for performance")
 }     
         
-func     main() {
+func main() {
     flag.Parse()
     time.Sleep(time.Second)
 
+	if useShm{
+		logger.Info("USE SHARE MEMORY")
+		fmt.Println("USE SHARE MEMORY")
+	}else{
+		logger.Info("USE PIPE")
+		fmt.Println("USE PIPE")
+	}
 	// pprof 鐢ㄤ簬鍒嗘瀽鎬ц兘
 	go func() {
 		logger.Info(http.ListenAndServe("0.0.0.0:6061", nil))
@@ -38,8 +50,8 @@
 	go util.Init(initchan)
 
 	logger.Info("init ok !!!!, start sdk, task, camera init process ....", <-initchan)
-	sdk.Init()        //  鑾峰彇鎵�鏈夌畻娉昳d ,寤虹珛 sdk 涓婚锛� 寤虹珛sdk server(send, recv 杩愯)
+	sdk.Init(useShm)        //  鑾峰彇鎵�鏈夌畻娉昳d ,寤虹珛 sdk 涓婚锛� 寤虹珛sdk server(send, recv 杩愯)
 	tasktag.Init()   // 鑾峰彇鎵�鏈変换鍔★紝寤虹珛浠诲姟鏍囩锛� 鍦ㄦ暟鎹繘鍏ユ椂锛� 鎵撴爣绛�
-	camera.Init()   //鑾峰彇cid, taskid, sdkid ,鍏崇郴
+	camera.Init(useShm)   //鑾峰彇cid, taskid, sdkid ,鍏崇郴
 	select {}
 }

--
Gitblit v1.8.0