From 4243a710f702b601b1d5a7dc2c765191c191201d Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期三, 11 十二月 2019 17:59:48 +0800
Subject: [PATCH] 把插入es的so放进中间件
---
main.go | 51 ++++++++++++++++++++++++++++-----------------------
1 files changed, 28 insertions(+), 23 deletions(-)
diff --git a/main.go b/main.go
index ba6a6ec..fb30ecd 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,24 +29,24 @@
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")
- //}
- //logFile = logFile + "ruleprocess.log"
- //fmt.Println("鏃ュ織鍦板潃锛�",logFile)
- //logger.Config(logFile, logger.DebugLevel)
- //logger.SetSaveDays(7)
+ structure.Init(*env)
+ 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)
// log4go
- logger.LoadConfiguration("./logger/log.xml")
- logger.Info("鏃ュ織鍒濆鍖栨垚鍔燂紒")
+ //logger.LoadConfiguration("./logger/log.xml")
+ logger.Info("鏃ュ織鍒濆鍖栨垚鍔燂紒",logFile)
}
func main() {
@@ -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