From 7737d3be7814bd5df735a5c16c07885000c7ae32 Mon Sep 17 00:00:00 2001
From: meixiongfeng <766881921@qq.com>
Date: Tue, 14 Jun 2022 15:26:47 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BB=93=E6=9E=84=E8=B0=83=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/model/db_comp.go         |  4 ++--
 modules/model/dbservice_comp.go  | 10 +++++-----
 modules/{ => model}/mail_test.go |  0
 sys/cache/init_test.go           | 33 ++++++++++++++++++++++++++++++++
 4 files changed, 40 insertions(+), 7 deletions(-)
 rename modules/{ => model}/mail_test.go (100%)

diff --git a/modules/model/db_comp.go b/modules/model/db_comp.go
index a6f37b8d6..083e8fbba 100644
--- a/modules/model/db_comp.go
+++ b/modules/model/db_comp.go
@@ -5,9 +5,9 @@ import (
 	"errors"
 	"go_dreamfactory/comm"
 	"go_dreamfactory/lego/core"
-	"go_dreamfactory/lego/core/cbase"
 	"go_dreamfactory/lego/sys/log"
 	"go_dreamfactory/lego/sys/mgo"
+	"go_dreamfactory/modules"
 
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo/options"
@@ -18,7 +18,7 @@ const (
 )
 
 type DB_Comp struct {
-	cbase.ModuleCompBase
+	modules.MComp_DBComp
 	mgo mgo.ISys
 }
 
diff --git a/modules/model/dbservice_comp.go b/modules/model/dbservice_comp.go
index ade6efae5..cfaa543b4 100644
--- a/modules/model/dbservice_comp.go
+++ b/modules/model/dbservice_comp.go
@@ -8,13 +8,13 @@ import (
 
 type DBService_Comp struct {
 	cbase.ModuleCompBase
-	task chan string
-	DB_Comp
+	task   chan string
+	module *Model
 }
 
 func (this *DBService_Comp) Init(service core.IService, module core.IModule, comp core.IModuleComp, options core.IModuleOptions) (err error) {
 	this.ModuleCompBase.Init(service, module, comp, options)
-
+	this.module = module.(*Model)
 	return
 }
 
@@ -27,9 +27,9 @@ func (this *DBService_Comp) run() {
 	for {
 		select {
 		case v := <-this.task:
-			this.Model_UpdateUserDataByUid(v)
+			this.module.DB().(*DB_Comp).Model_UpdateUserDataByUid(v)
 		case <-time.After(time.Second):
-			this.Model_UpdateDBByLog()
+			this.module.DB().(*DB_Comp).Model_UpdateDBByLog()
 		}
 	}
 }
diff --git a/modules/mail_test.go b/modules/model/mail_test.go
similarity index 100%
rename from modules/mail_test.go
rename to modules/model/mail_test.go
diff --git a/sys/cache/init_test.go b/sys/cache/init_test.go
index af9cc838e..0e420d0bf 100644
--- a/sys/cache/init_test.go
+++ b/sys/cache/init_test.go
@@ -2,10 +2,16 @@ package cache_test
 
 import (
 	"fmt"
+	"go_dreamfactory/comm"
+	"go_dreamfactory/lego/sys/log"
+	"go_dreamfactory/pb"
 	"go_dreamfactory/sys/cache"
 	"go_dreamfactory/sys/db"
 	"os"
 	"testing"
+	"time"
+
+	"go.mongodb.org/mongo-driver/bson/primitive"
 )
 
 //测试环境下初始化db和cache 系统
@@ -19,6 +25,33 @@ func TestMain(m *testing.M) {
 		return
 	}
 
+	_mail := &pb.DB_MailData{
+
+		UserId: "uid123",
+		Title:  "系统邮件",
+
+		Contex:     "恭喜获得专属礼包一份",
+		CreateTime: uint64(time.Now().Unix()),
+		DueTime:    uint64(time.Now().Unix()) + 30*24*3600,
+		Check:      false,
+		Reward:     false,
+	}
+	//db.InsertModelLogs("mail", "uid123", _mail)
+	//InsertModelLogs("mail", "uid123", _mail)
+	data := &comm.Autogenerated{
+		ID:  primitive.NewObjectID().Hex(),
+		UID: "uid123",
+		Act: string(comm.LogHandleType_Insert),
+	}
+	data.D = append(data.D, "mail") // D[0]
+	data.D = append(data.D, _mail)  // D[1]
+	for i := 0; i < 100000; i++ {
+		_, err1 := db.Defsys.Mgo().InsertOne("model", data)
+		if err1 != nil {
+			log.Errorf("insert model db err %v", err1)
+		}
+	}
+
 	defer os.Exit(m.Run())
 
 }