Device/SERVER/Meter_Service/core/db/db.go

package db

import (
	"MeterService/core/utils"
	"database/sql"
	"sync"
)

var (
	err  error
	myDb *MyDB
)

var once sync.Once

type MyDB struct {
	DB *sql.DB
}

// OpenDb 获取数据库连接
func OpenDb(dbFun func() *sql.DB) {
	once.Do(func() {
		myDb = &MyDB{DB: dbFun()}
	})
}
func GetDb() *MyDB {
	return myDb
}
func CloseDb() {
	err := myDb.DB.Close()
	if err != nil {
		return
	} else {
		panic(err)
	}
}

// Exec 增、删、改
func (db *MyDB) Exec(SQL string, args ...interface{}) (sql.Result, error) {
	//DB := OpenDb().DB()
	var ret sql.Result
	if args == nil {
		ret, err = db.DB.Exec(SQL)
	} else {
		ret, err = db.DB.Exec(SQL, args...)
	}
	if err != nil {
		return nil, err
	}
	return ret, nil
}

// Query 查询
func (db *MyDB) Query(SQL string, args ...interface{}) ([]map[string]string, bool) { //通用查询
	var rows *sql.Rows
	if args == nil {
		rows, err = db.DB.Query(SQL)
	} else {
		rows, err = db.DB.Query(SQL, args...)
	}
	//rows, err := DB.Query(SQL, args) //执行SQL语句，比如select * from users
	if err != nil {
		panic(err)
	}
	columns, _ := rows.Columns()            //获取列的信息
	count := len(columns)                   //列的数量
	var values = make([]interface{}, count) //创建一个与列的数量相当的空接口
	for i, _ := range values {
		var ii interface{} //为空接口分配内存
		values[i] = &ii    //取得这些内存的指针，因后继的Scan函数只接受指针
	}
	ret := make([]map[string]string, 0) //创建返回值：不定长的map类型切片
	for rows.Next() {
		err := rows.Scan(values...)  //开始读行，Scan函数只接受指针变量
		m := make(map[string]string) //用于存放1列的 [键/值] 对
		if err != nil {
			panic(err)
		}
		for i, colName := range columns {
			var rawValue = *(values[i].(*interface{})) //读出raw数据，类型为byte
			//b, _ := rawValue.([]byte)
			//v := string(b) //将raw数据转换成字符串
			m[colName], err = utils.ToString(rawValue) //colName是键，v是值
			if err != nil {
				panic(err)
			}
		}
		ret = append(ret, m) //将单行所有列的键值对附加在总的返回值上（以行为单位）
	}

	defer func(rows *sql.Rows) {
		_ = rows.Close()
	}(rows)

	if len(ret) != 0 {
		return ret, true
	}
	return nil, false
}

func NewDB(db *sql.DB) *MyDB {
	return &MyDB{DB: db}
}