Go语言开发点餐系统中的订单查询功能实现方法

type Order struct {
    ID       uint   `gorm:"primary_key"`
    UserID   uint   `gorm:"not null"`
    Amount   uint   `gorm:"not null"`
    Status   string `gorm:"not null"`
    CreatedAt time.Time
    UpdatedAt time.Time
}

import (
    "fmt"
    "GitHub.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mysql"
)

func ConnectDB() (*gorm.DB, error) {
    db, err := gorm.Open("mysql", "root:@/orders_db?charset=utf8&parseTime=True&loc=Local")
    if err != nil {
        return nil, err
    }
    fmt.Println("Database connection established")
    return db, nil
}

import (
    "github.com/gin-gonic/gin"
    "net/http"
)

func GetOrders(c *gin.Context) {
    user_id := c.Query("user_id")
    db, err := ConnectDB()
    if err != nil {
        c.JSON(http.StatusInternalServerError, err.Error())
        return
    }
    defer db.Close()

    var orders []Order
    db.Where("user_id=?", user_id).Find(&orders)
    c.jsON(http.StatusOK, orders)
}

import (
    "encoding/json"
    "github.com/stretchr/testify/assert"
    "net/http"
    "net/http/httptest"
    "testing"
)

func TestGetOrders(t *testing.T) {
    router := gin.Default()
    router.GET("/orders", GetOrders)

    w := httptest.NewRecorder()
    req, _ := http.NewRequest("GET", "/orders?user_id=1", nil)
    router.ServeHTTP(w, req)

    assert.Equal(t, http.StatusOK, w.Code)

    var orders []Order
    json.Unmarshal(w.Body.Bytes(), &orders)
    assert.Equal(t, 1, len(orders))
}