From c0852d839f7aba0ef92eaa60f6b0c26efde5a49e Mon Sep 17 00:00:00 2001
From: kzkzzzz <test@test.local>
Date: Sun, 27 Jul 2025 20:50:37 +0800
Subject: [PATCH] update

---
 mymysql/batchwriter.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/mymysql/batchwriter.go b/mymysql/batchwriter.go
index d7b1835..197e638 100644
--- a/mymysql/batchwriter.go
+++ b/mymysql/batchwriter.go
@@ -223,7 +223,11 @@ func (bw *BatchWriter[T]) writeToDb(bd *batchData[T]) {
 
 }
 
-var disableGormLog = logger.Default.LogMode(logger.Silent)
+var DisableGormLog = logger.Default.LogMode(logger.Silent)
+
+func SessionDisableLog() *gorm.Session {
+	return &gorm.Session{Logger: DisableGormLog}
+}
 
 func (bw *BatchWriter[T]) asyncWriteToDb(jobIndex int, copyDataList []T) {
 	if len(copyDataList) == 0 {
@@ -240,7 +244,7 @@ func (bw *BatchWriter[T]) asyncWriteToDb(jobIndex int, copyDataList []T) {
 	if bw.config.noPrepare {
 		_sql := query.ToSQL(func(tx *gorm.DB) *gorm.DB {
 			return tx.Session(&gorm.Session{
-				Logger: disableGormLog,
+				Logger: DisableGormLog,
 			}).Create(&copyDataList)
 		})