首页 > 数据库> 文章详细

GO的MySql预处理

2021-07-02 09:34:15 阅读：226 来源： 互联网

标签：return err nil fmt Age MySql GO 预处理 Name

说明：使用预处理方式执行，是将存储过程预先发送到MySql服务器上执行，之后再传参数上去，可以有效预防数据库的注入。多次循环执行也提高了运行效率。

代码

package main

import (
    "database/sql"
    "fmt"

    _ "github.com/go-sql-driver/mysql"
)

//GO链接MySql
var Db *sql.DB //连接池对象

func initDB() (err error) {
    dsn := "username:password@tcp(127.0.0.1:3306)/database?charset=utf8"
    db, err := sql.Open("mysql", dsn)
    if err != nil {
        return err
    }
    err = db.Ping()
    if err != nil {
        return err
    }
    //设置数据库连接池最大连接数
    db.SetMaxOpenConns(10)
    //设置最大链接空闲数
    db.SetMaxIdleConns(5)
    Db = db
    return nil
}

type User struct {
    Id   int
    Name string
    Age  int
}


//查询
func query(where string) {
    sqlStr := "select Id,Name,Age from User " + where + ";"
    //执行
    rows, err := Db.Query(sqlStr)
    if err != nil {
        fmt.Println("查询错误：%v\n", err)
        return
    }
    //释放连接
    defer rows.Close()
    for rows.Next() {
        var u User
        //赋值并关闭链接（Scan方法）
        err := rows.Scan(&u.Id, &u.Name, &u.Age)
        if err != nil {
            fmt.Println("查询循环错误")
            continue
        }
        fmt.Printf("Id:%v;Name:%v;Age:%v;\n", u.Id, u.Name, u.Age)
    }
}

//预处理（批量插入）
func prepareInsert() {
    sqlStr := `insert into User(Name,Age) value(?,?)`
    stmt, err := Db.Prepare(sqlStr)
    if err != nil {
        fmt.Printf("预处理Sql语句错误:%v\n", err)
        return
    }
    defer stmt.Close()
    var m = map[string]int{
        "张三": 12,
        "李四": 10,
        "王五": 11,
        "赵六": 18,
    }
    //循环执行预处理数据
    for v, k := range m {
        stmt.Exec(v, k)
    }
}

func main() {
    err := initDB()
    if err != nil {
        fmt.Printf("初始化失败，错误%v\n", err)
        return
    }
    fmt.Println("数据库链接成功。")
    fmt.Println("-----------------------------------")
    query("where id > 0")
    prepareInsert()
    fmt.Println("-----------------------------------")
    query("where id > 0")

}

运行结果：

参考：https://www.bilibili.com/video/BV1fz4y1m7Pm?p=136

标签：return,err,nil,fmt,Age,MySql,GO,预处理,Name
来源： https://www.cnblogs.com/dzw159/p/14961846.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

GO的MySql预处理