Go mysql返回map類型數據優化

這個版本是根據上個版本修改"database/sql"源碼優化後得到的，暫時還是隻支持返回map[string]string結果。

本次修改沒有修改"database/sql"下的源碼，是對底層查詢結果的一層包裝。

下面是主要代碼，使用方法參考測試代碼。

package dbMap

import (
	. "database/sql"
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"time"
)

//數據庫類型
const driverName = "mysql"

type ConnectPool struct {
	USERNAME string
	PASSWORD string
	NETWORK  string
	SERVER   string
	PORT     string
	DATABASE string
	con      *DB
}

func NewConnectPool(username, password, host, port, database string) *ConnectPool {
	/**
	username
	返回一個連接池的地址
	 */
	return &ConnectPool{
		username, password, "tcp", host, port, database, nil,
	}
}
func (c *ConnectPool) connect() error {
	ServerInfo := fmt.Sprintf("%s:%s@%s(%s:%s)/%s?charset=utf8", c.USERNAME, c.PASSWORD, c.NETWORK, c.SERVER, c.PORT, c.DATABASE)
	DB, err := Open(driverName, ServerInfo)
	if err != nil {
		return err
	}
	DB.SetConnMaxLifetime(100 * time.Second) //最大連接週期，超過時間的連接就close
	DB.SetMaxOpenConns(100)                  //設置最大連接數
	DB.SetMaxIdleConns(16)                   //設置閒置連接數
	c.con = DB
	return nil
}
func (c *ConnectPool) Get() (*DB, error) {
	/**
	返回一個數據庫連接
	*/
	if c.con != nil {
		return c.con, nil
	}
	err := c.connect()
	return c.con, err
}
func (c *ConnectPool) Close() {
	/**
	關閉數據庫連接
	*/
	if c.con != nil {
		_ = c.con.Close()
	}
}
func Query(db *DB, Sql string, args []interface{}, dest []string) ([]map[string]string, error) {
	/*
		數據庫查詢入口 只能返回map[string]string類型結果集
		@params
		Sql: 被執行的sql
		args: Sql的參數  沒有參數傳空interface{}數組 順序和類型一定要和sql中一致
		dest: 返回結果集map中key的名稱
		@result
		[]map[string]string: 查詢結果map對象數組
		error: 錯誤信息,有錯誤返回錯誤信息   沒有錯誤返回nil
	*/
	//定義返回的結果集
	var results []map[string]string
	//定義接收查詢結果切片 interface{}類型
	destP := make([]interface{}, len(dest))
	//把每個結果初始化爲*string
	for i := range dest {
		destP[i] = new(string)
	}
	//調用DB查詢 得到遊標rows
	rows, err := db.Query(Sql, args...)
	if err != nil {
		return results, err
	}
	//循環取出所有行
	for rows.Next() {
		//定義存儲單行查詢結果的map對象
		result := make(map[string]string)
		//取出當前行的查詢結果到destP中
		err = rows.Scan(destP...)
		if err != nil {
			return results, err
		}
		//循環把destP中的結果按照dest的key存儲到result中
		for i := range dest {
			//destP[i]中存儲的是interface{}類型的string指針
			//destP[i].(*string)是string指針值
			//*destP[i].(*string)是string的值
			result[dest[i]] = *destP[i].(*string)
		}
		//把單行查詢結果添加到results中
		results = append(results, result)
	}
	return results, err
}

附測試代碼：

package dbMap_test

import (
	"dbMap"
	"fmt"
	"testing"
)

const (
	USERNAME = "mysqlsa"
	PASSWORD = "Lansi123"
	NETWORK  = "tcp"
	SERVER   = "rm-uf6sjzuqdryz62fs7lo.mysql.rds.aliyuncs.com"
	PORT     = "3306"
	DATABASE = "lansi_sy_evaluate"
)

func TestQuery(t *testing.T) {
	var pool = dbMap.ConnectPool{USERNAME: USERNAME, PASSWORD: PASSWORD, NETWORK: NETWORK, SERVER: SERVER, PORT: PORT, DATABASE: DATABASE}
	db, err := pool.Get()
	if err != nil {
		fmt.Println("數據庫連接失敗")
		fmt.Println(err)
		return
	}
	//sql格式化參數
	args := []interface{}{}
	//查詢結果字段名稱  傳入Database.Query時的順序必須要和sql查詢字段順序一致
	resultNames := []string{"nickname", "inserttime"}
	res, err := dbMap.Query(db, "select nickname,inserttime from sy_account", args, resultNames)
	if err != nil {
		fmt.Println(err)
	} else {
		fmt.Println("批量查詢")
		for _, r := range res {
			fmt.Println(r)
		}
	}

	args = []interface{}{25, "nick_admin"}
	res, err = dbMap.Query(db, "select nickname,inserttime from sy_account where id=? and nickname=? limit 1", args, resultNames)
	if err != nil {
		fmt.Println(err)
	} else {
		fmt.Println("單條查詢")
		fmt.Println((res)[0])
	}
}
func TestConnectPool_Get(t *testing.T) {
	var pool = dbMap.ConnectPool{USERNAME: USERNAME, PASSWORD: PASSWORD, NETWORK: NETWORK, SERVER: SERVER, PORT: PORT, DATABASE: DATABASE}
	_, err := pool.Get()
	if err != nil {
		fmt.Println("數據庫連接失敗")
		fmt.Println(err)
		return
	}
	pool.Close()
}

嫌複製麻煩點這裏直接去github下載代碼