go语言gin web 对mysql增删改查
1、连接mysql go 文件
package dbs
import (
"database/sql"
"fmt"
"log"
"time"
_ "github.com/go-sql-driver/mysql"
)
var MysqlDb *sql.DB
var MysqlDbErr error
// 初始化链接
func InitDB() (err error) {
dbDSN := fmt.Sprintf("root:root_pwd,1q,1@tcp(*.*.*.*:port)/tty01?charset=utf8")
// 打开连接失败
MysqlDb, MysqlDbErr = sql.Open("mysql", dbDSN)
//defer MysqlDb.Close();
if MysqlDbErr != nil {
log.Println("dbDSN: " + dbDSN)
panic("数据源配置不正确: " + MysqlDbErr.Error())
}
// 最大连接数
MysqlDb.SetMaxOpenConns(100)
// 闲置连接数
MysqlDb.SetMaxIdleConns(20)
// 最大连接周期
MysqlDb.SetConnMaxLifetime(100 * time.Second)
if MysqlDbErr = MysqlDb.Ping(); nil != MysqlDbErr {
panic("数据库链接失败: " + MysqlDbErr.Error())
}
return
}
2、 web main入口文件 操作mysql
package main
import (
"fmt"
"net/http"
"web_restful_api_mysql/dbs"
"github.com/gin-gonic/gin"
_ "github.com/go-sql-driver/mysql"
)
type User struct {
Id int `json:"id"`
Name string `json:"name"`
City string `json:"city"`
Age int `json:"age"`
}
func main() {
err := dbs.InitDB()
if err != nil {
fmt.Println("err ", err)
}
// // 通过切片存储
// users := make([]User, 0)
// rows, _ := dbs.MysqlDb.Query("SELECT * FROM book")
// // 遍历
// var user User
// for rows.Next() {
// rows.Scan(&user.Id, &user.Name, &user.Age)
// users = append(users, user)
// }
// fmt.Println(users, "长度", len(users))
// for i := 0; i < len(users); i++ {
// fmt.Println("id ， title ， price")
// fmt.Println(users[i].Id, users[i].Name, users[i].Age)
// }
router := gin.Default()
//根据id 展示
// router.GET("/user/:id", FindUser)
// 查询all
router.GET("user/all", UserList)
// 新增
router.POST("/user/add", UserAdd)
//修改
router.POST("/user/update", UpdateUser)
// 删除
router.POST("/user/delete", DeleteUser)
router.Run(":8001")
}
func FindUser(c *gin.Context) {
var (
user User
result gin.H
)
id := c.Param("id")
row := dbs.MysqlDb.QueryRow("select * from t_user where id = ?;", id)
err := row.Scan(&user.Id, &user.Name, &user.City, &user.Age)
if err != nil {
result = gin.H{
"result": nil,
"count": 0,
}
} else {
result = gin.H{
"result": user,
"count": 1,
}
}
c.JSON(http.StatusOK, result)
}
func UserList(c *gin.Context) {
var (
user User
users []User
)
rows, err := dbs.MysqlDb.Query("select * from t_user")
if err != nil {
fmt.Println("err==>", err.Error())
}
for rows.Next() {
err = rows.Scan(&user.Id, &user.Name, &user.City, &user.Age)
if err != nil {
fmt.Println("err ==>", err.Error())
}
users = append(users, user)
}
defer rows.Close()
c.JSON(http.StatusOK, gin.H{
"result": users,
"count": len(users),
})
}
func UserAdd(c *gin.Context) {
//获取 post接入值
// var buffer bytes.Buffer
fmt.Println("来了")
name := c.PostForm("name")
city := c.PostForm("city")
age := c.PostForm("age")
fmt.Println("来了", name, city, age)
stmt, err := dbs.MysqlDb.Prepare("insert into t_user (name,city,age) values (?,?,?)")
addRet, _ := stmt.Exec(name, city, age)
ins_nums, _ := addRet.RowsAffected()
if ins_nums > 0 {
if err != nil {
fmt.Println("err ->", err.Error())
}
_, err = stmt.Exec(name, city, age)
if err != nil {
fmt.Println("err-<", err.Error())
}
c.JSON(http.StatusOK, gin.H{
"message": "success",
"code": fmt.Sprintf("新插入数据： %s", name),
})
} else {
c.JSON(http.StatusOK, gin.H{
"message": fmt.Sprintf("%s add error created", name),
"code": 333,
})
}
}
func UpdateUser(c *gin.Context) {
id := c.PostForm("id")
// id := c.Param("id")
name := c.PostForm("name")
city := c.PostForm("city")
age := c.PostForm("age")
stmt, err := dbs.MysqlDb.Prepare("update t_user set name=?,city=?,age=? where id=?")
if err != nil {
fmt.Println("err--", err.Error())
}
update, _ := stmt.Exec(name, city, age, id)
ins_nus, _ := update.RowsAffected()
if ins_nus > 0 {
fmt.Println("修改成功", id)
c.JSON(http.StatusOK,gin.H{
"message": fmt.Sprintf("id : %s 修改成功",id),
"code" : 200,
})
} else {
fmt.Println("修改失败")
c.JSON(http.StatusOK,gin.H{
"message" : fmt.Sprintf("id ： %s 修改失败",id),
"code": 200,
})
}
}
func DeleteUser(c *gin.Context) {
id := c.PostForm("id")
stmt, err := dbs.MysqlDb.Prepare("delete from t_user where id=?")
if err != nil {
fmt.Println("err", err)
}
deleteUser, _ := stmt.Exec(id)
ins_nus, _ := deleteUser.RowsAffected()
if ins_nus > 0 {
fmt.Println("删除成功")
c.JSON(http.StatusOK, gin.H{
"message": "delete ok",
"code": fmt.Sprintf("删除id: %s", id),
})
} else {
fmt.Println("删除失败")
c.JSON(http.StatusOK, gin.H{
"message": "delete error",
"code": fmt.Sprintf("删除 id : %s 失败", id),
})
}
}
