From 97d26083462c8b3a4ff8eebedeffad225a94b381 Mon Sep 17 00:00:00 2001 From: azw Date: Sun, 20 Aug 2023 14:59:56 +0800 Subject: [PATCH] 1 --- dbpool.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/dbpool.go b/dbpool.go index d7404f7..faa00d6 100644 --- a/dbpool.go +++ b/dbpool.go @@ -7,6 +7,7 @@ import ( "math" "strings" "sync" + "errors" ) type DBStyle int32 @@ -243,6 +244,10 @@ func (this *dbPool) joinInsertFields(fieldsKv [][]string, params *[]string) stri func (this *dbPool) internalExec(dataSource string, sql string, params []string, cb ExecResultCb) { ds := this.borrowConn(dataSource) + if ds == nil { + cb(errors.New("borrowConn error"), int64(0), int64(0)) + return + } result, err := ds.conn.Exec(sql, q5.ToInterfaces(params)...) this.returnConn(ds) @@ -269,6 +274,10 @@ func (this *dbPool) internalExec(dataSource string, sql string, params []string, func (this *dbPool) internalQuery(dataSource string, sql string, params []string, cb QueryResultCb) { ds := this.borrowConn(dataSource) + if ds == nil { + cb(errors.New("borrowConn error"), int64(0), int64(0)) + return + } rows, err := ds.conn.Query(sql, q5.ToInterfaces(params)...) this.returnConn(ds) @@ -285,6 +294,10 @@ func (this *dbPool) internalQuery(dataSource string, sql string, params []string func (this *dbPool) internalQueryOne(dataSource string, sql string, params []string, cb QueryOneCb) { ds := this.borrowConn(dataSource) + if ds == nil { + cb(errors.New("borrowConn error"), int64(0), int64(0)) + return + } rows, err := ds.conn.Query(sql, q5.ToInterfaces(params)...) this.returnConn(ds)