客服系统即时通讯IM开发(六)Glang Gorm 执行原生Sql语句增删改查封装库【唯一客服】网站在线客服系统
2023-06-13 09:17:02 时间
在开发在线客服系统的时候,有某些地方需要使用脚本去批量执行SQL语句,这个时候就需要使用简单的执行SQL的封装函数了
查询操作是使用的原生的sql库,没用Gorm,原因是Gorm的Raw Scan必须要定义好结构体,原生的可以不需要了。执行查询语句时,将查询结果返回为[]map[string]interface{},其中每个map表示一条记录,key为列名,value为该列的字符串值。
- 数据库连接(使用gorm.Open)
- 执行查询语句(使用rows.Scan)
- 执行增删改语句(使用db.Exec)
- 执行SQL文件(读取文件内容,执行每个分号分隔的SQL语句)
package lib
import (
"database/sql"
"errors"
"fmt"
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/mysql"
"io/ioutil"
"log"
"os"
"strings"
"time"
)
type DBTool struct {
SqlPath string
Username, Password, Server, Port, Database string
}
func (this *DBTool) connect() (*gorm.DB, error) {
dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", this.Username, this.Password, this.Server, this.Port, this.Database)
db, err := gorm.Open("mysql", dsn)
if err != nil {
log.Println("数据库连接失败:", err)
//panic("数据库连接失败!")
return nil, err
}
db.SingularTable(true)
db.LogMode(true)
db.DB().SetMaxIdleConns(10)
db.DB().SetMaxOpenConns(100)
db.DB().SetConnMaxLifetime(59 * time.Second)
return db, nil
}
//执行查询语句
func (this *DBTool) QuerySql(sqlStr string) ([]map[string]interface{}, error) {
db, err := this.connect()
var result []map[string]interface{}
if err != nil {
return result, err
}
sqlStr = strings.TrimSpace(sqlStr)
if sqlStr == "" {
return result, errors.New("sql语句为空")
}
rows, err := db.DB().Query(sqlStr)
if err != nil {
return result, err
}
//获取列名
columns, _ := rows.Columns()
//定义一个切片,长度是字段的个数,切片里面的元素类型是sql.RawBytes
values := make([]sql.RawBytes, len(columns))
//定义一个切片,元素类型是interface{} 接口
scanArgs := make([]interface{}, len(values))
for i := range values {
//把sql.RawBytes类型的地址存进去了
scanArgs[i] = &values[i]
}
//获取字段值
for rows.Next() {
res := make(map[string]interface{})
rows.Scan(scanArgs...)
for i, col := range values {
res[columns[i]] = string(col)
}
result = append(result, res)
}
return result, nil
}
//执行增删改
func (this *DBTool) ExecuteSql(sql string) error {
db, err := this.connect()
sql = strings.TrimSpace(sql)
if sql == "" {
return errors.New("sql语句为空")
}
err = db.Exec(sql).Error
if err != nil {
log.Println("执行失败:" + err.Error())
return err
} else {
log.Println(sql, "\t success!")
}
return nil
}
//执行sql文件
func (this *DBTool) ImportSql() error {
_, err := os.Stat(this.SqlPath)
if os.IsNotExist(err) {
log.Println("数据库SQL文件不存在:", err)
return err
}
db, err := this.connect()
sqls, _ := ioutil.ReadFile(this.SqlPath)
sqlArr := strings.Split(string(sqls), ";")
for _, sql := range sqlArr {
sql = strings.TrimSpace(sql)
if sql == "" {
continue
}
err := db.Exec(sql).Error
if err != nil {
log.Println("数据库导入失败:" + err.Error())
return err
} else {
log.Println(sql, "\t success!")
}
}
return nil
}
使用案例
db := &DBTool{
Username: "BT_DB_USERNAME",
Password: "BT_DB_PASSWORD",
Server: "127.0.0.1",
Port: "3306",
Database: "BT_DB_NAME",
}
res, err := db.QuerySql("select * from user")
log.Println(res, err)
相关文章
- SQL开发知识:oracle多条sql执行中间的语句出现错误时的处理解决方式
- SQL Server 获取服务器时间的sql语句
- Sql Server中的系统视图详细介绍
- 比较SQL和MySQL:谁更适合你学习?(sql和mysql学哪个)
- 轻松掌握:使用SQL Server执行SQL脚本(sqlserver执行sql脚本)
- Linux运行 SQL: 获取自动化数据操作能力(linux执行sql)
- 语句MySQL 创建表:使用SQL语句实现.(mysql创建表sql)
- Oracle数据库SQL语句的导出与备份(oracle导出sql)
- MySQL查找提升系统性能的慢SQL(mysql查看慢sql)
- 运算SQL Server 不等号运算实现准确查询(sqlserver不等)
- 语句执行记录MySQL 查看SQL语句执行记录(mysql查看sql)
- 使用MySQL执行SQL指令:简单易学的基础教程(mysql运行sql命令)
- MySQL如何输出SQL语句?25字教程速成!(mysql输出sql)
- 语句执行情况分析Oracle中当天SQL语句的执行情况(oracle当天sql)
- SQL服务开启:突破MS SQL的瓶颈(sql服务开启mssql)
- Linux环境下执行SQL语句的步骤与方法(linux下执行sql)
- 深入了解MySQL事务SQL的特性和用法(mysql事务sql)
- 安装SQL Server月初新系统安装实践经验(sqlserver 月初)
- MSSQL实现SQL注入防护的完美解决方案(mssql 防sql注入)
- MSSQL恢复数据:从SQL文件进行还原(mssql还原sql文件)
- 改善MySQL慢SQL语句,提升系统性能(mysql慢sql语句)
- MySQL如何还原SQL文件(mysql还原sql文件)
- 一步搞定!MySQL命令行快速导入SQL文件(mysql命令导入sql文件)
- Linux下执行SQL脚本:灵活、快捷、方便!(linux执行sql脚本)
- MySQL报错无法执行SQL,该如何解决(mysql不能执行sql)
- SQL与Redis结合构建高效存储模型(在sql 中加redis)
- 掌握Oracle SQL写法,激发你的SQL能力(oracle sql写法)
- Sql学习第一天——SQL练习题(建表/sql语句)