Gin篇:gorm 使用

 

簡單使用

創建 main.go

 go get github.com/go-sql-driver/mysql 
 go get github.com/jinzhu/gorm

代碼

package main

import (
    _ "github.com/go-sql-driver/mysql"//加載mysql
    "github.com/jinzhu/gorm"//gorm 擴展包
    "fmt"
)
//注意如果 定義成小寫username 引用時 無法調用
type User struct {
    ID       int64  // 列名爲 `id`
    Username string // 列名爲 `username`
    Password string // 列名爲 `password`
}

//設置表名
func (User) TableName() string {
    return "users"
}

func main() {
    db, err := gorm.Open("mysql", "數據庫用戶名:數據庫密碼@tcp(ip地址:端口號)/數據庫名?     charset=utf8&parseTime=True&loc=Local&timeout=10ms")
     defer db.Close()
    if err != nil {
        fmt.Printf("mysql connect error %v", err)
    }

    //執行遷移文件 生成數據表
    db.CreateTable(&User{})

    //添加數據
    user := User{Username: "root", Password: "root"}
    result := db.Create(&user)

    if result.Error != nil {
        fmt.Printf("insert row err %v", result.Error)
        return
    }

    fmt.Println(user.ID) //返回id

    //查詢單條數據
    getUser := User{}

    //SELECT id, first FROM users WHERE id = 1 LIMIT 1;
    db.Select([]string{"id", "username"}).First(&getUser, 1)
    fmt.Println(getUser) //打印查詢數據

    //修改數據
    user.Username = "update username"
    user.Password = "update password"
    db.Save(&user)

    //查詢列表數據
    users := []User{}
    db.Find(&users)
    fmt.Println(&users)//獲取所有數據

  //刪除數據
    db.Delete(&user)
}

執行

go run main.go

gin+gorm+router 快速搭建 crud restful API 接口

├──api
│    ├── apis  
│    │    └── user.go                   
│    ├── database 
│    │    └── mysql.go          
│    ├── models 
│    │    └── user.go       
│    └── router 
│         └──  router.go
└──main.go

apis/apis/user.go

package apis

import (
    "github.com/gin-gonic/gin"
    model "api/models"
    "net/http"
    "strconv"
)

//列表數據
func Users(c *gin.Context) {
    var user model.User
    user.Username = c.Request.FormValue("username")
    user.Password = c.Request.FormValue("password")
    result, err := user.Users()

    if err != nil {
        c.JSON(http.StatusOK, gin.H{
            "code":    -1,
            "message": "抱歉未找到相關信息",
        })
        return
    }

    c.JSON(http.StatusOK, gin.H{
        "code": 1,
        "data":   result,
    })
}

//添加數據
func Store(c *gin.Context) {
    var user model.User
    user.Username = c.Request.FormValue("username")
    user.Password = c.Request.FormValue("password")
    id, err := user.Insert()

    if err != nil {
        c.JSON(http.StatusOK, gin.H{
            "code":    -1,
            "message": "添加失敗",
        })
        return
    }
    c.JSON(http.StatusOK, gin.H{
        "code":  1,
        "message": "添加成功",
        "data":    id,
    })
}

//修改數據
func Update(c *gin.Context) {
    var user model.User
    id, err := strconv.ParseInt(c.Param("id"), 10, 64)
    user.Password = c.Request.FormValue("password")
    result, err := user.Update(id)
    if err != nil || result.ID == 0 {
        c.JSON(http.StatusOK, gin.H{
            "code":    -1,
            "message": "修改失敗",
        })
        return
    }
    c.JSON(http.StatusOK, gin.H{
        "code":  1,
        "message": "修改成功",
    })
}

//刪除數據
func Destroy(c *gin.Context) {
    var user model.User
    id, err := strconv.ParseInt(c.Param("id"), 10, 64)
    result, err := user.Destroy(id)
    if err != nil || result.ID == 0 {
        c.JSON(http.StatusOK, gin.H{
            "code":    -1,
            "message": "刪除失敗",
        })
        return
    }
    c.JSON(http.StatusOK, gin.H{
        "code":  1,
        "message": "刪除成功",
    })
}

database/mysql.go

package database

import (
    _ "github.com/go-sql-driver/mysql" //加載mysql
    "github.com/jinzhu/gorm"
    "fmt"
)

var Eloquent *gorm.DB

func init() {
    var err error
    Eloquent, err = gorm.Open("mysql", "root:root@tcp(127.0.0.1:3306)/test?charset=utf8&parseTime=True&loc=Local&timeout=10ms")

    if err != nil {
        fmt.Printf("mysql connect error %v", err)
    }

    if Eloquent.Error != nil {
        fmt.Printf("database error %v", Eloquent.Error)
    }
}

models/user.go

    package models

    import (
        orm "api/database"
    )

    type User struct {
        ID       int64  `json:"id"`       // 列名爲 `id`
        Username string `json:"username"` // 列名爲 `username`
        Password string `json:"password"` // 列名爲 `password`
    }

    var Users []User

    //添加
    func (user User) Insert() (id int64, err error) {

        //添加數據
        result := orm.Eloquent.Create(&user)
        id =user.ID
        if result.Error != nil {
            err = result.Error
            return
        }
        return
    }

    //列表
    func (user *User) Users() (users []User, err error) {
        if err = orm.Eloquent.Find(&users).Error; err != nil {
            return
        }
        return
    }

    //修改
    func (user *User) Update(id int64) (updateUser User, err error) {

        if err = orm.Eloquent.Select([]string{"id", "username"}).First(&updateUser, id).Error; err != nil {
            return
        }

        //參數1:是要修改的數據
        //參數2:是修改的數據
        if err = orm.Eloquent.Model(&updateUser).Updates(&user).Error; err != nil {
            return
        }
        return
    }

    //刪除數據
    func (user *User) Destroy(id int64) (Result User, err error) {

        if err = orm.Eloquent.Select([]string{"id"}).First(&user, id).Error; err != nil {
            return
        }

        if err = orm.Eloquent.Delete(&user).Error; err != nil {
            return
        }
        Result = *user
        return
    }

  

router/router.go

package router

import (
    "github.com/gin-gonic/gin"
    . "api/apis"
)

func InitRouter() *gin.Engine {
    router := gin.Default()

    router.GET("/users", Users)

    router.POST("/user", Store)

    router.PUT("/user/:id", Update)

    router.DELETE("/user/:id", Destroy)

    return router
}

  

main.go

    package main

    import (
        _ "api/database"
        "api/router"
        orm "api/database"
    )

    func main() {
        defer orm.Eloquent.Close()
        router := router.InitRouter()
        router.Run(":8000")
    }

執行 go run main.go

訪問地址

POST localhost:8006/user 添加
GET localhost:8006/users 列表
DELETE localhost:8006/user/id 刪除
PUT localhost:8006/user/id 修改 

原文作者:yourself
轉自鏈接:https://learnku.com/articles/23548/gingormrouter-quickly-build-crud-restful-api-interface
 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章