From c1a82cf1b089aa2ef2fa857e87d8e58e5a320be7 Mon Sep 17 00:00:00 2001
From: guzeng <gz@flyrise.cn>
Date: Wed, 10 Mar 2021 17:16:07 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=9F=A5=E8=AF=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 chain.go | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/chain.go b/chain.go
index a9e4a76..5046c16 100644
--- a/chain.go
+++ b/chain.go
@@ -29,9 +29,23 @@ type Query struct {
 	conn      *sql.DB
 }
 
-func NewQuery() *Query {
+func NewQuery(t ...string) *Query {
+
+	var conn_type *sql.DB = DB
+
+	if len(t) > 0 {
+		switch t[0] {
+		case "mysql":
+			conn_type = DB
+
+		case "mssql": //sql server
+			conn_type = MSDB_CONNDB
+
+		}
+	}
+
 	return &Query{
-		conn: DB,
+		conn: conn_type,
 	}
 }