golang web 开发 —— gin 框架 （gorm 链接 mysql）

在现代Web开发中，Go语言因其高性能和简洁性而受到越来越多开发者的青睐。其中，Gin框架因其高效的路由和中间件支持，成为了Go语言Web开发的热门选择。同时，GORM作为一个强大的Go语言ORM（对象关系映射）库，简化了与数据库的交互，使得开发者可以更加专注于业务逻辑的实现。本文将通过一个示例，介绍如何使用Gin和GORM连接MySQL数据库。

环境准备

首先，确保你已经安装了Go语言环境以及MySQL数据库。在你的项目目录中初始化Go模块：

go mod init my-go-web

然后，安装Gin和GORM以及MySQL驱动：

go get -u github.com/gin-gonic/gin
go get -u gorm.io/gorm
go get -u gorm.io/driver/mysql

创建数据库

在MySQL中创建一个示例数据库和表：

CREATE DATABASE mydb;
USE mydb;

CREATE TABLE users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(100) NOT NULL,
    email VARCHAR(100) NOT NULL UNIQUE
);

代码实现

接下来，在项目目录下创建一个 main.go 文件，并编写以下内容：

package main

import (
    "github.com/gin-gonic/gin"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
    "net/http"
)

// User 模型
type User struct {
    ID    uint   `json:"id" gorm:"primaryKey"`
    Name  string `json:"name"`
    Email string `json:"email" gorm:"unique"`
}

// 初始化 GORM 连接
func setupDatabase() (*gorm.DB, error) {
    dsn := "user:password@tcp(127.0.0.1:3306)/mydb?charset=utf8mb4&parseTime=True&loc=Local"
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        return nil, err
    }
    db.AutoMigrate(&User{})
    return db, nil
}

func main() {
    r := gin.Default()

    db, err := setupDatabase()
    if err != nil {
        panic("failed to connect database")
    }

    // 获取用户列表
    r.GET("/users", func(c *gin.Context) {
        var users []User
        if err := db.Find(&users).Error; err != nil {
            c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
            return
        }
        c.JSON(http.StatusOK, users)
    })

    // 创建新用户
    r.POST("/users", func(c *gin.Context) {
        var user User
        if err := c.ShouldBindJSON(&user); err != nil {
            c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
            return
        }
        db.Create(&user)
        c.JSON(http.StatusCreated, user)
    })

    // 启动服务器
    r.Run(":8080")
}

代码解析

1. 模型定义：我们定义了一个 User 结构体，使用GORM的标签定义了数据库字段和约束。

2. 数据库连接：setupDatabase 函数尝试连接到MySQL数据库，并自动迁移 User 模型。记得将 dsn 中的 user:password 替换为你的MySQL用户名和密码。

3. 路由设置：

4. GET /users 接口返回所有用户。

5. POST /users 接口接收用户JSON数据并创建新用户。

6. 启动服务器：程序监听8080端口，准备接受请求。

测试接口

你可以使用Postman或cURL测试API：

• 获取用户列表：

curl http://localhost:8080/users

• 创建用户：

curl -X POST http://localhost:8080/users -H "Content-Type: application/json" -d '{"name": "Alice", "email": "alice@example.com"}'

总结

通过上述示例，我们成功地利用Gin框架搭建了一个简单的Web应用，使用GORM连接MySQL数据库。这种架构使得开发者能够快速构建高效、易维护的Web应用。随着项目的复杂性增加，可以根据需要添加更多的中间件和功能，进一步拓展Gin和GORM的能力。