From 0347e7652c415463c2c2c5ec95331b6df6faba8a Mon Sep 17 00:00:00 2001
From: boyce <boyce.duan@gmail.com>
Date: Mon, 18 Feb 2019 13:26:01 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=BC=82=E6=AD=A5=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sysmodule/DBModule.go | 61 +++++++++++++++++++++++++++++++++++++++----
 1 file changed, 56 insertions(+), 5 deletions(-)

diff --git a/sysmodule/DBModule.go b/sysmodule/DBModule.go
index 0e5ed64..73f2e21 100644
--- a/sysmodule/DBModule.go
+++ b/sysmodule/DBModule.go
@@ -14,6 +14,12 @@ import (
 	_ "github.com/go-sql-driver/mysql"
 )
 
+type SyncFun func()
+
+const (
+	MAX_EXECUTE_FUN = 10000
+)
+
 // DBModule ...
 type DBModule struct {
 	service.BaseModule
@@ -22,6 +28,9 @@ type DBModule struct {
 	UserName string
 	Password string
 	DBName   string
+
+	syncExecuteFun   chan SyncFun
+	syncCoroutineNum int
 }
 
 // DBResult ...
@@ -37,7 +46,8 @@ type DBResult struct {
 
 //OnInit ...
 func (slf *DBModule) OnInit() error {
-	return slf.Connect(100)
+	slf.syncExecuteFun = make(chan SyncFun, MAX_EXECUTE_FUN)
+	return slf.Connect(10)
 }
 
 // Next ...
@@ -238,6 +248,10 @@ func (slf *DBModule) Connect(maxConn int) error {
 	db.SetMaxIdleConns(maxConn)
 	db.SetConnMaxLifetime(time.Second * 90)
 
+	slf.syncCoroutineNum = maxConn
+	for i := 0; i < slf.syncCoroutineNum; i++ {
+		go slf.RunExecuteDBCoroutine()
+	}
 	return nil
 }
 
@@ -274,10 +288,20 @@ func (slf *DBModule) SyncQuery(query string, args ...interface{}) SyncDBResult {
 	ret := SyncDBResult{
 		sres: make(chan DBResult, 1),
 	}
-	go func() {
+
+	if len(slf.syncExecuteFun) >= MAX_EXECUTE_FUN {
+		dbret := DBResult{}
+		dbret.Err = fmt.Errorf("chan is full,sql:%s", query)
+		ret.sres <- dbret
+
+		return ret
+	}
+
+	slf.syncExecuteFun <- func() {
 		rsp := slf.Query(query, args...)
 		ret.sres <- rsp
-	}()
+	}
+
 	return ret
 }
 
@@ -301,9 +325,36 @@ func (slf *DBModule) SyncExec(query string, args ...interface{}) SyncDBResult {
 	ret := SyncDBResult{
 		sres: make(chan DBResult, 1),
 	}
-	go func() {
+
+	if len(slf.syncExecuteFun) >= MAX_EXECUTE_FUN {
+		dbret := DBResult{}
+		dbret.Err = fmt.Errorf("chan is full,sql:%s", query)
+		ret.sres <- dbret
+
+		return ret
+	}
+
+	slf.syncExecuteFun <- func() {
 		rsp := slf.Exec(query, args...)
 		ret.sres <- rsp
-	}()
+	}
+
 	return ret
 }
+
+func (slf *DBModule) RunExecuteDBCoroutine() {
+	slf.WaitGroup.Add(1)
+	defer slf.WaitGroup.Done()
+	for {
+		select {
+		case <-slf.ExitChan:
+			service.GetLogger().Printf(LEVER_WARN, "stopping module %s...", fmt.Sprintf("%T", slf))
+			fmt.Println("stopping module %s...", fmt.Sprintf("%T", slf))
+			return
+		case fun := <-slf.syncExecuteFun:
+			fun()
+		default:
+		}
+	}
+
+}