From 7a10088075f5d9a51ec92c8692bf46895aa3e9c3 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期一, 09 十二月 2019 14:33:37 +0800
Subject: [PATCH] ---
---
main.go | 43 ++++++++++++++++++++++++-------------------
1 files changed, 24 insertions(+), 19 deletions(-)
diff --git a/main.go b/main.go
index ba6a6ec..c42e868 100644
--- a/main.go
+++ b/main.go
@@ -3,13 +3,16 @@
import (
"flag"
+ "fmt"
+ "github.com/spf13/viper"
"sync"
"net/http"
_ "net/http/pprof"
"plugin"
//"github.com/spf13/viper"
- logger "github.com/alecthomas/log4go"
+ //"basic.com/valib/logger.git"
+ "basic.com/valib/logger.git"
"github.com/panjf2000/ants/v2"
"basic.com/pubsub/protomsg.git"
@@ -26,23 +29,23 @@
var surveyPort = flag.Int("surveyPort", 40007, "survey port") //蹇冭烦
var pubPort = flag.Int("pubPort", 50007, "pubsub port") //鏁版嵁鏇存柊
var initchan = make(chan bool)
-var env = flag.String("env","pro","env set")
+var env = flag.String("env","dev","env set")
func init() {
flag.Parse()
// 鏃ュ織鍒濆鍖�
insertdata.Init(*env)
- //var logFile = "./logger/"
- //if viper.GetString("LogBasePath") != "" {
- // logFile = viper.GetString("LogBasePath")
- //}
+ var logFile = "./logger/"
+ if viper.GetString("LogBasePath") != "" {
+ logFile = viper.GetString("LogBasePath")
+ }
//logFile = logFile + "ruleprocess.log"
//fmt.Println("鏃ュ織鍦板潃锛�",logFile)
- //logger.Config(logFile, logger.DebugLevel)
- //logger.SetSaveDays(7)
+ logger.Config(logFile, logger.DebugLevel)
+ logger.SetSaveDays(7)
// log4go
- logger.LoadConfiguration("./logger/log.xml")
+ //logger.LoadConfiguration("./logger/log.xml")
logger.Info("鏃ュ織鍒濆鍖栨垚鍔燂紒")
}
@@ -75,20 +78,22 @@
func nRecvImpl(c deliver.Deliver, index int) {
var msg []byte
- //var wg1 sync.WaitGroup
- //p,_ := ants.NewPool(100)
- //syncCalculateSum := func() {
- // Task(msg)
- // wg1.Done()
- //}
- //wg1.Wait()
+ var wg1 sync.WaitGroup
+ p,_ := ants.NewPool(20)
+ syncCalculateSum := func() {
+ Task(msg)
+ wg1.Done()
+ }
+ wg1.Wait()
var err error
for {
msg, err = c.Recv()
if err == nil {
- //wg1.Add(1)
- //_ = p.Submit(syncCalculateSum)
- Task(msg)
+ wg1.Add(1)
+ _ = p.Submit(syncCalculateSum)
+ //go Task(msg)
+ } else {
+ fmt.Println("鎺ユ敹鏁版嵁澶辫触",err)
}
}
}
--
Gitblit v1.8.0