From 5601ab5ae21327f555b373f8c7b1004e9b878b0b Mon Sep 17 00:00:00 2001
From: orgin <boyce.duan@gmail.com>
Date: Thu, 17 Nov 2022 09:26:10 +0800
Subject: [PATCH] =?UTF-8?q?=E6=89=A9=E5=B1=95CountDocument=E6=9D=A1?=
 =?UTF-8?q?=E4=BB=B6=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sysmodule/mongodbmodule/mongodbmodule.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sysmodule/mongodbmodule/mongodbmodule.go b/sysmodule/mongodbmodule/mongodbmodule.go
index c1daa9f..a882538 100644
--- a/sysmodule/mongodbmodule/mongodbmodule.go
+++ b/sysmodule/mongodbmodule/mongodbmodule.go
@@ -52,10 +52,10 @@ func (mm *MongoModule) TakeSession() Session {
 	return Session{Client: mm.client, maxOperatorTimeOut: mm.maxOperatorTimeOut}
 }
 
-func (s *Session) CountDocument(db string, collection string) (int64, error) {
+func (s *Session) CountDocument(db string, collection string, filter interface{}) (int64, error) {
 	ctxTimeout, cancel := s.GetDefaultContext()
 	defer cancel()
-	return s.Database(db).Collection(collection).CountDocuments(ctxTimeout, bson.D{})
+	return s.Database(db).Collection(collection).CountDocuments(ctxTimeout, filter)
 }
 
 func (s *Session) NextSeq(db string, collection string, id interface{}) (int, error) {