From e5dfbb23ac6fd3fc857fa93c5f6c4cb934db1458 Mon Sep 17 00:00:00 2001
From: gigibox <gigibox@163.com>
Date: 星期二, 20 六月 2023 16:18:48 +0800
Subject: [PATCH] 添加sql语句校验, 查询接口只允许查询操作
---
kingdee/db.go | 25 ++++++++++++++++++++-----
1 files changed, 20 insertions(+), 5 deletions(-)
diff --git a/kingdee/db.go b/kingdee/db.go
index 5637219..d932284 100644
--- a/kingdee/db.go
+++ b/kingdee/db.go
@@ -2,8 +2,13 @@
import (
"fmt"
- "github.com/jinzhu/gorm"
+ "kingdee-dbapi/logger"
+
+ //"gorm.io/driver/sqlserver"
+ //"gorm.io/gorm"
+
+ "github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/mssql"
)
@@ -17,8 +22,11 @@
sqlServer := fmt.Sprintf("sqlserver://%s:%s@%s:1433?database=%s;encrypt=disable;",
username, password, addr, dbName)
+ logger.Debug("sqlServer鍦板潃:%s", sqlServer)
// 鎵撳紑鏁版嵁搴撹繛鎺�
db, err = gorm.Open("mssql", sqlServer)
+ // gorm 2.0
+ //db, err = gorm.Open(sqlserver.Open(sqlServer), &gorm.Config{})
if err != nil {
return err
}
@@ -30,9 +38,16 @@
return db
}
-// CloseDB .
+// 2.0 CloseDB .
+//func CloseDB() {
+// sqlDB, err := db.DB()
+// if err != nil {
+// return
+// }
+//
+// sqlDB.Close()
+//}
+
func CloseDB() {
- if db != nil {
- db.Close()
- }
+ db.Close()
}
--
Gitblit v1.8.0